run delete authtoken in background (#627)

* try 3 times to create authtoken prefect input * remove returning response * increase axios timeout * remove timeout option in request options * delete authtoken in prefect in the background * add try..catch --------- Co-authored-by: suwarno ong <[email protected]>
data2evidence · Jan 16, 2025 · 6f4000f · 6f4000f
1 parent fc0ca35
commit 6f4000f
Show file tree

Hide file tree

Showing 5 changed files with 63 additions and 26 deletions.
diff --git a/functions/jobplugins/src/services/CohortService.ts b/functions/jobplugins/src/services/CohortService.ts
@@ -19,12 +19,21 @@ export class CohortService {
     );
 
     await prefectApi.createInputAuthToken(flowRunId);
-    await new Promise((resolve) => {
-      setTimeout(async () => {
-        await prefectApi.deleteInputAuthToken(flowRunId);
-        resolve(`Deleted the input of ${flowRunId}`);
-      }, 5000);
-    });
+    Promise.any([
+      new Promise(() => {
+        setTimeout(async () => {
+          const msg = "Prefect input authtoken deletion";
+          try {
+            (await prefectApi.deleteInputAuthToken(flowRunId))
+              ? console.log(`${msg} successful`)
+              : console.log(`${msg} failed`);
+          } catch (error) {
+            console.log(`${msg} failed`);
+            console.error(error);
+          }
+        }, 1000 * 60 * 5);
+      }),
+    ]);
 
     return { flowRunId };
   }

diff --git a/functions/jobplugins/src/services/DataCharacterizationService.ts b/functions/jobplugins/src/services/DataCharacterizationService.ts
@@ -160,12 +160,19 @@ export class DataCharacterizationService {
 
     await prefectApi.createInputAuthToken(flowRunId);
 
-    await Promise.any([
-      new Promise((resolve) => {
+    Promise.any([
+      new Promise(() => {
         setTimeout(async () => {
-          await prefectApi.deleteInputAuthToken(flowRunId);
-          resolve(`Deleted the input of ${flowRunId}`);
-        }, 5000);
+          const msg = "Prefect input authtoken deletion";
+          try {
+            (await prefectApi.deleteInputAuthToken(flowRunId))
+              ? console.log(`${msg} successful`)
+              : console.log(`${msg} failed`);
+          } catch (error) {
+            console.log(`${msg} failed`);
+            console.error(error);
+          }
+        }, 1000 * 60 * 5);
       }),
     ]);
 

diff --git a/functions/jobplugins/src/services/DataModelFlowService.ts b/functions/jobplugins/src/services/DataModelFlowService.ts
@@ -69,12 +69,19 @@ export class DataModelFlowService {
 
     await prefectApi.createInputAuthToken(flowRunId);
 
-    await Promise.any([
-      new Promise((resolve) => {
+    Promise.any([
+      new Promise(() => {
         setTimeout(async () => {
-          await prefectApi.deleteInputAuthToken(flowRunId);
-          resolve(`Deleted the input of ${flowRunId}`);
-        }, 5000);
+          const msg = "Prefect input authtoken deletion";
+          try {
+            (await prefectApi.deleteInputAuthToken(flowRunId))
+              ? console.log(`${msg} successful`)
+              : console.log(`${msg} failed`);
+          } catch (error) {
+            console.log(`${msg} failed`);
+            console.error(error);
+          }
+        }, 1000 * 60 * 5);
       }),
     ]);
 

diff --git a/functions/jobplugins/src/services/DbSvcService.ts b/functions/jobplugins/src/services/DbSvcService.ts
@@ -53,12 +53,19 @@ export class DbSvcService {
     );
 
     await prefectApi.createInputAuthToken(flowrunId);
-    await Promise.any([
-      new Promise((resolve) => {
+    Promise.any([
+      new Promise(() => {
         setTimeout(async () => {
-          await prefectApi.deleteInputAuthToken(flowrunId);
-          resolve(`Deleted the input of ${flowrunId}`);
-        }, 5000);
+          const msg = "Prefect input authtoken deletion";
+          try {
+            (await prefectApi.deleteInputAuthToken(flowrunId))
+              ? console.log(`${msg} successful`)
+              : console.log(`${msg} failed`);
+          } catch (error) {
+            console.log(`${msg} failed`);
+            console.error(error);
+          }
+        }, 1000 * 60 * 5);
       }),
     ]);
 

diff --git a/functions/jobplugins/src/services/DqdService.ts b/functions/jobplugins/src/services/DqdService.ts
@@ -154,12 +154,19 @@ export class DqdService {
 
     await prefectApi.createInputAuthToken(flowRunId);
 
-    await Promise.any([
-      new Promise((resolve) => {
+    Promise.any([
+      new Promise(() => {
         setTimeout(async () => {
-          await prefectApi.deleteInputAuthToken(flowRunId);
-          resolve(`Deleted the input of ${flowRunId}`);
-        }, 5000);
+          const msg = "Prefect input authtoken deletion";
+          try {
+            (await prefectApi.deleteInputAuthToken(flowRunId))
+              ? console.log(`${msg} successful`)
+              : console.log(`${msg} failed`);
+          } catch (error) {
+            console.log(`${msg} failed`);
+            console.error(error);
+          }
+        }, 1000 * 60 * 5);
       }),
     ]);