umami/queries/analytics/pageview/savePageView.js

58 lines
1.4 KiB
JavaScript
Raw Normal View History

2022-08-05 02:43:47 +02:00
import { CLICKHOUSE, RELATIONAL, KAFKA, URL_LENGTH } from 'lib/constants';
2022-07-22 23:43:19 +02:00
import {
getDateFormatClickhouse,
prisma,
rawQueryClickhouse,
runAnalyticsQuery,
runQuery,
2022-08-05 02:43:47 +02:00
kafkaProducer,
2022-07-22 23:43:19 +02:00
} from 'lib/db';
2022-07-16 01:47:38 +02:00
2022-07-22 23:43:19 +02:00
export async function savePageView(...args) {
return runAnalyticsQuery({
2022-07-25 18:47:11 +02:00
[RELATIONAL]: () => relationalQuery(...args),
[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-07-12 23:14:36 +02:00
return runQuery(
prisma.pageview.create({
data: {
website_id,
session_id,
url: url?.substr(0, URL_LENGTH),
referrer: referrer?.substr(0, URL_LENGTH),
},
}),
);
}
2022-07-22 23:43:19 +02:00
async function clickhouseQuery(website_id, { session_uuid, url, referrer }) {
const params = [
website_id,
session_uuid,
url?.substr(0, URL_LENGTH),
referrer?.substr(0, URL_LENGTH),
];
return rawQueryClickhouse(
`
insert into umami_dev.pageview (created_at, website_id, session_uuid, url, referrer)
values (${getDateFormatClickhouse(new Date())}, $1, $2, $3, $4);`,
params,
);
}
2022-08-05 02:43:47 +02:00
async function kafkaQuery(website_id, { session_uuid, url, referrer }) {
const params = {
website_id: website_id,
session_uuid: session_uuid,
url: url?.substr(0, URL_LENGTH),
referrer: referrer?.substr(0, URL_LENGTH),
};
await kafkaProducer(params, 'pageview');
}