umami/queries/analytics/stats/getActiveVisitors.js

40 lines
1.0 KiB
JavaScript
Raw Normal View History

2022-07-12 23:14:36 +02:00
import { subMinutes } from 'date-fns';
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-28 06:38:35 +02:00
import { runQuery, CLICKHOUSE, PRISMA } from 'lib/db';
2022-07-12 23:14:36 +02:00
2022-07-21 06:31:26 +02:00
export async function getActiveVisitors(...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-07-21 06:31:26 +02:00
});
}
async function relationalQuery(websiteId) {
2022-07-12 23:14:36 +02:00
const date = subMinutes(new Date(), 5);
const params = [date];
2022-07-12 23:14:36 +02:00
2022-08-28 06:38:35 +02:00
return prisma.rawQuery(
`select count(distinct session_id) x
2022-07-12 23:14:36 +02:00
from pageview
2022-10-12 04:37:38 +02:00
join website
on pageview.website_id = website.website_id
where website.website_uuid = '${websiteId}'
and pageview.created_at >= $1`,
2022-07-12 23:14:36 +02:00
params,
);
}
2022-07-21 06:31:26 +02:00
async function clickhouseQuery(websiteId) {
2022-08-28 06:38:35 +02:00
const { rawQuery, getDateFormat } = clickhouse;
const params = [websiteId];
2022-07-21 06:31:26 +02:00
2022-08-28 06:38:35 +02:00
return rawQuery(
2022-10-09 01:12:33 +02:00
`select count(distinct session_id) x
2022-09-12 18:55:34 +02:00
from event
2022-07-21 06:31:26 +02:00
where website_id = $1
2022-08-28 06:38:35 +02:00
and created_at >= ${getDateFormat(subMinutes(new Date(), 5))}`,
2022-07-21 06:31:26 +02:00
params,
);
}