aljaz/ditto
1
0
Fork 0
mirror of https://github.com/aljazceru/ditto.git synced 2025-12-19 06:24:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

relay: fix connection metrics

Browse Source
This commit is contained in:
Alex Gleason
2024-09-21 14:48:50 -05:00
parent 9c9b87bc94
commit 5a0a2087e5
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/controllers/nostr/relay.ts
View File

@@ -26,14 +26,16 @@ const LIMITER_LIMIT = 300;
const limiter = new TTLCache<string, number>();
/** Connections for metrics purposes. */
const connections = new Set<WebSocket>();
/** Set up the Websocket connection. */
function connectStream(socket: WebSocket, ip: string | undefined) {
let opened = false;
const controllers = new Map<string, AbortController>();
socket.onopen = () => {
opened = true;
relayConnectionsGauge.inc();
connections.add(socket);
relayConnectionsGauge.set(connections.size);
};
socket.onmessage = (e) => {
@@ -63,9 +65,8 @@ function connectStream(socket: WebSocket, ip: string | undefined) {
};
socket.onclose = () => {
if (opened) {
relayConnectionsGauge.dec();
}
connections.delete(socket);
relayConnectionsGauge.set(connections.size);
for (const controller of controllers.values()) {
controller.abort();