aljaz/ditto
1
0
Fork 0
mirror of https://github.com/aljazceru/ditto.git synced 2025-12-18 22:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

prometheus: add gauges for websocket connections

Browse Source
This commit is contained in:
Alex Gleason
2024-06-24 09:23:26 -05:00
parent 4ed289e5c3
commit e6cd3d9e47
3 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,7 +10,7 @@ import {
import { AppController } from '@/app.ts';
import { relayInfoController } from '@/controllers/nostr/relay-info.ts';
import { relayEventCounter, relayMessageCounter } from '@/metrics.ts';
import { relayConnectionsGauge, relayEventCounter, relayMessageCounter } from '@/metrics.ts';
import * as pipeline from '@/pipeline.ts';
import { RelayError } from '@/RelayError.ts';
import { Storages } from '@/storages.ts';
@@ -22,6 +22,10 @@ const FILTER_LIMIT = 100;
function connectStream(socket: WebSocket) {
const controllers = new Map<string, AbortController>();
socket.onopen = () => {
relayConnectionsGauge.inc();
};
socket.onmessage = (e) => {
const result = n.json().pipe(n.clientMsg()).safeParse(e.data);
if (result.success) {
@@ -34,6 +38,8 @@ function connectStream(socket: WebSocket) {
};
socket.onclose = () => {
relayConnectionsGauge.dec();
for (const controller of controllers.values()) {
controller.abort();
}