Merge pull request #86 from pinax-network/feature/substreams-specific…

…-metrics Substreams specific metrics
pinax-network · Nov 23, 2023 · a19ae45 · a19ae45
2 parents 8e9f939 + d95516b
commit a19ae45
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 10 deletions.
diff --git a/bun.lockb b/bun.lockb
diff --git a/src/clickhouse/handleSinkRequest.ts b/src/clickhouse/handleSinkRequest.ts
@@ -19,7 +19,10 @@ export async function handleSinkRequest({ data, ...metadata }: PayloadBody) {
     sqlite.triggerTransaction();
   }
 
-  prometheus.sink_requests.inc();
+  prometheus.sink_requests.inc({
+    chain: metadata.manifest.chain,
+    module_hash: metadata.manifest.moduleHash,
+  });
   bufferedItems++;
 
   // EntityChanges
@@ -196,7 +199,10 @@ function insertEntityChange(
     )
   );
 
-  prometheus.entity_changes_inserted.inc();
+  prometheus.entity_changes_inserted.inc({
+    chain: metadata.manifest.chain,
+    module_hash: metadata.manifest.moduleHash,
+  });
 }
 
 // TODO: implement function

diff --git a/src/prometheus.ts b/src/prometheus.ts
@@ -8,10 +8,10 @@ export async function metrics() {
   return new Response(await registry.metrics(), { status: 200, headers });
 }
 
-export function registerCounter(name: string, help: string) {
+export function registerCounter<T extends string = string>(name: string, help: string, labels?: T[]) {
   try {
-    registry.registerMetric(new client.Counter({ name, help }));
-    return registry.getSingleMetric(name) as Counter;
+    registry.registerMetric(new client.Counter({ name, help, labelNames: labels ?? [] }));
+    return registry.getSingleMetric(name) as Counter<T>;
   } catch (error) {}
 }
 
@@ -22,12 +22,11 @@ export function registerGauge(name: string, help: string) {
   } catch (error) {}
 }
 
-export const requests = registerCounter("requests", "Total requests")!
+export const requests = registerCounter("requests", "Total requests")!;
 export const request_errors = registerCounter("request_errors", "Total failed requests")!;
-export const sink_requests = registerCounter("sink_requests", "Total sink requests")!;
-export const queue_size = registerGauge("queue_size", "Amount of promises being processed")!;
+export const sink_requests = registerCounter("sink_requests", "Total sink requests", ["chain", "module_hash"])!;
 
-export const entity_changes_inserted = registerCounter("entity_changes_inserted", "Total inserted entity changes")!;
+export const entity_changes_inserted = registerCounter("entity_changes_inserted", "Total inserted entity changes", ["chain", "module_hash"])!;
 export const entity_changes_updated = registerCounter("entity_changes_updated", "Total updated entity changes")!;
 export const entity_changes_deleted = registerCounter("entity_changes_deleted", "Total deleted entity changes")!;
-export const entity_changes_unsupported = registerCounter("entity_changes_unsupported", "Total unsupported entity changes")!
+export const entity_changes_unsupported = registerCounter("entity_changes_unsupported", "Total unsupported entity changes")!;