2022-08-28 06:38:35 +02:00
|
|
|
import prisma from 'lib/prisma';
|
2022-08-26 07:04:32 +02:00
|
|
|
import clickhouse from 'lib/clickhouse';
|
2022-08-26 07:23:19 +02:00
|
|
|
import kafka from 'lib/kafka';
|
2022-08-28 06:38:35 +02:00
|
|
|
import { runQuery, CLICKHOUSE, KAFKA, PRISMA } from 'lib/db';
|
|
|
|
import { URL_LENGTH } from 'lib/constants';
|
2022-07-16 01:47:38 +02:00
|
|
|
|
2022-07-22 23:43:19 +02:00
|
|
|
export async function savePageView(...args) {
|
2022-08-28 06:38:35 +02:00
|
|
|
return runQuery({
|
|
|
|
[PRISMA]: () => relationalQuery(...args),
|
2022-07-25 18:47:11 +02:00
|
|
|
[CLICKHOUSE]: () => clickhouseQuery(...args),
|
2022-08-05 02:43:47 +02:00
|
|
|
[KAFKA]: () => kafkaQuery(...args),
|
2022-07-22 23:43:19 +02:00
|
|
|
});
|
|
|
|
}
|
2022-07-12 23:14:36 +02:00
|
|
|
|
2022-07-22 23:43:19 +02:00
|
|
|
async function relationalQuery(website_id, { session_id, url, referrer }) {
|
2022-08-28 06:38:35 +02:00
|
|
|
return prisma.client.pageview.create({
|
|
|
|
data: {
|
|
|
|
website_id,
|
|
|
|
session_id,
|
|
|
|
url: url?.substring(0, URL_LENGTH),
|
|
|
|
referrer: referrer?.substring(0, URL_LENGTH),
|
|
|
|
},
|
|
|
|
});
|
2022-07-12 23:14:36 +02:00
|
|
|
}
|
2022-07-22 23:43:19 +02:00
|
|
|
|
|
|
|
async function clickhouseQuery(website_id, { session_uuid, url, referrer }) {
|
|
|
|
const params = [
|
|
|
|
website_id,
|
|
|
|
session_uuid,
|
2022-08-28 06:38:35 +02:00
|
|
|
url?.substring(0, URL_LENGTH),
|
|
|
|
referrer?.substring(0, URL_LENGTH),
|
2022-07-22 23:43:19 +02:00
|
|
|
];
|
|
|
|
|
2022-08-26 07:04:32 +02:00
|
|
|
return clickhouse.rawQuery(
|
2022-08-28 06:38:35 +02:00
|
|
|
`insert into umami.pageview (created_at, website_id, session_uuid, url, referrer)
|
2022-08-26 07:04:32 +02:00
|
|
|
values (${clickhouse.getDateFormat(new Date())}, $1, $2, $3, $4);`,
|
2022-07-22 23:43:19 +02:00
|
|
|
params,
|
|
|
|
);
|
|
|
|
}
|
2022-08-05 02:43:47 +02:00
|
|
|
|
|
|
|
async function kafkaQuery(website_id, { session_uuid, url, referrer }) {
|
2022-08-28 06:38:35 +02:00
|
|
|
const { getDateFormat, sendMessage } = kafka;
|
2022-08-05 02:43:47 +02:00
|
|
|
const params = {
|
|
|
|
website_id: website_id,
|
|
|
|
session_uuid: session_uuid,
|
2022-08-28 06:38:35 +02:00
|
|
|
created_at: getDateFormat(new Date()),
|
|
|
|
url: url?.substring(0, URL_LENGTH),
|
|
|
|
referrer: referrer?.substring(0, URL_LENGTH),
|
2022-08-05 02:43:47 +02:00
|
|
|
};
|
|
|
|
|
2022-08-28 06:38:35 +02:00
|
|
|
await sendMessage(params, 'pageview');
|
2022-08-05 02:43:47 +02:00
|
|
|
}
|