Add pow metrics

src/file/file.controller.ts

@@ -3,9 +3,11 @@ import { FileService } from './file.service';
 import { Post } from '@nestjs/common';
 import { UploadedObjectInfo } from 'minio';
 import { FileInterceptor } from '@nestjs/platform-express';
-import { ApiConsumes } from '@nestjs/swagger';
+import { ApiConsumes, ApiTags } from '@nestjs/swagger';
+import { InjectMetric } from '@willsoto/nestjs-prometheus';
 
 @Controller('file')
+@ApiTags('file')
 export class FileController {
     constructor(private readonly fileService: FileService) { }
 

src/kv/kv.controller.ts

@@ -12,7 +12,7 @@ import {
 } from '@nestjs/common';
 import { KvService } from './kv.service';
 import { Request } from 'express';
-import { ApiTags } from '@nestjs/swagger';
+import { ApiQuery, ApiTags } from '@nestjs/swagger';
 import { FileInterceptor } from '@nestjs/platform-express';
 
 @Controller('kv')
@@ -21,6 +21,8 @@ export class KvController {
   constructor(private readonly kvService: KvService) { }
 
   @Get(':namespace/:key/metadata')
+  @ApiQuery({ name: 'namespace', required: true })
+  @ApiQuery({ name: 'key', required: true })
   async getMetadata(
     @Param('namespace') namespace: string,
     @Param('key') key: string,

src/pow/pow.controller.ts

@@ -1,4 +1,4 @@
-import { Body, Controller, Get, Post, Query } from '@nestjs/common';
+import { Body, Controller, Get, Post, Query, Render } from '@nestjs/common';
 import { PowService } from './pow.service';
 import { ApiBody, ApiQuery, ApiTags } from '@nestjs/swagger';
 
@@ -9,6 +9,13 @@ export class PowController {
         private readonly powService: PowService,
     ) { }
 
+    @Get('')
+    @Render('pow/index')
+    async index() {
+        return {
+            difficulty: this.powService.getDifficulty(),
+        };
+    }
 
     @Get('challenge')
     async generateChallenge() {

src/pow/pow.module.ts

@@ -2,6 +2,7 @@ import { Module } from '@nestjs/common';
 import { PowService } from './pow.service';
 import { CacheModule } from '@nestjs/cache-manager';
 import { PowController } from './pow.controller';
+import { PrometheusModule, makeGaugeProvider } from '@willsoto/nestjs-prometheus';
 
 @Module({
   imports: [
@@ -9,8 +10,35 @@ import { PowController } from './pow.controller';
       ttl: 5,
       max: 10,
     }),
+    PrometheusModule.register({
+      customMetricPrefix: 'pow',
+      defaultMetrics: {
+        enabled: false,
+      },
+    }),
+  ],
+  providers: [PowService,
+    makeGaugeProvider({
+      name: 'challenges_generated',
+      help: 'The total number of POW challenges generated',
+    }),
+    makeGaugeProvider({
+      name: 'challenges_completed',
+      help: 'The total number of POW challenges completed',
+    }),
+    makeGaugeProvider({
+      name: 'successful_verifies',
+      help: 'The total number of successful POW challenge verifications',
+    }),
+    makeGaugeProvider({
+      name: 'failed_verifies',
+      help: 'The total number of failed POW challenge verifications',
+    }),
+    makeGaugeProvider({
+      name: 'difficulty',
+      help: 'The current POW difficulty',
+    }),
   ],
-  providers: [PowService],
   controllers: [PowController]
 })
 export class PowModule { }

src/pow/pow.service.ts

@@ -1,5 +1,6 @@
 import { CACHE_MANAGER } from '@nestjs/cache-manager';
 import { Inject, Injectable, Logger } from '@nestjs/common';
+import { InjectMetric } from '@willsoto/nestjs-prometheus';
 import { Cache } from 'cache-manager';
 import { randomBytes, createHash } from 'crypto';
 
@@ -11,7 +12,14 @@ export class PowService {
 
     constructor(
         @Inject(CACHE_MANAGER) private cacheManager: Cache,
-    ) { }
+        @InjectMetric('challenges_generated') private challengesGenerated: any,
+        @InjectMetric('challenges_completed') private challengesCompleted: any,
+        @InjectMetric('successful_verifies') private successfulVerifies: any,
+        @InjectMetric('failed_verifies') private failedVerifies: any,
+        @InjectMetric('difficulty') private powDifficulty: any,
+    ) {
+        this.powDifficulty.set(this.difficulty);
+    }
 
     /**
      * Generate a proof of work challenge, stored to redis for verification within
@@ -19,7 +27,8 @@ export class PowService {
      */
     async generateChallenge() {
         const challenge = this.generateRandom256BitString();
-        console.log(challenge)
+        this.logger.verbose(`Generated challenge: ${challenge}`);
+        this.challengesGenerated.inc();
         await this.cacheManager.set(challenge, true, 60 * 1000);
         return challenge;
     }
@@ -44,10 +53,17 @@ export class PowService {
      */
     async verifyChallenge(challenge: string, proof: string): Promise<boolean> {
         const expected = await this.cacheManager.get<boolean>(challenge);
-        return expected ? this.hashAndCheck(proof + challenge) : false;
+        const success = expected ? this.hashAndCheck(proof + challenge) : false;
+        if (success) {
+            this.successfulVerifies.inc();
+        } else {
+            this.failedVerifies.inc();
+        }
+        return success;
     }
 
     async markChallengeAsComplete(challenge: string) {
+        this.challengesCompleted.inc();
         await this.cacheManager.del(challenge);
     }
 
@@ -76,6 +92,7 @@ export class PowService {
      */
     setDifficulty(difficulty: number) {
         this.difficulty = difficulty;
+        this.powDifficulty.set(this.difficulty);
     }
 
     /**