commit 2437e2a9ab896c79f4796fa644214e53c67725e0
author kezhenxu94 <kezhenxu94@163.com> Thu Jun 25 11:50:49 2020 +0800
committer kezhenxu94 <kezhenxu94@163.com> Thu Jun 25 11:50:49 2020 +0800
tree 23caad261a2995832a2aae5584d925d4466f08e8
parent 3caff640dbfb8de9fdc29330189a38752bbb1099

Delete the context when an event loop is done

src/trace/context/ContextManager.ts

diff --git a/src/trace/context/ContextManager.ts b/src/trace/context/ContextManager.ts
index 3360031..e49bbb0 100644
--- a/src/trace/context/ContextManager.ts
+++ b/src/trace/context/ContextManager.ts
@@ -19,11 +19,19 @@
 
 import Context from '../../trace/context/Context';
 import SpanContext from '../../trace/context/SpanContext';
-import { executionAsyncId } from 'async_hooks';
+import { executionAsyncId, createHook } from 'async_hooks';
 
 class ContextManager {
   contextKeyedByAsyncId: { [asyncId: number]: Context } = {};
 
+  constructor() {
+    createHook({
+      destroy: (asyncId: number) => {
+        delete this.contextKeyedByAsyncId[asyncId];
+      },
+    }).enable();
+  }
+
   get current(): Context {
     const thisAsyncId = executionAsyncId();