111 lines
2.8 KiB
TypeScript
111 lines
2.8 KiB
TypeScript
import type { FastifyReply, FastifyRequest } from 'fastify';
|
|
import { escape } from 'sqlstring';
|
|
import type * as WebSocket from 'ws';
|
|
|
|
import { getSafeJson } from '@openpanel/common';
|
|
import type { IServiceCreateEventPayload } from '@openpanel/db';
|
|
import { getEvents, getLiveVisitors } from '@openpanel/db';
|
|
import { redis, redisPub, redisSub } from '@openpanel/redis';
|
|
|
|
export function getLiveEventInfo(key: string) {
|
|
return key.split(':').slice(2) as [string, string];
|
|
}
|
|
|
|
export async function test(
|
|
req: FastifyRequest<{
|
|
Params: {
|
|
projectId: string;
|
|
};
|
|
}>,
|
|
reply: FastifyReply
|
|
) {
|
|
const [event] = await getEvents(
|
|
`SELECT * FROM events WHERE project_id = ${escape(req.params.projectId)} AND name = 'screen_view' LIMIT 1`
|
|
);
|
|
if (!event) {
|
|
return reply.status(404).send('No event found');
|
|
}
|
|
redisPub.publish('event', JSON.stringify(event));
|
|
redis.set(
|
|
`live:event:${event.projectId}:${Math.random() * 1000}`,
|
|
'',
|
|
'EX',
|
|
10
|
|
);
|
|
reply.status(202).send(event);
|
|
}
|
|
|
|
export function wsVisitors(
|
|
connection: {
|
|
socket: WebSocket;
|
|
},
|
|
req: FastifyRequest<{
|
|
Params: {
|
|
projectId: string;
|
|
};
|
|
}>
|
|
) {
|
|
const { params } = req;
|
|
|
|
redisSub.subscribe('event');
|
|
redisSub.psubscribe('__key*:expired');
|
|
|
|
const message = (channel: string, message: string) => {
|
|
if (channel === 'event') {
|
|
const event = getSafeJson<IServiceCreateEventPayload>(message);
|
|
if (event?.projectId === params.projectId) {
|
|
getLiveVisitors(params.projectId).then((count) => {
|
|
connection.socket.send(String(count));
|
|
});
|
|
}
|
|
}
|
|
};
|
|
const pmessage = (pattern: string, channel: string, message: string) => {
|
|
const [projectId] = getLiveEventInfo(message);
|
|
if (projectId && projectId === params.projectId) {
|
|
getLiveVisitors(params.projectId).then((count) => {
|
|
connection.socket.send(String(count));
|
|
});
|
|
}
|
|
};
|
|
|
|
redisSub.on('message', message);
|
|
redisSub.on('pmessage', pmessage);
|
|
|
|
connection.socket.on('close', () => {
|
|
redisSub.unsubscribe('event');
|
|
redisSub.punsubscribe('__key*:expired');
|
|
redisSub.off('message', message);
|
|
redisSub.off('pmessage', pmessage);
|
|
});
|
|
}
|
|
|
|
export function wsEvents(
|
|
connection: {
|
|
socket: WebSocket;
|
|
},
|
|
req: FastifyRequest<{
|
|
Params: {
|
|
projectId: string;
|
|
};
|
|
}>
|
|
) {
|
|
const { params } = req;
|
|
|
|
redisSub.subscribe('event');
|
|
|
|
const message = (channel: string, message: string) => {
|
|
const event = getSafeJson<IServiceCreateEventPayload>(message);
|
|
if (event?.projectId === params.projectId) {
|
|
connection.socket.send(JSON.stringify(event));
|
|
}
|
|
};
|
|
|
|
redisSub.on('message', message);
|
|
|
|
connection.socket.on('close', () => {
|
|
redisSub.unsubscribe('event');
|
|
redisSub.off('message', message);
|
|
});
|
|
}
|