66 lines
1.4 KiB
JavaScript
66 lines
1.4 KiB
JavaScript
import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
|
|
import kafka from 'lib/kafka';
|
|
import prisma from 'lib/prisma';
|
|
import redis from 'lib/redis';
|
|
|
|
export async function createSession(...args) {
|
|
return runQuery({
|
|
[PRISMA]: () => relationalQuery(...args),
|
|
[CLICKHOUSE]: () => clickhouseQuery(...args),
|
|
});
|
|
}
|
|
|
|
async function relationalQuery(websiteId, data) {
|
|
return prisma.client.session
|
|
.create({
|
|
data: {
|
|
websiteId,
|
|
...data,
|
|
},
|
|
select: {
|
|
id: true,
|
|
sessionUuid: true,
|
|
hostname: true,
|
|
browser: true,
|
|
os: true,
|
|
screen: true,
|
|
language: true,
|
|
country: true,
|
|
device: true,
|
|
},
|
|
})
|
|
.then(async res => {
|
|
if (redis.enabled && res) {
|
|
await redis.set(`session:${res.sessionUuid}`, 1);
|
|
}
|
|
|
|
return res;
|
|
});
|
|
}
|
|
|
|
async function clickhouseQuery(
|
|
websiteId,
|
|
{ sessionUuid, hostname, browser, os, screen, language, country, device },
|
|
) {
|
|
const { getDateFormat, sendMessage } = kafka;
|
|
|
|
const params = {
|
|
session_uuid: sessionUuid,
|
|
website_id: websiteId,
|
|
created_at: getDateFormat(new Date()),
|
|
hostname,
|
|
browser,
|
|
os,
|
|
device,
|
|
screen,
|
|
language,
|
|
country: country ? country : null,
|
|
};
|
|
|
|
await sendMessage(params, 'event');
|
|
|
|
if (redis.enabled) {
|
|
await redis.set(`session:${sessionUuid}`, 1);
|
|
}
|
|
}
|