From 8106afc7fdbd9e4f64b611a6b442a8ce2b206b6d Mon Sep 17 00:00:00 2001
From: Leon <leonvanginneken@gmail.com>
Date: Fri, 21 Oct 2022 15:09:34 +0200
Subject: [PATCH] feat: add initialDelay as option

Signed-off-by: Leon <leonvanginneken@gmail.com>
---
 .../src/service/fact/FactRetrieverEngine.ts            | 10 ++++++++--
 plugins/tech-insights-node/src/facts.ts                |  7 +++++++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/plugins/tech-insights-backend/src/service/fact/FactRetrieverEngine.ts b/plugins/tech-insights-backend/src/service/fact/FactRetrieverEngine.ts
index a316b984ed..9078a72fba 100644
--- a/plugins/tech-insights-backend/src/service/fact/FactRetrieverEngine.ts
+++ b/plugins/tech-insights-backend/src/service/fact/FactRetrieverEngine.ts
@@ -76,6 +76,7 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
     private readonly scheduler: PluginTaskScheduler,
     private readonly defaultCadence?: string,
     private readonly defaultTimeout?: Duration,
+    private readonly defaultInitialDelay?: Duration,
   ) {}
 
   static async create(options: {
@@ -85,6 +86,7 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
     scheduler: PluginTaskScheduler;
     defaultCadence?: string;
     defaultTimeout?: Duration;
+    defaultInitialDelay?: Duration;
   }) {
     const {
       repository,
@@ -93,6 +95,7 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
       scheduler,
       defaultCadence,
       defaultTimeout,
+      defaultInitialDelay,
     } = options;
 
     const retrievers = await factRetrieverRegistry.listRetrievers();
@@ -106,6 +109,7 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
       scheduler,
       defaultCadence,
       defaultTimeout,
+      defaultInitialDelay,
     );
   }
 
@@ -115,11 +119,13 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
 
     await Promise.all(
       registrations.map(async registration => {
-        const { factRetriever, cadence, lifecycle, timeout } = registration;
+        const { factRetriever, cadence, lifecycle, timeout, initialDelay } = registration;
         const cronExpression =
           cadence || this.defaultCadence || randomDailyCron();
         const timeLimit =
           timeout || this.defaultTimeout || Duration.fromObject({ minutes: 5 });
+        const initialDelaySetting =
+          initialDelay || this.defaultInitialDelay || Duration.fromObject({ seconds: 5 });
         try {
           await this.scheduler.scheduleTask({
             id: factRetriever.id,
@@ -128,7 +134,7 @@ export class DefaultFactRetrieverEngine implements FactRetrieverEngine {
             timeout: timeLimit,
             // We add a delay in order to prevent errors due to the
             // fact that the backend is not yet online in a cold-start scenario
-            initialDelay: Duration.fromObject({ seconds: 5 }),
+            initialDelay: initialDelaySetting,
           });
           newRegs.push(factRetriever.id);
         } catch (e) {
diff --git a/plugins/tech-insights-node/src/facts.ts b/plugins/tech-insights-node/src/facts.ts
index 41db302fd5..1eb58b08ce 100644
--- a/plugins/tech-insights-node/src/facts.ts
+++ b/plugins/tech-insights-node/src/facts.ts
@@ -279,4 +279,11 @@ export type FactRetrieverRegistration = {
    * If defined this value will be used to determine expired items which will deleted when this fact retriever is run
    */
   lifecycle?: FactLifecycle;
+
+  /**
+   * A duration to determine the initial delay for the fact retriever. Useful for cold start scenarios when e.g. the
+   * catalog backend is not yet available. Defaults to 5 seconds.
+   *
+   */
+   initialDelay?: Duration | HumanDuration;
 };