core: add balloonTask metric

Author: pandadtdyy

packages/server/handler/client.ts

@@ -99,6 +99,7 @@ class ClientBallloonConnectHandler extends Handler {
         await this.ctx.db.client.updateOne({ id: params.cid }, { $set: { updateAt: new Date().getTime(), ip } });
         await this.ctx.db.balloon.update({ balloonid: { $in: balloons.map((b) => b.balloonid) } },
             { $set: { receivedAt: new Date().getTime() } }, { multi: true });
+        await this.ctx.parallel('balloon/sendTask', client._id, balloons.length);
     }
 }
 
@@ -110,6 +111,7 @@ class ClientBalloonDoneHandler extends Handler {
         if (!balloon) throw new ValidationError('Balloon', params.bid, 'Balloon not found');
         await this.ctx.db.balloon.updateOne({ balloonid: +params.bid }, { $set: { printDone: 1, printDoneAt: new Date().getTime() } });
         if (!balloon.done) await this.ctx.fetcher.setBalloonDone(balloon.balloonid);
+        await this.ctx.parallel('balloon/doneTask', client._id, 1);
         this.response.body = { code: 1 };
         logger.info(`Client ${client.name} connected, balloon task ${balloon.teamid}#${balloon.balloonid} completed.`);
     }

packages/server/service/fetcher.ts

@@ -148,6 +148,7 @@ class DOMjudgeFetcher extends BasicFetcher {
                 },
             }, { upsert: true });
         }
+        await this.ctx.parallel('balloon/newTask', balloons.length);
         logger.debug(`Found ${balloons.length} balloons`);
     }
 
@@ -234,6 +235,7 @@ class HydroFetcher extends BasicFetcher {
                 },
             }, { upsert: true });
         }
+        await this.ctx.parallel('balloon/newTask', balloons.length);
         logger.debug(`Found ${balloons.length} balloons`);
     }
 

packages/server/utils/metrics.ts

@@ -11,6 +11,9 @@ declare module 'cordis' {
         'print/newTask': () => void;
         'print/sendTask': (client: string) => void;
         'print/doneTask': (client: string, printer: string) => void;
+        'balloon/newTask': (count: number) => void;
+        'balloon/sendTask': (client: string, count: number) => void;
+        'balloon/doneTask': (client: string, count: number) => void;
     }
 }
 
@@ -44,6 +47,15 @@ export function createMetricsRegistry(ctx: Context) {
 
     ctx.on('print/doneTask', (client, printer) => printTaskCounter.inc({ status: 'done', client, printer }));
 
+    const balloonTaskCounter = createMetric(Counter, 'xcpc_ballooncount', 'ballooncount', {
+        labelNames: ['status', 'client'],
+    });
+    ctx.on('balloon/newTask', (count) => balloonTaskCounter.inc({ status: 'new' }, count));
+
+    ctx.on('balloon/sendTask', (client, count) => balloonTaskCounter.inc({ status: 'sent', client }, count));
+
+    ctx.on('balloon/doneTask', (client, count) => balloonTaskCounter.inc({ status: 'done', client }, count));
+
     collectDefaultMetrics({ register: registry });
 
     ctx.set('metrics', registry);