110 lines
4.0 KiB
TypeScript
110 lines
4.0 KiB
TypeScript
import clickhouse from '@/lib/clickhouse';
|
|
import { EVENT_TYPE } from '@/lib/constants';
|
|
import { CLICKHOUSE, PRISMA, runQuery } from '@/lib/db';
|
|
import prisma from '@/lib/prisma';
|
|
import type { EventDataNumericStats, PropertyFilter, QueryFilters } from '@/lib/types';
|
|
|
|
const FUNCTION_NAME = 'getSessionDataNumericStats';
|
|
|
|
export async function getSessionDataNumericStats(
|
|
...args: [
|
|
websiteId: string,
|
|
propertyName: string,
|
|
filters: QueryFilters,
|
|
propertyFilters?: PropertyFilter[],
|
|
]
|
|
): Promise<EventDataNumericStats> {
|
|
return runQuery({
|
|
[PRISMA]: () => relationalQuery(...args),
|
|
[CLICKHOUSE]: () => clickhouseQuery(...args),
|
|
}).then(results => results?.[0]);
|
|
}
|
|
|
|
async function relationalQuery(
|
|
websiteId: string,
|
|
propertyName: string,
|
|
filters: QueryFilters,
|
|
propertyFilters: PropertyFilter[] = [],
|
|
) {
|
|
const { timezone = 'utc' } = filters;
|
|
const { rawQuery, parseFilters, getPropertyFilterQuery } = prisma;
|
|
const { filterQuery, cohortQuery, joinSessionQuery, queryParams } = parseFilters({
|
|
...filters,
|
|
websiteId,
|
|
timezone,
|
|
});
|
|
const { sql: pfSQL, params: pfParams } = getPropertyFilterQuery(propertyFilters, 'session', timezone);
|
|
|
|
return rawQuery(
|
|
`
|
|
with filtered_sessions as (
|
|
select distinct website_event.session_id, website_event.website_id
|
|
from website_event
|
|
${cohortQuery}
|
|
${joinSessionQuery}
|
|
where website_event.website_id = {{websiteId::uuid}}
|
|
and website_event.created_at between {{startDate}} and {{endDate}}
|
|
and website_event.event_type != ${EVENT_TYPE.performance}
|
|
${filterQuery}
|
|
${pfSQL}
|
|
)
|
|
select
|
|
coalesce(sum(cast(session_data.number_value as decimal)), 0) as "total",
|
|
coalesce(avg(cast(session_data.number_value as decimal)), 0) as "average",
|
|
coalesce(percentile_cont(0.5) within group (order by session_data.number_value), 0) as "median",
|
|
coalesce(max(session_data.number_value), 0) as "max",
|
|
coalesce(min(session_data.number_value), 0) as "min"
|
|
from session_data
|
|
join filtered_sessions
|
|
on filtered_sessions.session_id = session_data.session_id
|
|
and filtered_sessions.website_id = session_data.website_id
|
|
where session_data.website_id = {{websiteId::uuid}}
|
|
and session_data.data_key = {{propertyName}}
|
|
and session_data.data_type = 2
|
|
`,
|
|
{ ...queryParams, propertyName, ...pfParams },
|
|
FUNCTION_NAME,
|
|
);
|
|
}
|
|
|
|
async function clickhouseQuery(
|
|
websiteId: string,
|
|
propertyName: string,
|
|
filters: QueryFilters,
|
|
propertyFilters: PropertyFilter[] = [],
|
|
): Promise<EventDataNumericStats[]> {
|
|
const { timezone = 'UTC' } = filters;
|
|
const { rawQuery, parseFilters, getPropertyFilterQuery } = clickhouse;
|
|
const { filterQuery, cohortQuery, queryParams } = parseFilters({ ...filters, websiteId, timezone });
|
|
const { sql: pfSQL, params: pfParams } = getPropertyFilterQuery(propertyFilters, 'session', timezone);
|
|
|
|
return rawQuery(
|
|
`
|
|
with filtered_sessions as (
|
|
select distinct website_event.session_id
|
|
from website_event
|
|
${cohortQuery}
|
|
where website_event.website_id = {websiteId:UUID}
|
|
and website_event.created_at between {startDate:DateTime64} and {endDate:DateTime64}
|
|
and website_event.event_type != ${EVENT_TYPE.performance}
|
|
${filterQuery}
|
|
${pfSQL}
|
|
)
|
|
select
|
|
if(count() = 0, 0, sum(session_data.number_value)) as total,
|
|
if(count() = 0, 0, avg(session_data.number_value)) as average,
|
|
if(count() = 0, 0, median(session_data.number_value)) as median,
|
|
if(count() = 0, 0, max(session_data.number_value)) as max,
|
|
if(count() = 0, 0, min(session_data.number_value)) as min
|
|
from session_data final
|
|
join filtered_sessions
|
|
on filtered_sessions.session_id = session_data.session_id
|
|
where session_data.website_id = {websiteId:UUID}
|
|
and session_data.data_key = {propertyName:String}
|
|
and session_data.data_type = 2
|
|
`,
|
|
{ ...queryParams, propertyName, ...pfParams },
|
|
FUNCTION_NAME,
|
|
);
|
|
}
|