Merge pull request nasa-gcn#2364 from lpsinger/sigint-sigterm

Close Kafka connection on SIGINT or SIGTERM, not beforeExit
Vidushi-GitHub · Jun 12, 2024 · c94e2d6 · c94e2d6
2 parents 6a1113d + cba9b5c
commit c94e2d6
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/app/lib/kafka.server.ts b/app/lib/kafka.server.ts
@@ -51,11 +51,13 @@ if (process.env.ARC_SANDBOX) {
     async () => {
       const producer = kafka.producer()
       await producer.connect()
-      process.once('beforeExit', async () => {
-        console.log('Disconnecting from Kafka')
-        await producer.disconnect()
-        console.log('Disconnected from Kafka')
-      })
+      ;['SIGINT', 'SIGTERM'].forEach((event) =>
+        process.once(event, async () => {
+          console.log('Disconnecting from Kafka')
+          await producer.disconnect()
+          console.log('Disconnected from Kafka')
+        })
+      )
       return producer
     },
     { promise: true }