no collection metrics

JupiterOne · May 14, 2024 · 587a41a · 587a41a
1 parent fa9129b
commit 587a41a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/packages/integration-sdk-runtime/src/execution/executeIntegration.ts b/packages/integration-sdk-runtime/src/execution/executeIntegration.ts
@@ -285,6 +285,10 @@ export async function executeWithContext<
         }
       }
 
+      if (process.env.NO_COLLECTION_METRICS) {
+        logger.info('Disabling collection metrics');
+      }
+
       const integrationStepResults = await executeSteps({
         executionContext: context,
         integrationSteps: config.integrationSteps,
@@ -296,8 +300,12 @@ export async function executeWithContext<
         createStepGraphObjectDataUploader,
         beforeAddEntity: config.beforeAddEntity,
         beforeAddRelationship: config.beforeAddRelationship,
-        afterAddEntity: createAfterAddEntityInternalHook(logger),
-        afterAddRelationship: createAfterAddRelationshipInternalHook(logger),
+        afterAddEntity: process.env.NO_COLLECTION_METRICS
+          ? undefined
+          : createAfterAddEntityInternalHook(logger),
+        afterAddRelationship: process.env.NO_COLLECTION_METRICS
+          ? undefined
+          : createAfterAddRelationshipInternalHook(logger),
         dependencyGraphOrder: config.dependencyGraphOrder,
         executionHandlerWrapper: config.executionHandlerWrapper,
       });