mirror of
https://gitlab.com/soapbox-pub/ditto.git
synced 2025-12-06 11:29:46 +00:00
prometheus: track relay messages by verb
This commit is contained in:
parent
961cb070f0
commit
e0b1029953
2 changed files with 5 additions and 15 deletions
|
|
@ -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));
|
||||
|
|
|
|||
|
|
@ -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({
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue