add worker id to metrics

rudderlabs · Jun 24, 2024 · fc4669f · fc4669f
1 parent 78b6cd4
commit fc4669f
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 16 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -103,7 +103,7 @@
     "oauth-1.0a": "^2.2.6",
     "object-hash": "^3.0.0",
     "parse-static-imports": "^1.1.0",
-    "prom-client": "^15.1.2",
+    "prom-client": "^14.2.0",
     "qs": "^6.11.1",
     "rudder-transformer-cdk": "^1.4.11",
     "set-value": "^4.1.0",

diff --git a/src/middleware.js b/src/middleware.js
@@ -14,6 +14,11 @@ function initPyroscope() {
   });
   Pyroscope.start();
   Pyroscope.startHeapCollecting();
+  process.on('SIGINT', () => {
+    Pyroscope.stop();
+    Pyroscope.stopHeapCollecting();
+  });
+
 }
 
 function getCPUProfile(seconds) {

diff --git a/src/util/prometheus.js b/src/util/prometheus.js
@@ -1,12 +1,17 @@
 const prometheusClient = require('prom-client');
 const logger = require('../logger');
 const { MetricsAggregator } = require('./metricsAggregator');
+const cluster = require('cluster');
 
 const clusterEnabled = process.env.CLUSTER_ENABLED !== 'false';
 const useMetricsAggregator = process.env.USE_METRICS_AGGREGATOR === 'true';
 const instanceID = process.env.INSTANCE_ID || 'localhost';
 const prefix = 'transformer';
-const defaultLabels = { instanceName: instanceID };
+const defaultLabels = { 
+  instanceName: instanceID, 
+  workerId: cluster.worker ? cluster.worker.id : 0, 
+  isMaster: cluster.isMaster  
+};
 
 function appendPrefix(name) {
   return `${prefix}_${name}`;