enhance(prom): use params.operationName if available in the context

Author: ardatan

packages/plugins/prometheus/src/index.ts

@@ -229,7 +229,11 @@ export const usePrometheus = (config: PrometheusTracingPluginConfig = {}): Plugi
       const totalTime = (Date.now() - startTime) / 1000;
       let fillLabelsFnParams = fillLabelsFnParamsMap.get(params.result);
       if (!fillLabelsFnParams) {
-        fillLabelsFnParams = createFillLabelFnParams(params.result, filterFillParamsFnParams);
+        fillLabelsFnParams = createFillLabelFnParams(
+          params.result,
+          context,
+          filterFillParamsFnParams,
+        );
         fillLabelsFnParamsMap.set(context, fillLabelsFnParams);
       }
 

packages/plugins/prometheus/src/utils.ts

@@ -48,6 +48,7 @@ function getOperation(document: DocumentNode): OperationDefinitionNode {
 
 export function createFillLabelFnParams(
   parseResult: AfterParseEventPayload<any>['result'],
+  context: any,
   filterParams: (params: FillLabelsFnParams) => FillLabelsFnParams | null,
 ): FillLabelsFnParams | null {
   if (parseResult === null) {
@@ -59,7 +60,7 @@ export function createFillLabelFnParams(
   const operation = getOperation(parseResult);
   return filterParams({
     document: parseResult,
-    operationName: operation.name?.value || 'Anonymous',
+    operationName: context?.params?.operationName || operation.name?.value || 'Anonymous',
     operationType: operation.operation,
   });
 }