feat：现有的 KV 存储数据迁移到 D1 数据库

functions/utils/databaseAdapter.js

@@ -0,0 +1,206 @@
+/**
+ * 数据库适配器
+ * 提供统一的接口，可以在KV和D1之间切换
+ */
+
+import { D1Database } from './d1Database.js';
+
+/**
+ * 创建数据库适配器
+ * @param {Object} env - 环境变量
+ * @returns {Object} 数据库适配器实例
+ */
+export function createDatabaseAdapter(env) {
+    // 检查是否配置了D1数据库
+    if (env.DB && typeof env.DB.prepare === 'function') {
+        // 使用D1数据库
+        console.log('Using D1 Database');
+        return new D1Database(env.DB);
+    } else if (env.img_url) {
+        // 回退到KV存储
+        console.log('Using KV Storage (fallback)');
+        return new KVAdapter(env.img_url);
+    } else {
+        throw new Error('No database configured. Please configure either D1 (env.DB) or KV (env.img_url)');
+    }
+}
+
+/**
+ * KV适配器类
+ * 保持与原有KV接口的兼容性
+ */
+class KVAdapter {
+    constructor(kv) {
+        this.kv = kv;
+    }
+
+    // 直接代理到KV的方法
+    async put(key, value, options = {}) {
+        return await this.kv.put(key, value, options);
+    }
+
+    async get(key) {
+        return await this.kv.get(key);
+    }
+
+    async getWithMetadata(key) {
+        return await this.kv.getWithMetadata(key);
+    }
+
+    async delete(key) {
+        return await this.kv.delete(key);
+    }
+
+    async list(options = {}) {
+        return await this.kv.list(options);
+    }
+
+    // 为了兼容性，添加一些别名方法
+    async putFile(fileId, value, options) {
+        return await this.put(fileId, value, options);
+    }
+
+    async getFile(fileId) {
+        const result = await this.getWithMetadata(fileId);
+        return result;
+    }
+
+    async getFileWithMetadata(fileId) {
+        return await this.getWithMetadata(fileId);
+    }
+
+    async deleteFile(fileId) {
+        return await this.delete(fileId);
+    }
+
+    async listFiles(options) {
+        return await this.list(options);
+    }
+
+    async putSetting(key, value) {
+        return await this.put(key, value);
+    }
+
+    async getSetting(key) {
+        return await this.get(key);
+    }
+
+    async deleteSetting(key) {
+        return await this.delete(key);
+    }
+
+    async listSettings(options) {
+        return await this.list(options);
+    }
+
+    async putIndexOperation(operationId, operation) {
+        const key = 'manage@index@operation_' + operationId;
+        return await this.put(key, JSON.stringify(operation));
+    }
+
+    async getIndexOperation(operationId) {
+        const key = 'manage@index@operation_' + operationId;
+        const result = await this.get(key);
+        return result ? JSON.parse(result) : null;
+    }
+
+    async deleteIndexOperation(operationId) {
+        const key = 'manage@index@operation_' + operationId;
+        return await this.delete(key);
+    }
+
+    async listIndexOperations(options) {
+        const listOptions = {
+            ...options,
+            prefix: 'manage@index@operation_'
+        };
+        const result = await this.list(listOptions);
+        
+        // 转换格式以匹配D1Database的返回格式
+        const operations = [];
+        for (const item of result.keys) {
+            const operationData = await this.get(item.name);
+            if (operationData) {
+                const operation = JSON.parse(operationData);
+                operations.push({
+                    id: item.name.replace('manage@index@operation_', ''),
+                    type: operation.type,
+                    timestamp: operation.timestamp,
+                    data: operation.data,
+                    processed: false // KV中没有这个字段，默认为false
+                });
+            }
+        }
+        
+        return operations;
+    }
+}
+
+/**
+ * 获取数据库实例的便捷函数
+ * 这个函数可以在整个应用中使用，确保一致的数据库访问
+ * @param {Object} env - 环境变量
+ * @returns {Object} 数据库实例
+ */
+export function getDatabase(env) {
+    return createDatabaseAdapter(env);
+}
+
+/**
+ * 检查数据库配置
+ * @param {Object} env - 环境变量
+ * @returns {Object} 配置信息
+ */
+export function checkDatabaseConfig(env) {
+    const hasD1 = env.DB && typeof env.DB.prepare === 'function';
+    const hasKV = env.img_url && typeof env.img_url.get === 'function';
+    
+    return {
+        hasD1,
+        hasKV,
+        usingD1: hasD1,
+        usingKV: !hasD1 && hasKV,
+        configured: hasD1 || hasKV
+    };
+}
+
+/**
+ * 数据库健康检查
+ * @param {Object} env - 环境变量
+ * @returns {Promise<Object>} 健康检查结果
+ */
+export async function healthCheck(env) {
+    const config = checkDatabaseConfig(env);
+    
+    if (!config.configured) {
+        return {
+            healthy: false,
+            error: 'No database configured',
+            config
+        };
+    }
+    
+    try {
+        const db = getDatabase(env);
+        
+        if (config.usingD1) {
+            // D1健康检查 - 尝试查询一个简单的表
+            const stmt = db.db.prepare('SELECT 1 as test');
+            await stmt.first();
+        } else {
+            // KV健康检查 - 尝试列出键
+            await db.list({ limit: 1 });
+        }
+        
+        return {
+            healthy: true,
+            config
+        };
+    } catch (error) {
+        return {
+            healthy: false,
+            error: error.message,
+            config
+        };
+    }
+}