d1/sqlite支持

2026-01-30 14:27:43 +08:00
parent c4198e0954
commit c86f9cdca7
11 changed files with 2110 additions and 11 deletions
@@ -0,0 +1,170 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+
+/**
+ * 统一的数据库适配器接口
+ * 兼容 Cloudflare D1 和 better-sqlite3
+ *
+ * 注意：此模块仅在服务端使用，通过 webpack 配置排除客户端打包
+ */
+
+// D1 PreparedStatement 接口
+export interface D1PreparedStatement {
+  bind(...values: any[]): D1PreparedStatement;
+  first<T = any>(colName?: string): Promise<T | null>;
+  run<T = any>(): Promise<D1Result<T>>;
+  all<T = any>(): Promise<D1Result<T>>;
+}
+
+export interface D1Result<T = any> {
+  results?: T[];
+  success: boolean;
+  meta?: any;
+  error?: string;
+}
+
+// 统一的数据库接口
+export interface DatabaseAdapter {
+  prepare(query: string): D1PreparedStatement;
+  batch?(statements: D1PreparedStatement[]): Promise<D1Result[]>;
+  exec?(query: string): void;
+}
+
+/**
+ * Cloudflare D1 适配器（生产环境）
+ */
+export class CloudflareD1Adapter implements DatabaseAdapter {
+  constructor(private db: D1Database) {}
+
+  prepare(query: string): D1PreparedStatement {
+    return this.db.prepare(query);
+  }
+
+  async batch(statements: D1PreparedStatement[]): Promise<D1Result[]> {
+    return this.db.batch(statements as any);
+  }
+}
+
+/**
+ * SQLite 适配器（开发环境）
+ * 包装 better-sqlite3 以兼容 D1 API
+ */
+export class SQLiteAdapter implements DatabaseAdapter {
+  private db: any; // better-sqlite3 Database
+
+  constructor(db: any) {
+    this.db = db;
+  }
+
+  prepare(query: string): D1PreparedStatement {
+    const stmt = this.db.prepare(query);
+    return new SQLitePreparedStatement(stmt);
+  }
+
+  batch(statements: D1PreparedStatement[]): Promise<D1Result[]> {
+    // SQLite 使用事务模拟 batch
+    return new Promise((resolve, reject) => {
+      try {
+        const results: D1Result[] = [];
+        const transaction = this.db.transaction(() => {
+          for (const stmt of statements) {
+            const result = (stmt as any).runSync();
+            results.push(result);
+          }
+        });
+        transaction();
+        resolve(results);
+      } catch (err) {
+        reject(err);
+      }
+    });
+  }
+
+  exec(query: string): void {
+    this.db.exec(query);
+  }
+}
+
+/**
+ * SQLite PreparedStatement 包装器
+ * 将 better-sqlite3 API 转换为 D1 兼容 API
+ */
+class SQLitePreparedStatement implements D1PreparedStatement {
+  private stmt: any;
+  private params: any[] = [];
+
+  constructor(stmt: any) {
+    this.stmt = stmt;
+  }
+
+  bind(...values: any[]): D1PreparedStatement {
+    this.params = values;
+    return this;
+  }
+
+  async first<T = any>(colName?: string): Promise<T | null> {
+    try {
+      const result = this.stmt.get(...this.params);
+      if (!result) return null;
+      if (colName) return result[colName] ?? null;
+      return result;
+    } catch (err) {
+      console.error('SQLite first() error:', err);
+      return null;
+    }
+  }
+
+  async run<T = any>(): Promise<D1Result<T>> {
+    try {
+      const info = this.stmt.run(...this.params);
+      return {
+        success: true,
+        meta: {
+          changes: info.changes,
+          last_row_id: info.lastInsertRowid,
+        },
+      };
+    } catch (err: any) {
+      console.error('SQLite run() error:', err);
+      return {
+        success: false,
+        error: err.message,
+      };
+    }
+  }
+
+  async all<T = any>(): Promise<D1Result<T>> {
+    try {
+      const results = this.stmt.all(...this.params);
+      return {
+        success: true,
+        results: results || [],
+      };
+    } catch (err: any) {
+      console.error('SQLite all() error:', err);
+      return {
+        success: false,
+        error: err.message,
+        results: [],
+      };
+    }
+  }
+
+  // 同步版本（用于 batch）
+  runSync(): D1Result {
+    try {
+      const info = this.stmt.run(...this.params);
+      return {
+        success: true,
+        meta: {
+          changes: info.changes,
+          last_row_id: info.lastInsertRowid,
+        },
+      };
+    } catch (err: any) {
+      return {
+        success: false,
+        error: err.message,
+      };
+    }
+  }
+}