Allow schema reporting env var

apollographql · Jun 4, 2020 · 2d613cf · 2d613cf
1 parent 5519db7
commit 2d613cf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/packages/apollo-engine-reporting/src/agent.ts b/packages/apollo-engine-reporting/src/agent.ts
@@ -389,6 +389,7 @@ export class EngineReportingAgent<TContext = any> {
   };
 
   private readonly tracesEndpointUrl: string;
+  private readonly schemaReport: boolean;
 
   public constructor(options: EngineReportingOptions<TContext> = {}) {
     this.options = options;
@@ -407,6 +408,13 @@ export class EngineReportingAgent<TContext = any> {
       );
     }
 
+
+    if (options.experimental_schemaReporting !== undefined) {
+      this.schemaReport = options.experimental_schemaReporting;
+    } else {
+      this.schemaReport = process.env.APOLLO_SCHEMA_REPORTING === "true"
+    }
+
     // Since calculating the signature for Engine reporting is potentially an
     // expensive operation, we'll cache the signatures we generate and re-use
     // them based on repeated traces for the same `queryHash`.
@@ -469,6 +477,7 @@ export class EngineReportingAgent<TContext = any> {
     return plugin(this.options, this.addTrace.bind(this), {
       startSchemaReporting: this.startSchemaReporting.bind(this),
       executableSchemaIdGenerator: this.executableSchemaIdGenerator.bind(this),
+      schemaReport: this.schemaReport,
     });
   }
 

diff --git a/packages/apollo-engine-reporting/src/plugin.ts b/packages/apollo-engine-reporting/src/plugin.ts
@@ -34,6 +34,7 @@ export const plugin = <TContext>(
   {
     startSchemaReporting,
     executableSchemaIdGenerator,
+    schemaReport,
   }: {
     startSchemaReporting: ({
       executableSchema,
@@ -43,6 +44,7 @@ export const plugin = <TContext>(
       executableSchemaId: string;
     }) => void;
     executableSchemaIdGenerator: (schema: string | GraphQLSchema) => string;
+    schemaReport?: boolean;
   },
 ): ApolloServerPlugin<TContext> => {
   /**
@@ -57,7 +59,7 @@ export const plugin = <TContext>(
 
   return {
     serverWillStart: function({ schema }) {
-      if (!options.experimental_schemaReporting) return;
+      if (!schemaReport) return;
       startSchemaReporting({
         executableSchema:
           options.experimental_overrideReportedSchema || printSchema(schema),