aljaz/ditto
1
0
Fork 0
mirror of https://github.com/aljazceru/ditto.git synced 2026-01-16 03:44:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Streaming metrics

Browse Source
This commit is contained in:
Alex Gleason
2024-09-21 21:40:52 -05:00
parent 323d97e5e0
commit fc7228e183
4 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/controllers/api/streaming.ts
View File

@@ -61,6 +61,8 @@ const LIMITER_LIMIT = 100;
const limiter = new TTLCache<string, number>();
const connections = new Set<WebSocket>();
const streamingController: AppController = async (c) => {
const upgrade = c.req.header('upgrade');
const token = c.req.header('sec-websocket-protocol');
@@ -126,7 +128,8 @@ const streamingController: AppController = async (c) => {
}
socket.onopen = async () => {
streamingConnectionsGauge.inc();
connections.add(socket);
streamingConnectionsGauge.set(connections.size);
if (!stream) return;
const topicFilter = await topicToFilter(stream, c.req.query(), pubkey);
@@ -186,7 +189,8 @@ const streamingController: AppController = async (c) => {
};
socket.onclose = () => {
streamingConnectionsGauge.dec();
connections.delete(socket);
streamingConnectionsGauge.set(connections.size);
controller.abort();
};