umami/queries/analytics/session/getSessionByUuid.js
2022-09-12 09:55:34 -07:00

58 lines
1.2 KiB
JavaScript

import clickhouse from 'lib/clickhouse';
import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
import prisma from 'lib/prisma';
import redis from 'lib/redis';
export async function getSessionByUuid(...args) {
return runQuery({
[PRISMA]: () => relationalQuery(...args),
[CLICKHOUSE]: () => clickhouseQuery(...args),
});
}
async function relationalQuery(session_uuid) {
return prisma.client.session
.findUnique({
where: {
session_uuid,
},
})
.then(async res => {
if (redis.client && res) {
await redis.client.set(`session:${res.session_uuid}`, res.session_id);
}
return res;
});
}
async function clickhouseQuery(session_uuid) {
const { rawQuery, findFirst } = clickhouse;
const params = [session_uuid];
return rawQuery(
`select distinct
session_uuid,
website_id,
created_at,
hostname,
browser,
os,
device,
screen,
language,
country
from event
where session_uuid = $1`,
params,
)
.then(result => findFirst(result))
.then(async res => {
if (redis.client && res) {
await redis.client.set(`session:${res.session_uuid}`, 1);
}
return res;
});
}