refactor: mark session broken when error or close

mybigday · Sep 20, 2023 · 2d42884 · 2d42884
1 parent caf5d89
commit 2d42884
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/simple-mq.ts b/src/simple-mq.ts
@@ -101,15 +101,14 @@ export class SimpleMQBroker extends EventEmitter {
   protected handleClient(client: Client) {
     let id: string | undefined;
     let session: MQClientSession | undefined;
-    client.on('error', (err) => {
-      this.emit('error', err);
-    });
-    client.on('close', () => {
+    const connectionBroken = () => {
       if (id && session) {
         delete this.clients[id!];
         session!.alive = false;
       }
-    });
+    };
+    client.on('error', connectionBroken);
+    client.on('close', connectionBroken);
     client.on('disconnect', () => {
       client.end();
       this.emit('clientDisconnected', id);