fix: wrapped route handlers were forced to be async or return a promi…

…se (#62)
autotelic · Feb 6, 2024 · d5f59ef · d5f59ef
1 parent a12cf85
commit d5f59ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/index.js b/index.js
@@ -141,7 +141,7 @@ async function openTelemetryPlugin (fastify, opts = {}) {
 
   if (wrapRoutes) {
     function wrapRoute (routeHandler) {
-      return async function (request, ...args) {
+      return function (request, ...args) {
         const reqContext = getContext(request)
         return context.with(reqContext, routeHandler.bind(this, request, ...args))
       }

diff --git a/test/index.test.js b/test/index.test.js
@@ -199,10 +199,12 @@ test('should wrap all routes when wrapRoutes is true', async ({ equal, same, tea
 
   await fastify.register(openTelemetryPlugin, { wrapRoutes: true })
 
-  async function testHandlerOne (request, reply) {
+  function testHandlerOne (request, reply) {
     request.openTelemetry()
     reply.headers({ one: 'ok' })
-    return { body: 'one' }
+    setTimeout(() => {
+      reply.send({ body: 'one' })
+    }, 0)
   }
 
   async function testHandlerTwo (request, reply) {