enhance(runtime): use in-context-sdk for wrapping resolvers to have b…

…etter tracing
ardatan · Sep 6, 2022 · 44b8681 · 44b8681
1 parent a6690c6
commit 44b8681
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/.changeset/tall-dingos-hammer.md b/.changeset/tall-dingos-hammer.md
@@ -0,0 +1,6 @@
+---
+'@graphql-mesh/runtime': minor
+'@graphql-mesh/types': minor
+---
+
+Use In Context SDK for wrapping resolvers for better tracing
diff --git a/packages/runtime/src/get-mesh.ts b/packages/runtime/src/get-mesh.ts
@@ -24,7 +24,7 @@ import {
   getHeadersObj,
 } from '@graphql-mesh/utils';
 
-import { SubschemaConfig } from '@graphql-tools/delegate';
+import { CreateProxyingResolverFn, SubschemaConfig } from '@graphql-tools/delegate';
 import { AggregateError, ExecutionResult, isAsyncIterable, mapAsyncIterator, memoize1 } from '@graphql-tools/utils';
 import { enableIf, envelop, PluginOrDisabledPlugin, useExtendContext } from '@envelop/core';
 import { OneOfInputObjectsRule, useExtendedValidation } from '@envelop/extended-validation';
@@ -92,6 +92,15 @@ export function wrapFetchWithPlugins(plugins: MeshPlugin<any>[]): MeshFetch {
   };
 }
 
+// Use in-context-sdk for tracing
+function createProxyingResolverFactory(apiName: string): CreateProxyingResolverFn {
+  return function createProxyingResolver() {
+    return function proxyingResolver(root, args, context, info) {
+      return context[apiName][info.parentType.name][info.fieldName]({ root, args, context, info });
+    };
+  };
+}
+
 export async function getMesh(options: GetMeshOptions): Promise<MeshInstance> {
   const rawSources: RawSourceOutput[] = [];
   const {
@@ -185,6 +194,7 @@ export async function getMesh(options: GetMeshOptions): Promise<MeshInstance> {
           handler: apiSource.handler,
           batch: 'batch' in source ? source.batch : true,
           merge: apiSource.merge,
+          createProxyingResolver: createProxyingResolverFactory(apiName),
         });
       } catch (e: any) {
         sourceLogger.error(`Failed to generate the schema`, e);

diff --git a/packages/types/src/index.ts b/packages/types/src/index.ts
@@ -2,7 +2,13 @@
 import { IResolvers, Executor } from '@graphql-tools/utils';
 import { GraphQLSchema, GraphQLResolveInfo, DocumentNode, SelectionSetNode } from 'graphql';
 import * as YamlConfig from './config';
-import { Transform, MergedTypeConfig, SubschemaConfig, IDelegateToSchemaOptions } from '@graphql-tools/delegate';
+import {
+  Transform,
+  MergedTypeConfig,
+  SubschemaConfig,
+  IDelegateToSchemaOptions,
+  CreateProxyingResolverFn,
+} from '@graphql-tools/delegate';
 import { TypedDocumentNode } from '@graphql-typed-document-node/core';
 import { MeshStore } from '@graphql-mesh/store';
 import configSchema from './config-schema.json';
@@ -186,6 +192,7 @@ export type RawSourceOutput = {
   handler: MeshHandler;
   batch: boolean;
   merge?: Record<string, MergedTypeConfig>;
+  createProxyingResolver: CreateProxyingResolverFn<any>;
 };
 
 export type GraphQLOperation<TData, TVariables> = TypedDocumentNode<TData, TVariables> | string;