Merge pull request #703 from iwaiktos/fix-fastify-plugin

Fix (aws-xray-sdk-fastify): enable automatic mode and expose hooks

Author: mxiamxia

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();
+  }
 };

sdk_contrib/fastify/lib/plugin.js

@@ -1,19 +1,20 @@
 // @ts-check
+const fp = require('fastify-plugin').default;
 const configureAWSXRaySync = require('./private/configure-aws-x-ray-sync');
 const onRequestHook = require('./hooks/on-request.hook');
 const onResponseHook = require('./hooks/on-response.hook');
 const onErrorHook = require('./hooks/on-error.hook');
 
 /** @type {import('fastify').FastifyPluginAsync} */
-const xRayFastifyPlugin = async (fastify, opts) => {
+const xRayFastifyPlugin = fp(async (fastify, opts) => {
   configureAWSXRaySync(fastify, opts);
 
   fastify.decorateRequest('segment', null);
   fastify
     .addHook('onRequest', onRequestHook)
     .addHook('onResponse', onResponseHook)
     .addHook('onError', onErrorHook);
-};
+});
 
 module.exports = xRayFastifyPlugin;