Adds the root sentry span to a resolver's context

packages/plugins/sentry/__tests__/sentry.spec.ts

@@ -257,3 +257,41 @@ describe('sentry', () => {
     });
   });
 });
+
+test('sets the span in the query context', async () => {
+  const { testkit: sentryTestkit, sentryTransport } = createSentryTestkit();
+  Sentry.init({
+    dsn: 'https://[email protected]/2',
+    transport: sentryTransport,
+  });
+
+  // The test verifies that the span is set in the context, which would be
+  // available as the third argument
+  const schema = makeExecutableSchema({
+    typeDefs: /* GraphQL */ `
+      type Query {
+        hasSentryInContext: String!
+      }
+    `,
+    resolvers: {
+      Query: {
+        hasSentryInContext: async (args, info, ctx) => {
+          return !!ctx.sentry;
+        },
+      },
+    },
+  });
+
+  const envelopTestkit = createTestkit([useSentry()], schema);
+  const result = await envelopTestkit.execute('{ hasSentryInContext }');
+  expect(result).toMatchInlineSnapshot(`
+    Object {
+      "data": Object {
+        "hasSentryInContext": "true",
+      },
+    }
+  `);
+
+  // run sentry flush
+  await new Promise(res => setTimeout(res, 10));
+});

packages/plugins/sentry/src/index.ts

@@ -104,7 +104,7 @@ export const useSentry = (options: SentryPluginOptions = {}): Plugin => {
   }
 
   return {
-    onExecute({ args }) {
+    onExecute({ args, extendContext }) {
       if (skipOperation(args)) {
         return;
       }
@@ -179,6 +179,8 @@ export const useSentry = (options: SentryPluginOptions = {}): Plugin => {
         Sentry.configureScope(scope => options.configureScope!(args, scope));
       }
 
+      extendContext({ sentry: { span: rootSpan } });
+
       return {
         onExecuteDone(payload) {
           const handleResult: OnExecuteDoneHookResultOnNextHook<{}> = ({ result, setResult }) => {