57 lines
1.4 KiB
TypeScript
57 lines
1.4 KiB
TypeScript
import Database from 'better-sqlite3';
|
|
import path from 'path';
|
|
import fs from 'fs';
|
|
|
|
class DatabaseWrapper {
|
|
private db: Database.Database;
|
|
private dbPath: string;
|
|
|
|
constructor() {
|
|
this.dbPath = process.env.DB_PATH || path.join(process.cwd(), 'data/database.sqlite');
|
|
const dbDir = path.dirname(this.dbPath);
|
|
|
|
if (!fs.existsSync(dbDir)) {
|
|
fs.mkdirSync(dbDir, { recursive: true });
|
|
}
|
|
|
|
this.db = new Database(this.dbPath, { verbose: console.log });
|
|
}
|
|
|
|
get<T = any>(sql: string, params: any[] = []): T | undefined {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
return stmt.get(params) as T | undefined;
|
|
} catch (error) {
|
|
console.error('数据库查询错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
all<T = any>(sql: string, params: any[] = []): T[] {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
return stmt.all(params) as T[];
|
|
} catch (error) {
|
|
console.error('数据库查询错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
run(sql: string, params: any[] = []): { id: number; changes: number } {
|
|
try {
|
|
const stmt = this.db.prepare(sql);
|
|
const result = stmt.run(params);
|
|
return { id: result.lastInsertRowid as number, changes: result.changes };
|
|
} catch (error) {
|
|
console.error('数据库操作错误:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
close(): void {
|
|
this.db.close();
|
|
}
|
|
}
|
|
|
|
export default new DatabaseWrapper();
|