refactor(database): unify database connections and add retry logic

Refactor LocalDataSource to reuse the shared database connection from database.ts, avoiding OPFS file handle conflicts. Add retry logic with exponential backoff for database open operations, particularly important for Web/OPFS environments. Also add dynamic theme support to chat input and message bubble styles, using theme colors instead of hardcoded values for better light/dark mode support.

BREAKING CHANGE: Database initialization now requires userId to be passed explicitly for user-specific databases

src/data/datasources/LocalDataSource.ts

@@ -1,41 +1,37 @@
 /**
  * 本地数据库数据源实现
- * 封装所有 SQLite 操作，提供统一的错误处理和事务支持
+ * 复用 database.ts 的共享连接，避免 OPFS 句柄冲突
  */
 
 import * as SQLite from 'expo-sqlite';
 import { ILocalDataSource, DataSourceError } from './interfaces';
+import {
+  initDatabase,
+  getCurrentDbUserId,
+  isDbInitialized,
+  getDbInstance,
+} from '../../services/database';
 
-// 数据库实例管理
-let dbInstance: SQLite.SQLiteDatabase | null = null;
-let currentDbName: string | null = null;
 let writeQueue: Promise<void> = Promise.resolve();
 
 export interface LocalDataSourceConfig {
-  dbName?: string;
   userId?: string;
 }
 
 export class LocalDataSource implements ILocalDataSource {
-  private db: SQLite.SQLiteDatabase | null = null;
-  private dbName: string;
+  private userId: string | null = null;
   private initialized = false;
   private initializingPromise: Promise<void> | null = null;
 
   constructor(config: LocalDataSourceConfig = {}) {
-    // 如果提供了userId，使用用户专属数据库
-    this.dbName = config.dbName || (config.userId ? `carrot_bbs_${config.userId}.db` : 'carrot_bbs.db');
+    this.userId = config.userId || null;
   }
 
-  /**
-   * 初始化数据库连接
-   */
   async initialize(): Promise<void> {
-    if (this.initialized && this.db) {
+    if (this.initialized && isDbInitialized()) {
       return;
     }
 
-    // 防止并发初始化导致连接状态竞争
     if (this.initializingPromise) {
       await this.initializingPromise;
       return;
@@ -51,131 +47,39 @@ export class LocalDataSource implements ILocalDataSource {
 
   private async doInitialize(): Promise<void> {
     try {
-      // 使用全局实例管理
-      if (dbInstance && currentDbName === this.dbName) {
-        this.db = dbInstance;
-        this.initialized = true;
-        return;
-      }
-
-      // 关闭旧连接
-      if (dbInstance) {
-        try {
-          await dbInstance.closeAsync();
-        } catch (e) {
-          console.warn('关闭旧数据库连接失败:', e);
+      // 如果 database.ts 已经初始化，复用它的连接
+      if (isDbInitialized()) {
+        const currentUserId = getCurrentDbUserId();
+        if (this.userId && this.userId !== currentUserId) {
+          await initDatabase(this.userId);
+        }
+      } else if (this.userId) {
+        await initDatabase(this.userId);
+      } else {
+        const currentUserId = getCurrentDbUserId();
+        if (!currentUserId) {
+          console.warn('[LocalDataSource] 没有用户登录，跳过初始化');
+          return;
         }
       }
 
-      // 创建新连接
-      this.db = await SQLite.openDatabaseAsync(this.dbName);
-      dbInstance = this.db;
-      currentDbName = this.dbName;
+      const db = getDbInstance();
+      if (db) {
+        await db.execAsync(`
+          CREATE TABLE IF NOT EXISTS posts_cache (
+            id TEXT PRIMARY KEY NOT NULL,
+            data TEXT NOT NULL,
+            updatedAt TEXT NOT NULL
+          )
+        `);
+      }
+      
       this.initialized = true;
-
-      // 初始化数据库表结构
-      await this.createTables();
     } catch (error) {
       this.handleError(error, 'INITIALIZE');
     }
   }
 
-  /**
-   * 创建数据库表结构
-   */
-  private async createTables(): Promise<void> {
-    if (!this.db) return;
-
-    const tables = [
-      // 消息表
-      `CREATE TABLE IF NOT EXISTS messages (
-        id TEXT PRIMARY KEY NOT NULL,
-        conversationId TEXT NOT NULL,
-        senderId TEXT NOT NULL,
-        content TEXT,
-        type TEXT DEFAULT 'text',
-        isRead INTEGER DEFAULT 0,
-        createdAt TEXT NOT NULL,
-        seq INTEGER DEFAULT 0,
-        status TEXT DEFAULT 'normal',
-        segments TEXT
-      )`,
-      // 会话表
-      `CREATE TABLE IF NOT EXISTS conversations (
-        id TEXT PRIMARY KEY NOT NULL,
-        participantId TEXT NOT NULL,
-        lastMessageId TEXT,
-        lastSeq INTEGER DEFAULT 0,
-        unreadCount INTEGER DEFAULT 0,
-        createdAt TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 会话缓存表
-      `CREATE TABLE IF NOT EXISTS conversation_cache (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 会话列表缓存表
-      `CREATE TABLE IF NOT EXISTS conversation_list_cache (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 用户缓存表
-      `CREATE TABLE IF NOT EXISTS users (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 当前登录用户缓存
-      `CREATE TABLE IF NOT EXISTS current_user_cache (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 群组缓存表
-      `CREATE TABLE IF NOT EXISTS groups (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-      // 群成员缓存表
-      `CREATE TABLE IF NOT EXISTS group_members (
-        groupId TEXT NOT NULL,
-        userId TEXT NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL,
-        PRIMARY KEY (groupId, userId)
-      )`,
-      // 帖子缓存表
-      `CREATE TABLE IF NOT EXISTS posts_cache (
-        id TEXT PRIMARY KEY NOT NULL,
-        data TEXT NOT NULL,
-        updatedAt TEXT NOT NULL
-      )`,
-    ];
-
-    for (const sql of tables) {
-      await this.db.execAsync(sql);
-    }
-
-    // 创建索引
-    const indexes = [
-      `CREATE INDEX IF NOT EXISTS idx_messages_conversationId ON messages(conversationId)`,
-      `CREATE INDEX IF NOT EXISTS idx_messages_seq ON messages(seq)`,
-      `CREATE INDEX IF NOT EXISTS idx_messages_conversation_seq ON messages(conversationId, seq)`,
-      `CREATE INDEX IF NOT EXISTS idx_group_members_groupId ON group_members(groupId)`,
-    ];
-
-    for (const sql of indexes) {
-      await this.db.execAsync(sql);
-    }
-  }
-
-  /**
-   * 处理错误
-   */
   private handleError(error: unknown, operation: string): never {
     const message = error instanceof Error ? error.message : 'Unknown error';
     throw new DataSourceError(
@@ -186,24 +90,21 @@ export class LocalDataSource implements ILocalDataSource {
     );
   }
 
-  /**
-   * 检查数据库连接
-   */
   private ensureDb(): SQLite.SQLiteDatabase {
-    if (!this.db) {
+    const db = getDbInstance();
+    if (!db) {
       throw new DataSourceError(
         'Database not initialized',
         'DB_NOT_INITIALIZED',
         'LocalDataSource'
       );
     }
-    return this.db;
+    return db;
   }
 
-  // ==================== ILocalDataSource 实现 ====================
-
   async query<T>(sql: string, params?: any[]): Promise<T[]> {
     try {
+      await this.initialize();
       const db = this.ensureDb();
       if (params && params.length > 0) {
         return await db.getAllAsync<T>(sql, params as any);
@@ -216,6 +117,7 @@ export class LocalDataSource implements ILocalDataSource {
 
   async getFirst<T>(sql: string, params?: any[]): Promise<T | null> {
     try {
+      await this.initialize();
       const db = this.ensureDb();
       if (params && params.length > 0) {
         return await db.getFirstAsync<T>(sql, params as any);
@@ -228,6 +130,7 @@ export class LocalDataSource implements ILocalDataSource {
 
   async execute(sql: string, params?: any[]): Promise<void> {
     try {
+      await this.initialize();
       const db = this.ensureDb();
       await db.execAsync(sql);
     } catch (error) {
@@ -237,6 +140,7 @@ export class LocalDataSource implements ILocalDataSource {
 
   async run(sql: string, params?: any[]): Promise<{ lastInsertRowId: number; changes: number }> {
     try {
+      await this.initialize();
       const db = this.ensureDb();
       if (params && params.length > 0) {
         return await db.runAsync(sql, params as any);
@@ -244,9 +148,7 @@ export class LocalDataSource implements ILocalDataSource {
       return await db.runAsync(sql);
     } catch (error) {
       if (this.isRecoverableError(error)) {
-        this.db = null;
         this.initialized = false;
-        dbInstance = null;
         await this.initialize();
         const db = this.ensureDb();
         if (params && params.length > 0) {
@@ -260,6 +162,7 @@ export class LocalDataSource implements ILocalDataSource {
 
   async transaction(operations: () => Promise<void>): Promise<void> {
     try {
+      await this.initialize();
       const db = this.ensureDb();
       await db.withTransactionAsync(async () => {
         await operations();
@@ -279,21 +182,15 @@ export class LocalDataSource implements ILocalDataSource {
     }
   }
 
-  // ==================== 队列写入支持 ====================
-
-  /**
-   * 使用队列执行写入操作（避免并发写入冲突）
-   */
   async enqueueWrite<T>(operation: () => Promise<T>): Promise<T> {
+    await this.initialize();
     const wrappedOperation = async () => {
       try {
         return await operation();
       } catch (error) {
-        // 尝试重连后重试
         if (this.isRecoverableError(error)) {
-          console.error('数据库写入异常，尝试重连后重试:', error);
-          this.db = null;
-          dbInstance = null;
+          console.error('[LocalDataSource] 数据库写入异常，尝试重连后重试:', error);
+          this.initialized = false;
           await this.initialize();
           return operation();
         }
@@ -306,18 +203,16 @@ export class LocalDataSource implements ILocalDataSource {
     return queued;
   }
 
-  /**
-   * 判断错误是否可恢复
-   */
   private isRecoverableError(error: unknown): boolean {
     const message = String(error);
     return (
       message.includes('NativeDatabase.prepareAsync') ||
       message.includes('NullPointerException') ||
-      message.includes('database is locked')
+      message.includes('database is locked') ||
+      message.includes('finalizing statement') ||
+      message.includes('cannot create file')
     );
   }
 }
 
-// 导出单例实例
 export const localDataSource = new LocalDataSource();