Call done() at inside of the active namespace context, fixes #701

aws · Dec 18, 2024 · 6aa0de0 · 6aa0de0
1 parent 3b89008
commit 6aa0de0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sdk_contrib/fastify/lib/hooks/on-request.hook.js b/sdk_contrib/fastify/lib/hooks/on-request.hook.js
@@ -16,6 +16,7 @@ module.exports = function onRequestHook(request, reply, done) {
 
     ns.run(() => {
       AWSXRay.setSegment(segment);
+      done();
     });
   } else {
     request.log.info('Manual mode, skipping segment');
@@ -25,7 +26,7 @@ module.exports = function onRequestHook(request, reply, done) {
     } else {
       request.log.warn('Request already has a segment, skipping');
     }
-  }
 
-  done();
+    done();
+  }
 };