修正分頁、Too many connection 問題

2025-09-19 16:27:54 +08:00
parent bac364a667
commit 95c0c4cb23
20 changed files with 912 additions and 291 deletions
--- a/lib/database-service-base.ts
+++ b/lib/database-service-base.ts
@@ -0,0 +1,308 @@
+// =====================================================
+// 資料庫服務基類 - 確保連線正確關閉
+// =====================================================
+
+import { db } from './database';
+import { PoolConnection } from 'mysql2/promise';
+
+export abstract class DatabaseServiceBase {
+  // 靜態安全查詢方法 - 確保連線關閉
+  static async safeQuery<T = any>(sql: string, params?: any[]): Promise<T[]> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      // 獲取連線
+      connection = await db.getConnection();
+      
+      // 執行查詢
+      const [rows] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return rows as T[];
+    } catch (error) {
+      // 確保連線被釋放
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 實例安全查詢方法 - 確保連線關閉
+  protected async safeQuery<T = any>(sql: string, params?: any[]): Promise<T[]> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      // 獲取連線
+      connection = await db.getConnection();
+      
+      // 執行查詢
+      const [rows] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return rows as T[];
+    } catch (error) {
+      // 確保連線被釋放
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 靜態安全單一查詢方法
+  static async safeQueryOne<T = any>(sql: string, params?: any[]): Promise<T | null> {
+    const results = await DatabaseServiceBase.safeQuery<T>(sql, params);
+    return results.length > 0 ? results[0] : null;
+  }
+
+  // 實例安全單一查詢方法
+  protected async safeQueryOne<T = any>(sql: string, params?: any[]): Promise<T | null> {
+    const results = await this.safeQuery<T>(sql, params);
+    return results.length > 0 ? results[0] : null;
+  }
+
+  // 靜態安全插入方法
+  static async safeInsert(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 實例安全插入方法
+  protected async safeInsert(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 靜態安全更新方法
+  static async safeUpdate(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 實例安全更新方法
+  protected async safeUpdate(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 靜態安全刪除方法
+  static async safeDelete(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 實例安全刪除方法
+  protected async safeDelete(sql: string, params?: any[]): Promise<any> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const [result] = await connection.execute(sql, params);
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 事務處理方法
+  protected async withTransaction<T>(
+    callback: (connection: PoolConnection) => Promise<T>
+  ): Promise<T> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      await connection.beginTransaction();
+      
+      const result = await callback(connection);
+      
+      await connection.commit();
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return result;
+    } catch (error) {
+      if (connection) {
+        try {
+          await connection.rollback();
+        } catch (rollbackError) {
+          console.error('回滾事務時發生錯誤:', rollbackError);
+        }
+        
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 批次查詢方法 - 減少連線使用
+  protected async batchQuery<T = any>(
+    queries: Array<{ sql: string; params?: any[] }>
+  ): Promise<T[][]> {
+    let connection: PoolConnection | null = null;
+    
+    try {
+      connection = await db.getConnection();
+      const results: T[][] = [];
+      
+      for (const query of queries) {
+        const [rows] = await connection.execute(query.sql, query.params);
+        results.push(rows as T[]);
+      }
+      
+      // 立即釋放連線
+      connection.release();
+      connection = null;
+      
+      return results;
+    } catch (error) {
+      if (connection) {
+        try {
+          connection.release();
+        } catch (releaseError) {
+          console.error('釋放連線時發生錯誤:', releaseError);
+        }
+      }
+      throw error;
+    }
+  }
+
+  // 連線健康檢查
+  protected async checkConnectionHealth(): Promise<boolean> {
+    try {
+      await this.safeQueryOne('SELECT 1 as health_check');
+      return true;
+    } catch (error) {
+      console.error('資料庫連線健康檢查失敗:', error);
+      return false;
+    }
+  }
+}