Squashing client close errors

KurtzL · Jun 22, 2023 · 35076bb · 35076bb
1 parent bf6c9fd
commit 35076bb
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/lib/utils/client.util.ts b/lib/utils/client.util.ts
@@ -1,12 +1,20 @@
-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 () =>
+      client.connection
+        ?.close()
+        .catch((reason) =>
+          console.error(
+            `Temporal client connection was not cleanly closed: ${reason}`,
+          ),
+        );
   return client;
 }
 
 export function getWorkflowClient(options?: WorkflowClientOptions) {
   const client = new WorkflowClient(options);
   return assignOnAppShutdownHook(client);
-}
+}