fix: enable prometheus metrics (#7)

hirosystems · Jul 2, 2024 · e3533ef · e3533ef
1 parent bd4d666
commit e3533ef
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 15 deletions.
diff --git a/api/src/index.ts b/api/src/index.ts
@@ -1,8 +1,13 @@
-import { isProdEnv, logger, registerShutdownConfig } from '@hirosystems/api-toolkit';
+import {
+  buildPrometheusServer,
+  isProdEnv,
+  logger,
+  registerShutdownConfig,
+} from '@hirosystems/api-toolkit';
 import { buildApiServer } from './api/init';
 import { ENV } from './env';
 import { PgStore } from './pg/pg-store';
-// import { ApiMetrics } from './metrics/metrics';
+import { ApiMetrics } from './metrics/metrics';
 
 async function initApiService(db: PgStore) {
   logger.info('Initializing API service...');
@@ -17,19 +22,18 @@ async function initApiService(db: PgStore) {
 
   await fastify.listen({ host: ENV.API_HOST, port: ENV.API_PORT });
 
-  // if (isProdEnv) {
-  //   const promServer = await buildPromServer({ metrics: fastify.metrics });
-  //   registerShutdownConfig({
-  //     name: 'Prometheus Server',
-  //     forceKillable: false,
-  //     handler: async () => {
-  //       await promServer.close();
-  //     },
-  //   });
-
-  //   // ApiMetrics.configure(db);
-  //   await promServer.listen({ host: ENV.API_HOST, port: 9153 });
-  // }
+  if (isProdEnv) {
+    const promServer = await buildPrometheusServer({ metrics: fastify.metrics });
+    registerShutdownConfig({
+      name: 'Prometheus Server',
+      forceKillable: false,
+      handler: async () => {
+        await promServer.close();
+      },
+    });
+    ApiMetrics.configure(db);
+    await promServer.listen({ host: ENV.API_HOST, port: 9153 });
+  }
 }
 
 async function initApp() {

diff --git a/api/src/metrics/metrics.ts b/api/src/metrics/metrics.ts
@@ -0,0 +1,22 @@
+import * as prom from 'prom-client';
+import { PgStore } from '../pg/pg-store';
+
+export class ApiMetrics {
+  /** The most recent Bitcoin block height ingested by the API */
+  readonly runes_api_block_height: prom.Gauge;
+
+  static configure(db: PgStore): ApiMetrics {
+    return new ApiMetrics(db);
+  }
+
+  private constructor(db: PgStore) {
+    this.runes_api_block_height = new prom.Gauge({
+      name: `runes_api_block_height`,
+      help: 'The most recent Bitcoin block height ingested by the API',
+      async collect() {
+        const height = await db.getChainTipBlockHeight();
+        this.set(parseInt(height ?? '0'));
+      },
+    });
+  }
+}