feat: add event names fetcher

Author: moonrailgun

src/server/model/insights/warehouse.ts

@@ -11,6 +11,7 @@ import dayjs from 'dayjs';
 import { processGroupedTimeSeriesData } from './utils.js';
 import { env } from '../../utils/env.js';
 import { FilterInfoType, FilterInfoValue } from '@tianji/shared';
+import { compact, get } from 'lodash-es';
 
 // MySQL date formats for different time units
 const MYSQL_DATE_FORMATS = {
@@ -66,21 +67,20 @@ export function getWarehouseApplications() {
   return applications;
 }
 
-export class WarehouseInsightsSqlBuilder extends InsightsSqlBuilder {
-  private connection: Pool | null = null;
-
-  private getConnection(): Pool {
-    if (!env.insights.warehouse.enable || !env.insights.warehouse.url) {
-      throw new Error('Warehouse is not enabled');
-    }
-
-    if (!this.connection) {
-      this.connection = createPool(env.insights.warehouse.url);
-    }
+let connection: Pool | null = null;
+export function getWarehouseConnection() {
+  if (!env.insights.warehouse.enable || !env.insights.warehouse.url) {
+    throw new Error('Warehouse is not enabled');
+  }
 
-    return this.connection;
+  if (!connection) {
+    connection = createPool(env.insights.warehouse.url);
   }
 
+  return connection;
+}
+
+export class WarehouseInsightsSqlBuilder extends InsightsSqlBuilder {
   getApplication(): WarehouseInsightsApplication {
     const name = this.query.insightId;
 
@@ -388,7 +388,7 @@ export class WarehouseInsightsSqlBuilder extends InsightsSqlBuilder {
   }
 
   async executeQuery(sql: Prisma.Sql): Promise<any[]> {
-    const connection = this.getConnection();
+    const connection = getWarehouseConnection();
 
     const [rows] = await connection.query(
       sql.sql.replaceAll('"', '`'), // avoid mysql and pg sql syntax error about double quote
@@ -412,3 +412,31 @@ export async function insightsWarehouse(
 
   return result;
 }
+
+export async function insightsWarehouseEvents(
+  applicationId: string
+): Promise<string[]> {
+  const connection = getWarehouseConnection();
+  const eventTable = getWarehouseApplications().find(
+    (app) => app.name === applicationId
+  )?.eventTable;
+
+  if (!eventTable) {
+    throw new Error(`Event table not found for application ${applicationId}`);
+  }
+
+  const [rows] = await connection.query(`
+SELECT DISTINCT event_name
+FROM (
+    SELECT \`${eventTable.eventNameField}\` as event_name, \`${eventTable.dateBasedCreatedAtField ?? eventTable.createdAtField}\` as date
+    FROM \`${eventTable.name}\`
+    ORDER BY \`${eventTable.dateBasedCreatedAtField ?? eventTable.createdAtField}\` DESC
+) t
+LIMIT 100;`);
+
+  if (!Array.isArray(rows)) {
+    throw new Error(`Invalid rows from warehouse`);
+  }
+
+  return compact(rows.map((row) => get(row, 'event_name', '')));
+}

src/server/trpc/routers/insights.ts

@@ -11,7 +11,10 @@ import { stringifyDateType } from '../../utils/common.js';
 import { queryEvents, queryInsight } from '../../model/insights/index.js';
 import { insightsSurveyBuiltinFields } from '../../model/insights/utils.js';
 import { uniq } from 'lodash-es';
-import { getWarehouseApplications } from '../../model/insights/warehouse.js';
+import {
+  getWarehouseApplications,
+  insightsWarehouseEvents,
+} from '../../model/insights/warehouse.js';
 
 export const insightsRouter = router({
   query: workspaceProcedure
@@ -67,6 +70,14 @@ export const insightsRouter = router({
         return [];
       }
 
+      if (insightType === 'warehouse') {
+        const events = await insightsWarehouseEvents(insightId);
+        return events.map((item) => ({
+          name: item,
+          count: 0,
+        }));
+      }
+
       return [];
     }),
   filterParams: workspaceProcedure