prom-client: add category for connection handler

packages/hydrooj/src/handler/home.ts

@@ -363,6 +363,8 @@ class HomeDomainCreateHandler extends Handler {
 }
 
 class HomeMessagesHandler extends Handler {
+    category = '#message';
+
     async get() {
         // TODO(iceboy): projection, pagination.
         const messages = await message.getByUser(this.user._id);

packages/hydrooj/src/handler/judge.ts

@@ -172,6 +172,7 @@ export class SubmissionDataDownloadHandler extends Handler {
 }
 
 class JudgeConnectionHandler extends ConnectionHandler {
+    category = '#judge';
     processing: any = null;
     closed = false;
     query: any = { type: 'judge' };

packages/prom-client/metrics.ts

@@ -40,10 +40,10 @@ export function createRegistry(ctx: Context) {
         labelNames: ['domainId'],
     });
     ctx.on('connection/active', (h) => {
-        connectionGauge.inc({ domainId: h.args.domainId });
+        connectionGauge.inc({ domainId: (h as any).category || h.args.domainId });
     });
     ctx.on('connection/close', (h) => {
-        connectionGauge.dec({ domainId: h.args.domainId });
+        connectionGauge.dec({ domainId: (h as any).category || h.args.domainId });
     });
 
     const taskColl = db.collection('task');