add event-data support and charts for arrays, booleans, dates

2026-04-24 15:04:59 -07:00
1 parent 43fb351ac7
commit 5a1ef5e674
@@ -0,0 +1,110 @@
+import clickhouse from '@/lib/clickhouse';
+import { DATA_TYPE } from '@/lib/constants';
+import { CLICKHOUSE, PRISMA, runQuery } from '@/lib/db';
+import prisma from '@/lib/prisma';
+import type { EventDataSeriesPoint, EventPropertyFilter, QueryFilters } from '@/lib/types';
+
+const FUNCTION_NAME = 'getEventDataArraySeries';
+
+export async function getEventDataArraySeries(
+  ...args: [
+    websiteId: string,
+    eventName: string,
+    propertyName: string,
+    filters: QueryFilters,
+    eventFilters?: EventPropertyFilter[],
+  ]
+): Promise<EventDataSeriesPoint[]> {
+  return runQuery({
+    [PRISMA]: () => relationalQuery(...args),
+    [CLICKHOUSE]: () => clickhouseQuery(...args),
+  });
+}
+
+async function relationalQuery(
+  websiteId: string,
+  eventName: string,
+  propertyName: string,
+  filters: QueryFilters,
+  eventFilters: EventPropertyFilter[] = [],
+): Promise<EventDataSeriesPoint[]> {
+  const { timezone = 'utc', unit = 'day' } = filters;
+  const { rawQuery, getDateSQL, parseFilters, getEventPropertyFilterQuery } = prisma;
+  const { filterQuery, cohortQuery, joinSessionQuery, queryParams } = parseFilters({
+    ...filters,
+    websiteId,
+  });
+  const { sql: epfSQL, params: epfParams } = getEventPropertyFilterQuery(eventFilters);
+
+  return rawQuery(
+    `
+    select
+      array_item.value as x,
+      ${getDateSQL('event_data.created_at', unit, timezone)} as t,
+      count(*) as y
+    from event_data
+    join website_event on website_event.event_id = event_data.website_event_id
+      and website_event.website_id = {{websiteId::uuid}}
+      and website_event.created_at between {{startDate}} and {{endDate}}
+      and website_event.event_type = 2
+      and website_event.event_name = {{eventName}}
+    cross join lateral jsonb_array_elements_text(coalesce(event_data.string_value, '[]')::jsonb) as array_item(value)
+    ${cohortQuery}
+    ${joinSessionQuery}
+    where event_data.website_id = {{websiteId::uuid}}
+      and event_data.created_at between {{startDate}} and {{endDate}}
+      and event_data.data_key = {{propertyName}}
+      and event_data.data_type = ${DATA_TYPE.array}
+      ${filterQuery}
+      ${epfSQL}
+    group by 1, 2
+    order by 2
+    `,
+    { ...queryParams, eventName, propertyName, ...epfParams },
+    FUNCTION_NAME,
+  );
+}
+
+async function clickhouseQuery(
+  websiteId: string,
+  eventName: string,
+  propertyName: string,
+  filters: QueryFilters,
+  eventFilters: EventPropertyFilter[] = [],
+): Promise<EventDataSeriesPoint[]> {
+  const { timezone = 'UTC', unit = 'day' } = filters;
+  const { rawQuery, getDateSQL, parseFilters, getEventPropertyFilterQuery } = clickhouse;
+  const { filterQuery, cohortQuery, queryParams } = parseFilters({ ...filters, websiteId });
+  const { sql: epfSQL, params: epfParams } = getEventPropertyFilterQuery(eventFilters);
+
+  return rawQuery(
+    `
+    select
+      arrayJoin(JSONExtract(ifNull(event_data.string_value, '[]'), 'Array(String)')) as x,
+      ${getDateSQL('event_data.created_at', unit, timezone)} as t,
+      count() as y
+    from event_data
+    any left join (
+          select *
+          from website_event
+          where website_id = {websiteId:UUID}
+            and created_at between {startDate:DateTime64} and {endDate:DateTime64}
+            and event_type = 2
+            and event_name = {eventName:String}) website_event
+    on website_event.event_id = event_data.event_id
+      and website_event.session_id = event_data.session_id
+      and website_event.website_id = event_data.website_id
+    ${cohortQuery}
+    where event_data.website_id = {websiteId:UUID}
+      and event_data.created_at between {startDate:DateTime64} and {endDate:DateTime64}
+      and event_data.data_key = {propertyName:String}
+      and event_data.data_type = ${DATA_TYPE.array}
+    ${filterQuery}
+    ${epfSQL}
+    group by x, t
+    order by t
+    `,
+    { ...queryParams, eventName, propertyName, ...epfParams },
+    FUNCTION_NAME,
+  );
+}