umami/src/queries/analytics/sessions/getSessionData.ts

import prisma from 'lib/prisma';
import clickhouse from 'lib/clickhouse';
import { runQuery, PRISMA, CLICKHOUSE } from 'lib/db';

export async function getSessionData(...args: [websiteId: string, sessionId: string]) {
  return runQuery({
    [PRISMA]: () => relationalQuery(...args),
    [CLICKHOUSE]: () => clickhouseQuery(...args),
  });
}

async function relationalQuery(websiteId: string, sessionId: string) {
  return prisma.client.sessionData.findMany({
    where: {
      id: sessionId,
      websiteId,
    },
  });
}

async function clickhouseQuery(websiteId: string, sessionId: string) {
  const { rawQuery } = clickhouse;

  return rawQuery(
    `
    select
        website_id as websiteId,
        session_id as sessionId,
        data_key as dataKey,
        data_type as dataType,
        string_value as stringValue,
        number_value as numberValue,
        date_value as dateValue,
        created_at as createdAt
    from session_data final
    where website_id = {websiteId:UUID}
    and session_id = {sessionId:UUID}
    order by data_key asc
    `,
    { websiteId, sessionId },
  );
}