fix(grpc-sdk): admin_grpc_requests_total metric (#372)

ConduitPlatform · Oct 7, 2022 · 5af2ada · 5af2ada
1 parent 29b3d87
commit 5af2ada
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/libraries/grpc-sdk/src/routing/RoutingManager.ts b/libraries/grpc-sdk/src/routing/RoutingManager.ts
@@ -121,7 +121,7 @@ export class RoutingManager {
     if (Object.keys(this._routeHandlers).length === 0) return;
     const modifiedFunctions: {
       [name: string]: (call: any, callback: any) => void;
-    } = wrapFunctionsAsync(this._routeHandlers);
+    } = wrapFunctionsAsync(this._routeHandlers, this.isAdmin ? 'admin' : 'client');
     const protoDescriptions = constructProtoFile(
       this._router.moduleName,
       Object.values(this._moduleRoutes),

diff --git a/libraries/grpc-sdk/src/routing/RoutingUtilities.ts b/libraries/grpc-sdk/src/routing/RoutingUtilities.ts
@@ -134,14 +134,17 @@ function getFormattedModuleName(moduleName: string) {
   return moduleName.replace('-', '_');
 }
 
-export function wrapFunctionsAsync(functions: { [name: string]: RequestHandlers }): {
+export function wrapFunctionsAsync(
+  functions: { [name: string]: RequestHandlers },
+  routerType: 'admin' | 'client',
+): {
   [name: string]: (call: Indexable, callback?: Indexable) => void;
 } {
   const modifiedFunctions: {
     [name: string]: (call: Indexable, callback?: Indexable) => void;
   } = {};
   Object.keys(functions).forEach(key => {
-    modifiedFunctions[key] = wrapRouterGrpcFunction(functions[key], 'client');
+    modifiedFunctions[key] = wrapRouterGrpcFunction(functions[key], routerType);
   });
   return modifiedFunctions;
 }