50 lines
1.2 KiB
JavaScript
50 lines
1.2 KiB
JavaScript
const Database = require('better-sqlite3');
|
|
const path = require('path');
|
|
|
|
class DatabaseWrapper {
|
|
constructor() {
|
|
this.dbPath = process.env.DB_PATH || path.join(__dirname, '../data/database.sqlite');
|
|
this.db = new Database(this.dbPath, { verbose: console.log });
|
|
}
|
|
|
|
// 查询单个记录
|
|
get(sql, params = []) {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
return stmt.get(params);
|
|
} catch (error) {
|
|
console.error('数据库查询错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
// 查询多个记录
|
|
all(sql, params = []) {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
return stmt.all(params);
|
|
} catch (error) {
|
|
console.error('数据库查询错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
// 执行插入、更新、删除操作
|
|
run(sql, params = []) {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
const result = stmt.run(params);
|
|
return { id: result.lastInsertRowid, changes: result.changes };
|
|
} catch (error) {
|
|
console.error('数据库操作错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
// 关闭数据库连接
|
|
close() {
|
|
this.db.close();
|
|
}
|
|
}
|
|
|
|
module.exports = new DatabaseWrapper(); |