feat: Metrics - WorkClaim Counts (#627)

poetapp · Oct 23, 2018 · 47b99f2 · 47b99f2
1 parent a3b0dcb
commit 47b99f2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/API/Router.ts b/src/API/Router.ts
@@ -68,6 +68,7 @@ export class Router {
     this.koaRouter.get('/works', RequestValidationMiddleware(getWorksSchema), this.getWorks)
     this.koaRouter.post('/works', this.postWork)
     this.koaRouter.get('/health', this.getHealth)
+    this.koaRouter.get('/metrics', this.getWorkCounts)
 
     this.koa.use(helmet(SecurityHeaders))
     this.koa.use(KoaCors({ expose: ['X-Total-Count'] }))
@@ -113,6 +114,14 @@ export class Router {
     context.body = works
   }
 
+  private getWorkCounts = async (context: KoaRouter.IRouterContext, next: () => Promise<any>) => {
+    this.logger.trace('/metrics')
+    const TotalWorkClaims = await this.workController.getWorksCountByFilters({
+      ...context.query,
+    })
+    context.body = { TotalWorkClaims }
+  }
+
   private postWork = async (context: KoaRouter.IRouterContext, next: () => Promise<any>) => {
     const { body } = context.request
 

diff --git a/src/API/WorkController.ts b/src/API/WorkController.ts
@@ -62,6 +62,12 @@ export class WorkController {
     return { count, works }
   }
 
+  async getWorksCountByFilters(worksFilters: WorksFilters = {}): Promise<number> {
+    this.logger.trace({ method: 'getWorksCountByFilters', worksFilters }, 'Getting Work Counts by Filters from DB')
+    const count = await this.collection.find(worksFilters, { projection: { _id: false } }).count()
+    return count
+  }
+
   async create(work: Work): Promise<void> {
     this.logger.trace({ method: 'create', work }, 'Creating Work')
     // TODO: verify id, publicKey, signature and createdDate