fix: update connection issue for new connection logic which will parse driver

Author: moonrailgun

src/client/routes/insights/warehouse/connections/index.tsx

@@ -32,7 +32,7 @@ type ConnectionItem = {
   id: string;
   name: string;
   description: string;
-  dbDriver: string;
+  dbDriver: 'mysql' | 'postgresql';
   createdAt: string;
   updatedAt: string;
 };
@@ -41,7 +41,7 @@ interface ConnectionFormValues {
   id?: string;
   name: string;
   description: string;
-  dbDriver: string;
+  dbDriver: 'mysql' | 'postgresql';
 }
 
 function PageComponent() {

src/server/model/insights/warehouse/ai.ts

@@ -139,6 +139,10 @@ export const createWarehouseAITools = (
 
               const connection = getWarehouseConnection(connectionUrl);
 
+              if (connection.driver === 'postgresql') {
+                throw new Error('PostgreSQL connection is not supported yet');
+              }
+
               // Auto-add LIMIT if not present
               let finalSql = sql.trim();
               const trimmedSqlLower = finalSql.toLowerCase();
@@ -155,7 +159,8 @@ export const createWarehouseAITools = (
               // const [res] = env.isDev
               //   ? [chatByCountryDemo] // TODO: remove this after testing
               //   : await connection.query(finalSql);
-              const [res] = await connection.query(finalSql);
+
+              const [res] = await connection.pool.query(finalSql);
 
               logger.info('[queryWarehouse] result:' + JSON.stringify(res));
 

src/server/model/insights/warehouse/longTable.ts

@@ -397,9 +397,17 @@ export class WarehouseLongTableInsightsSqlBuilder extends InsightsSqlBuilder {
 
   async executeQuery(sql: Prisma.Sql): Promise<any[]> {
     const application = this.getApplication();
+    if (!application.databaseUrl) {
+      throw new Error('Database url is not set');
+    }
+
     const connection = getWarehouseConnection(application.databaseUrl);
 
-    const [rows] = await connection.query(
+    if (connection.driver === 'postgresql') {
+      throw new Error('PostgreSQL connection is not supported yet');
+    }
+
+    const [rows] = await connection.pool.query(
       sql.sql.replaceAll('"', '`'), // avoid mysql and pg sql syntax error about double quote
       sql.values
     );
@@ -483,10 +491,18 @@ export async function insightsLongTableWarehouseEvents(
     throw new Error(`Event table not found for application ${applicationId}`);
   }
 
+  if (!application.databaseUrl) {
+    throw new Error('Database url is not set');
+  }
+
   const connection = getWarehouseConnection(application.databaseUrl);
   const eventTable = application.eventTable;
 
-  const [rows] = await connection.query(`
+  if (connection.driver === 'postgresql') {
+    throw new Error('PostgreSQL connection is not supported yet');
+  }
+
+  const [rows] = await connection.pool.query(`
 SELECT DISTINCT event_name
 FROM (
     SELECT \`${eventTable.eventNameField}\` as event_name, \`${eventTable.dateBasedCreatedAtField ?? eventTable.createdAtField}\` as date
@@ -515,10 +531,18 @@ export async function insightsLongTableWarehouseFilterParams(
     throw new Error(`Event table not found for application ${applicationId}`);
   }
 
+  if (!application.databaseUrl) {
+    throw new Error('Database url is not set');
+  }
+
   const connection = getWarehouseConnection(application.databaseUrl);
   const eventParametersTable = application.eventParametersTable;
 
-  const [rows] = await connection.query(`
+  if (connection.driver === 'postgresql') {
+    throw new Error('PostgreSQL connection is not supported yet');
+  }
+
+  const [rows] = await connection.pool.query(`
 SELECT DISTINCT param_value
 FROM (
     SELECT \`${eventParametersTable.paramsNameField}\` as param_value, \`${eventParametersTable.dateBasedCreatedAtField ?? eventParametersTable.createdAtField}\` as date

src/server/model/insights/warehouse/wideTable.ts

@@ -273,9 +273,16 @@ export class WarehouseWideTableInsightsSqlBuilder extends InsightsSqlBuilder {
 
   async executeQuery(sql: Prisma.Sql): Promise<any[]> {
     const application = this.getApplication();
+    if (!application.databaseUrl) {
+      throw new Error('Database url is not set');
+    }
     const connection = getWarehouseConnection(application.databaseUrl);
 
-    const [rows] = await connection.query(
+    if (connection.driver === 'postgresql') {
+      throw new Error('PostgreSQL connection is not supported yet');
+    }
+
+    const [rows] = await connection.pool.query(
       sql.sql.replaceAll('"', '`'), // avoid mysql and pg sql syntax error about double quote
       sql.values
     );