umami/queries/analytics/session/createSession.ts

import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
import kafka from 'lib/kafka';
import prisma from 'lib/prisma';
import cache from 'lib/cache';
import { Prisma } from '@prisma/client';

export async function createSession(args: Prisma.SessionCreateInput) {
  return runQuery({
    [PRISMA]: () => relationalQuery(args),
    [CLICKHOUSE]: () => clickhouseQuery(args),
  }).then(async data => {
    if (cache.enabled) {
      await cache.storeSession(data);
    }

    return data;
  });
}

async function relationalQuery(data: Prisma.SessionCreateInput) {
  const {
    id,
    websiteId,
    hostname,
    browser,
    os,
    device,
    screen,
    language,
    country,
    subdivision1,
    subdivision2,
    city,
  } = data;

  return prisma.client.session.create({
    data: {
      id,
      websiteId,
      hostname,
      browser,
      os,
      device,
      screen,
      language,
      country,
      subdivision1: country && subdivision1 ? `${country}-${subdivision1}` : null,
      subdivision2,
      city,
    },
  });
}

async function clickhouseQuery(data: {
  id: string;
  websiteId: string;
  hostname?: string;
  browser?: string;
  os?: string;
  device?: string;
  screen?: string;
  language?: string;
  country?: string;
  subdivision1?: string;
  subdivision2?: string;
  city?: string;
}) {
  const {
    id,
    websiteId,
    hostname,
    browser,
    os,
    device,
    screen,
    language,
    country,
    subdivision1,
    subdivision2,
    city,
  } = data;
  const { getDateFormat, sendMessage } = kafka;

  const msg = {
    session_id: id,
    website_id: websiteId,
    hostname,
    browser,
    os,
    device,
    screen,
    language,
    country,
    subdivision1,
    subdivision2,
    city,
    created_at: getDateFormat(new Date()),
  };

  await sendMessage(msg, 'event');

  return data;
}