Fix async context loss in WebSocketBus

2022-08-10 00:06:48 -05:00
parent 085fe04f90
commit 9b47d2ac99
2 changed files with 5 additions and 3 deletions
--- a/src/support/bus/WebSocketBus.ts
+++ b/src/support/bus/WebSocketBus.ts
@@ -154,8 +154,10 @@ export class WebSocketBus implements EventBus, AwareOfContainerLifecycle {
            })
        })

-        this.ws.on('close', () => this.bus.push(new WebSocketCloseEvent()))
-        this.ws.on('error', () => this.bus.push(new WebSocketCloseEvent()))
+        this.ws.on('close', () =>
+            resource.runInAsyncScope(() => this.bus.push(new WebSocketCloseEvent())))
+        this.ws.on('error', () =>
+            resource.runInAsyncScope(() => this.bus.push(new WebSocketCloseEvent())))

        await this.registerHealthCheck()