watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-18 19:45:35 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: support batch event

Browse source
This commit is contained in:
harry 2024-11-26 17:00:32 +07:00
parent 42824be546
commit 1a908b86fe
3 changed files with 156 additions and 116 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/lib/types.ts
View file

@ -52,12 +52,6 @@ export interface DynamicData {
[key: string]: number | string | number[] | string[]; [key: string]: number | string | number[] | string[];
} }
export interface JsonKeyDynamicData {
key: string;
value: any;
dataType: DynamicDataType;
}
export interface Auth { export interface Auth {
user?: { user?: {
id: string; id: string;

57
src/queries/analytics/events/saveEvent.ts
View file

@ -67,10 +67,44 @@ async function relationalQuery(data: {
eventBatchData, eventBatchData,
tag, tag,
} = data; } = data;
const websiteEventId = uuid();
const websiteEvent = prisma.client.websiteEvent.create({ const websiteEventData = [];
data: { const eventsData = [];
if (eventBatchData) {
for (const eventData of eventBatchData) {
const websiteEventId = uuid();
websiteEventData.push({
id: websiteEventId,
websiteId,
sessionId,
visitId,
urlPath: urlPath?.substring(0, URL_LENGTH),
urlQuery: urlQuery?.substring(0, URL_LENGTH),
referrerPath: referrerPath?.substring(0, URL_LENGTH),
referrerQuery: referrerQuery?.substring(0, URL_LENGTH),
referrerDomain: referrerDomain?.substring(0, URL_LENGTH),
pageTitle: pageTitle?.substring(0, PAGE_TITLE_LENGTH),
eventType: eventName ? EVENT_TYPE.customEvent : EVENT_TYPE.pageView,
eventName: eventName ? eventName?.substring(0, EVENT_NAME_LENGTH) : null,
tag,
});
eventsData.push({
websiteId,
sessionId,
visitId,
eventId: websiteEventId,
urlPath: urlPath?.substring(0, URL_LENGTH),
eventName: eventName?.substring(0, EVENT_NAME_LENGTH),
eventData,
});
}
} else {
const websiteEventId = uuid();
websiteEventData.push({
id: websiteEventId, id: websiteEventId,
websiteId, websiteId,
sessionId, sessionId,
@ -84,11 +118,9 @@ async function relationalQuery(data: {
eventType: eventName ? EVENT_TYPE.customEvent : EVENT_TYPE.pageView, eventType: eventName ? EVENT_TYPE.customEvent : EVENT_TYPE.pageView,
eventName: eventName ? eventName?.substring(0, EVENT_NAME_LENGTH) : null, eventName: eventName ? eventName?.substring(0, EVENT_NAME_LENGTH) : null,
tag, tag,
}, });
});
if (eventData || eventBatchData) { eventsData.push({
await saveEventData({
websiteId, websiteId,
sessionId, sessionId,
visitId, visitId,
@ -96,11 +128,18 @@ async function relationalQuery(data: {
urlPath: urlPath?.substring(0, URL_LENGTH), urlPath: urlPath?.substring(0, URL_LENGTH),
eventName: eventName?.substring(0, EVENT_NAME_LENGTH), eventName: eventName?.substring(0, EVENT_NAME_LENGTH),
eventData, eventData,
eventBatchData,
}); });
} }
return websiteEvent; const websiteEvents = prisma.client.websiteEvent.createMany({
data: websiteEventData,
});
if (eventData || eventBatchData) {
await saveEventData(eventsData);
}
return websiteEvents;
} }
async function clickhouseQuery(data: { async function clickhouseQuery(data: {

209
src/queries/analytics/events/saveEventData.ts
View file

@ -6,135 +6,142 @@ import { flattenDynamicData, flattenJSON, getStringValue } from 'lib/data';
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 prisma from 'lib/prisma'; import prisma from 'lib/prisma';
import { DynamicData, JsonKeyDynamicData } from 'lib/types'; import { DynamicData, DynamicDataType } from 'lib/types';
export async function saveEventData(data: { export async function saveEventData(
websiteId: string; data: Array<{
eventId: string; websiteId: string;
sessionId?: string; eventId: string;
visitId?: string; sessionId?: string;
urlPath?: string; visitId?: string;
eventName?: string; urlPath?: string;
eventData?: DynamicData; eventName?: string;
eventBatchData?: Array<DynamicData>; eventData?: DynamicData;
createdAt?: string; createdAt?: string;
}) { }>,
) {
return runQuery({ return runQuery({
[PRISMA]: () => relationalQuery(data), [PRISMA]: () => relationalQuery(data),
[CLICKHOUSE]: () => clickhouseQuery(data), [CLICKHOUSE]: () => clickhouseQuery(data),
}); });
} }
async function relationalQuery(data: { async function relationalQuery(
websiteId: string; data: Array<{
eventId: string; websiteId: string;
eventData?: DynamicData; eventId: string;
eventBatchData?: Array<DynamicData>; eventData?: DynamicData;
}): Promise<Prisma.BatchPayload> { }>,
const { websiteId, eventId, eventData, eventBatchData } = data; ): Promise<Prisma.BatchPayload> {
const listFlattenedData: {
id: string;
websiteEventId: string;
websiteId: string;
dataKey: string;
stringValue: string;
numberValue: number;
dateValue: Date;
dataType: DynamicDataType;
}[] = [];
let jsonKeys: Array<JsonKeyDynamicData> = []; for (const item of data) {
if (eventData) { const { websiteId, eventId, eventData } = item;
jsonKeys = flattenJSON(eventData); const jsonKeys = flattenJSON(eventData);
} else if (eventBatchData) {
jsonKeys = eventBatchData.flatMap(d => flattenJSON(d)); // id, websiteEventId, eventStringValue
const flattenedData = jsonKeys.map(a => ({
id: uuid(),
websiteEventId: eventId,
websiteId,
dataKey: a.key,
stringValue: getStringValue(a.value, a.dataType),
numberValue: a.dataType === DATA_TYPE.number ? a.value : null,
dateValue: a.dataType === DATA_TYPE.date ? new Date(a.value) : null,
dataType: a.dataType,
}));
listFlattenedData.push(...flattenedData);
} }
// id, websiteEventId, eventStringValue
const flattenedData = jsonKeys.map(a => ({
id: uuid(),
websiteEventId: eventId,
websiteId,
dataKey: a.key,
stringValue: getStringValue(a.value, a.dataType),
numberValue: a.dataType === DATA_TYPE.number ? a.value : null,
dateValue: a.dataType === DATA_TYPE.date ? new Date(a.value) : null,
dataType: a.dataType,
}));
return prisma.client.eventData.createMany({ return prisma.client.eventData.createMany({
data: flattenedData, data: listFlattenedData,
}); });
} }
async function clickhouseQuery(data: { async function clickhouseQuery(
websiteId: string; data: Array<{
eventId: string; websiteId: string;
sessionId?: string; eventId: string;
visitId?: string; sessionId?: string;
urlPath?: string; visitId?: string;
eventName?: string; urlPath?: string;
eventData?: DynamicData; eventName?: string;
eventBatchData?: Array<DynamicData>; eventData?: DynamicData;
createdAt?: string; createdAt?: string;
}) { }>,
const { ) {
websiteId, const { sendMessages } = kafka;
sessionId,
visitId,
eventId,
urlPath,
eventName,
eventData,
eventBatchData,
createdAt,
} = data;
const { sendMessages, sendMessage } = kafka;
const { insert, getUTCString } = clickhouse; const { insert, getUTCString } = clickhouse;
let jsonKeys: Array<JsonKeyDynamicData> = []; const messagesEventData = [];
if (eventData) { const messagesEventDataBlob = [];
jsonKeys = flattenJSON(eventData);
} else if (eventBatchData) {
jsonKeys = eventBatchData.flatMap(d => flattenJSON(d));
}
const messages = jsonKeys.map(({ key, value, dataType }) => { for (const item of data) {
return { const { websiteId, sessionId, visitId, eventId, urlPath, eventName, eventData, createdAt } =
item;
const jsonKeys = flattenJSON(eventData);
const messagesPerEventData = jsonKeys.map(({ key, value, dataType }) => {
return {
website_id: websiteId,
session_id: sessionId,
event_id: eventId,
visit_id: visitId,
url_path: urlPath,
event_name: eventName,
data_key: key,
data_type: dataType,
string_value: getStringValue(value, dataType),
number_value: dataType === DATA_TYPE.number ? value : null,
date_value: dataType === DATA_TYPE.date ? getUTCString(value) : null,
created_at: createdAt ?? getUTCString(),
};
});
messagesEventData.push(...messagesPerEventData);
const jsonBlobs = flattenDynamicData(jsonKeys);
const messageEventDataBlob: { [key: string]: string | number } = {
website_id: websiteId, website_id: websiteId,
session_id: sessionId, session_id: sessionId,
event_id: eventId, event_id: eventId,
visit_id: visitId, visit_id: visitId,
url_path: urlPath,
event_name: eventName, event_name: eventName,
data_key: key, created_at: createdAt ?? getUTCString(new Date()),
data_type: dataType,
string_value: getStringValue(value, dataType),
number_value: dataType === DATA_TYPE.number ? value : null,
date_value: dataType === DATA_TYPE.date ? getUTCString(value) : null,
created_at: createdAt ?? getUTCString(),
}; };
}); jsonBlobs.blobs.forEach((blob, i) => {
if (i >= 20) return; // 20 is the max number of blobs
messageEventDataBlob[`blob${i + 1}`] = blob;
});
jsonBlobs.doubles.forEach((double, i) => {
if (i >= 20) return; // 20 is the max number of doubles
messageEventDataBlob[`double${i + 1}`] = double;
});
if (kafka.enabled) { messagesEventDataBlob.push(messageEventDataBlob);
await sendMessages('event_data', messages);
} else {
await insert('event_data', messages);
} }
const jsonBlobs = flattenDynamicData(jsonKeys);
const message: { [key: string]: string | number } = {
website_id: websiteId,
session_id: sessionId,
event_id: eventId,
visit_id: visitId,
event_name: eventName,
created_at: createdAt ?? getUTCString(new Date()),
};
jsonBlobs.blobs.forEach((blob, i) => {
if (i >= 20) return; // 20 is the max number of blobs
message[`blob${i + 1}`] = blob;
});
jsonBlobs.doubles.forEach((double, i) => {
if (i >= 20) return; // 20 is the max number of doubles
message[`double${i + 1}`] = double;
});
if (kafka.enabled) { if (kafka.enabled) {
await sendMessage('event_data_blob', message); await Promise.all([
sendMessages('event_data', messagesEventData),
sendMessages('event_data_blob', messagesEventDataBlob),
]);
} else { } else {
await insert('event_data_blob', [message]); await Promise.all([
insert('event_data', messagesEventData),
insert('event_data_blob', messagesEventDataBlob),
]);
} }
return data; return data;