117 lines
3.9 KiB
TypeScript
117 lines
3.9 KiB
TypeScript
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, getPropertyFilterQuery } = prisma;
|
|
const { filterQuery, cohortQuery, joinSessionQuery, queryParams } = parseFilters({
|
|
...filters,
|
|
websiteId,
|
|
timezone,
|
|
});
|
|
const { sql: pfSQL, params: pfParams } = getPropertyFilterQuery(eventFilters, 'event', timezone);
|
|
|
|
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}
|
|
${pfSQL}
|
|
group by 1, 2
|
|
order by 2
|
|
`,
|
|
{ ...queryParams, eventName, propertyName, ...pfParams },
|
|
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, getPropertyFilterQuery } = clickhouse;
|
|
const { filterQuery, cohortQuery, queryParams } = parseFilters({
|
|
...filters,
|
|
websiteId,
|
|
timezone,
|
|
});
|
|
const { sql: pfSQL, params: pfParams } = getPropertyFilterQuery(eventFilters, 'event', timezone);
|
|
|
|
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.event_name = {eventName:String}
|
|
and event_data.data_key = {propertyName:String}
|
|
and event_data.data_type = ${DATA_TYPE.array}
|
|
${filterQuery}
|
|
${pfSQL}
|
|
group by x, t
|
|
order by t
|
|
`,
|
|
{ ...queryParams, eventName, propertyName, ...pfParams },
|
|
FUNCTION_NAME,
|
|
);
|
|
}
|