From d02934f766bcc077e73b459c253dc6eeb8ca35a9 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sun, 27 Nov 2022 01:05:07 +0800
Subject: [PATCH] prom: set category for reqCounter

---
 packages/center/index.ts        | 1 +
 packages/prom-client/metrics.ts | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/center/index.ts b/packages/center/index.ts
index 54e56c2cc..f51bc9083 100644
--- a/packages/center/index.ts
+++ b/packages/center/index.ts
@@ -25,6 +25,7 @@ const coll = db.collection('dataReport');
 
 class DataReportHandler extends Handler {
     noCheckPermView = true;
+    category = '#center';
 
     @post('installId', Types.String)
     @post('payload', Types.String)
diff --git a/packages/prom-client/metrics.ts b/packages/prom-client/metrics.ts
index 27268c18d..b4eea1325 100644
--- a/packages/prom-client/metrics.ts
+++ b/packages/prom-client/metrics.ts
@@ -26,7 +26,7 @@ export function createRegistry(ctx: Context) {
     const reqCounter = createMetric(Counter, 'hydro_reqcount', 'reqcount', {
         labelNames: ['domainId'],
     });
-    ctx.on('handler/create', (h) => reqCounter.inc({ domainId: h.args.domainId }));
+    ctx.on('handler/create', (h) => reqCounter.inc({ domainId: (h as any).category || h.args.domainId }));
 
     const judgeCounter = createMetric(Counter, 'hydro_judgecount', 'judgecount');
     ctx.on('record/judge', () => judgeCounter.inc());