narrowed allowed labels

pinax-network · Nov 22, 2023 · 75e65a7 · 75e65a7
1 parent 9dd1326
commit 75e65a7
Showing 1 changed file with 3 additions and 3 deletions.
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, labels?: string[]) {
+export function registerCounter<T extends string = string>(name: string, help: string, labels?: T[]) {
   try {
-    registry.registerMetric(new client.Counter({ name, help, labelNames: labels ?? []}));
-    return registry.getSingleMetric(name) as Counter;
+    registry.registerMetric(new client.Counter({ name, help, labelNames: labels ?? [] }));
+    return registry.getSingleMetric(name) as Counter<T>;
   } catch (error) {}
 }