mirror/ditto
mirror/ditto
1
0
Fork 0
mirror of https://gitlab.com/soapbox-pub/ditto.git synced 2025-12-06 11:29:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

prometheus: track relay messages by verb

Browse source
This commit is contained in:
Alex Gleason 2024-06-24 08:18:11 -05:00
parent 961cb070f0
commit e0b1029953
No known key found for this signature in database
GPG key ID: 7211D1F99744FBB7
2 changed files with 5 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
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 { relayCountCounter, relayEventCounter, relayMessageCounter, relayReqCounter } from '@/metrics.ts';
import { relayEventCounter, relayMessageCounter } from '@/metrics.ts';
import * as pipeline from '@/pipeline.ts';
import { RelayError } from '@/RelayError.ts';
import { Storages } from '@/storages.ts';
@ -23,11 +23,12 @@ function connectStream(socket: WebSocket) {
const controllers = new Map<string, AbortController>();
socket.onmessage = (e) => {
relayMessageCounter.inc();
const result = n.json().pipe(n.clientMsg()).safeParse(e.data);
if (result.success) {
relayMessageCounter.inc({ verb: result.data[0] });
handleMsg(result.data);
} else {
relayMessageCounter.inc();
send(['NOTICE', 'Invalid message.']);
}
};
@ -42,18 +43,15 @@ function connectStream(socket: WebSocket) {
function handleMsg(msg: NostrClientMsg) {
switch (msg[0]) {
case 'REQ':
relayReqCounter.inc();
handleReq(msg);
return;
case 'EVENT':
relayEventCounter.inc({ kind: msg[1].kind.toString() });
handleEvent(msg);
return;
case 'CLOSE':
handleClose(msg);
return;
case 'COUNT':
relayCountCounter.inc();
handleCount(msg);
return;
}
@ -93,6 +91,7 @@ function connectStream(socket: WebSocket) {
/** Handle EVENT. Store the event. */
async function handleEvent([_, event]: NostrClientEVENT): Promise<void> {
relayEventCounter.inc({ kind: event.kind.toString() });
try {
// This will store it (if eligible) and run other side-effects.
await pipeline.handleEvent(event, AbortSignal.timeout(1000));

11
src/metrics.ts
View file

@ -24,25 +24,16 @@ export const pipelineEventCounter = new Counter({
labelNames: ['kind'],
});
export const relayReqCounter = new Counter({
name: 'relay_reqs_total',
help: 'Total number of REQ messages processed by the relay',
});
export const relayEventCounter = new Counter({
name: 'relay_events_total',
help: 'Total number of EVENT messages processed by the relay',
labelNames: ['kind'],
});
export const relayCountCounter = new Counter({
name: 'relay_counts_total',
help: 'Total number of COUNT messages processed by the relay',
});
export const relayMessageCounter = new Counter({
name: 'relay_messages_total',
help: 'Total number of Nostr messages processed by the relay',
labelNames: ['verb'],
});
export const dbQueryCounter = new Counter({