Squashing client close errors

KurtzL · Jun 21, 2023 · be6057d · be6057d
1 parent bf6c9fd
commit be6057d
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/lib/utils/client.util.ts b/lib/utils/client.util.ts
@@ -1,12 +1,21 @@
-import { OnApplicationShutdown } from "@nestjs/common";
-import { WorkflowClient, WorkflowClientOptions } from "@temporalio/client";
+import { OnApplicationShutdown } from '@nestjs/common';
+import { WorkflowClient, WorkflowClientOptions } from '@temporalio/client';
 
 export function assignOnAppShutdownHook(client: WorkflowClient) {
-  (client as unknown as OnApplicationShutdown).onApplicationShutdown = client.connection.close;
+  (client as unknown as OnApplicationShutdown).onApplicationShutdown =
+    async () => {
+      try {
+        await client.connection?.close();
+      } catch (e) {
+        console.error(
+          `Temporal client connection was not cleanly closed: ${e.stack}`,
+        );
+      }
+    };
   return client;
 }
 
 export function getWorkflowClient(options?: WorkflowClientOptions) {
   const client = new WorkflowClient(options);
   return assignOnAppShutdownHook(client);
-}
+}