provide a results callback hook

JupiterOne · Dec 15, 2023 · 8ec344a · 8ec344a
1 parent 19bbe6a
commit 8ec344a
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/packages/integration-sdk-runtime/src/execution/executeIntegration.ts b/packages/integration-sdk-runtime/src/execution/executeIntegration.ts
@@ -62,13 +62,14 @@ export interface ExecuteIntegrationOptions {
   enableSchemaValidation?: boolean;
   graphObjectStore?: GraphObjectStore;
   createStepGraphObjectDataUploader?: CreateStepGraphObjectDataUploaderFunction;
+  resultsCallback?: (results: IntegrationStepResult[]) => Promise<void>;
   pretty?: boolean;
 }
 
-export interface ExecuteWithContextOptions {
-  graphObjectStore?: GraphObjectStore;
-  createStepGraphObjectDataUploader?: CreateStepGraphObjectDataUploaderFunction;
-}
+type ExecuteWithContextOptions = Pick<
+  ExecuteIntegrationOptions,
+  'graphObjectStore' | 'resultsCallback' | 'createStepGraphObjectDataUploader'
+>;
 
 const THIRTY_SECONDS_STORAGE_INTERVAL_MS = 60000 / 2;
 
@@ -251,6 +252,7 @@ export async function executeWithContext<
       const {
         graphObjectStore = new FileSystemGraphObjectStore(),
         createStepGraphObjectDataUploader,
+        resultsCallback,
       } = options;
 
       const integrationStepResults = await executeSteps({
@@ -288,6 +290,17 @@ export async function executeWithContext<
         data: summary,
       });
 
+      if (resultsCallback != null) {
+        try {
+          await resultsCallback(integrationStepResults);
+        } catch (err) {
+          context.logger.warn(
+            err,
+            'Unable to run results callback for integration job',
+          );
+        }
+      }
+
       processDeclaredTypesDiff(summary, (step, undeclaredTypes) => {
         if (
           step.status === StepResultStatus.SUCCESS &&