watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-10 23:57:12 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Support clickhouse for save session data.

Browse source
This commit is contained in:
Mike Cao 2024-07-05 02:04:51 -07:00
parent 9bf34bd5e4
commit 26462cffbc
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/queries/analytics/sessions/saveSessionData.ts
View file

@ -5,6 +5,7 @@ import prisma from 'lib/prisma';
import { DynamicData } from 'lib/types'; import { DynamicData } from 'lib/types';
import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db'; import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
import kafka from 'lib/kafka'; import kafka from 'lib/kafka';
import clickhouse from 'lib/clickhouse';
export async function saveSessionData(data: { export async function saveSessionData(data: {
websiteId: string; websiteId: string;
@ -81,6 +82,7 @@ async function clickhouseQuery(data: {
}) { }) {
const { websiteId, sessionId, sessionData, createdAt } = data; const { websiteId, sessionId, sessionData, createdAt } = data;
const { insert } = clickhouse;
const { getDateFormat, sendMessages } = kafka; const { getDateFormat, sendMessages } = kafka;
const jsonKeys = flattenJSON(sessionData); const jsonKeys = flattenJSON(sessionData);
@ -98,7 +100,11 @@ async function clickhouseQuery(data: {
}; };
}); });
await sendMessages(messages, 'session_data'); if (kafka.enabled) {
await sendMessages(messages, 'session_data');
} else {
await insert('event_data', messages);
}
return data; return data;
} }