Add session_data / session redis to CH.

This commit is contained in:
Brian Cao 2023-05-31 11:42:39 -07:00
parent 8937602f57
commit 2e0d1697e7
21 changed files with 328 additions and 282 deletions

View file

@ -1,43 +1,8 @@
import clickhouse from 'lib/clickhouse';
import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
import prisma from 'lib/prisma';
import { Prisma } from '@prisma/client';
import prisma from 'lib/prisma';
export async function getSession(args: { id: string }) {
return runQuery({
[PRISMA]: () => relationalQuery(args),
[CLICKHOUSE]: () => clickhouseQuery(args),
});
}
async function relationalQuery(where: Prisma.SessionWhereUniqueInput) {
export async function getSession(where: Prisma.SessionWhereUniqueInput) {
return prisma.client.session.findUnique({
where,
});
}
async function clickhouseQuery({ id: sessionId }: { id: string }) {
const { rawQuery, findFirst } = clickhouse;
const params = { sessionId };
return rawQuery(
`select
session_id,
website_id,
created_at,
hostname,
browser,
os,
device,
screen,
language,
country,
subdivision1,
subdivision2,
city
from website_event
where session_id = {sessionId:UUID}
limit 1`,
params,
).then(result => findFirst(result));
}