feat: network for build info

lidofinance · Apr 16, 2022 · d53d72d · d53d72d
1 parent a0f746e
commit d53d72d
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/src/app.service.ts b/src/app.service.ts
@@ -4,18 +4,21 @@ import { APP_NAME, APP_VERSION } from 'app.constants';
 import { METRIC_BUILD_INFO } from 'common/prometheus';
 import { WINSTON_MODULE_NEST_PROVIDER } from 'nest-winston';
 import { Gauge } from 'prom-client';
+import { ProviderService } from 'provider';
 
 export class AppService implements OnModuleInit {
   constructor(
+    private providerService: ProviderService,
     @Inject(WINSTON_MODULE_NEST_PROVIDER) private logger: LoggerService,
     @InjectMetric(METRIC_BUILD_INFO) private buildInfo: Gauge<string>,
   ) {}
 
-  onModuleInit() {
+  async onModuleInit(): Promise<void> {
+    const network = await this.providerService.getNetworkName();
     const version = APP_VERSION;
     const name = APP_NAME;
 
-    this.buildInfo.labels({ version, name }).inc();
+    this.buildInfo.labels({ version, name, network }).inc();
     this.logger.log('Init app', { name, version });
   }
 }
diff --git a/src/common/prometheus/prometheus.provider.ts b/src/common/prometheus/prometheus.provider.ts
@@ -57,7 +57,7 @@ export const PrometheusBlockDataErrorsCounterProvider = makeCounterProvider({
 export const PrometheusBuildInfoGaugeProvider = makeCounterProvider({
   name: METRIC_BUILD_INFO,
   help: 'Build information',
-  labelNames: ['version', 'name'] as const,
+  labelNames: ['version', 'name', 'network'] as const,
 });
 
 export const PrometheusDepositedKeysProvider = makeGaugeProvider({

diff --git a/src/provider/provider.service.ts b/src/provider/provider.service.ts
@@ -1,4 +1,5 @@
 import { Block } from '@ethersproject/abstract-provider';
+import { CHAINS } from '@lido-sdk/constants';
 import { Injectable } from '@nestjs/common';
 import { RpcBatchProvider, RpcProvider } from './interfaces';
 
@@ -48,4 +49,13 @@ export class ProviderService {
   public async getBlock(): Promise<Block> {
     return await this.provider.getBlock('latest');
   }
+
+  /**
+   * Returns network name
+   */
+  public async getNetworkName(): Promise<string> {
+    const network = await this.provider.getNetwork();
+    const name = CHAINS[network.chainId]?.toLocaleLowerCase();
+    return name || network.name;
+  }
 }