Merge pull request #2 from jeromemacias/fastify_29

Fix Fastify < 2.10.0 compatibility
jeromemacias · Dec 6, 2020 · 297f57e · 297f57e
2 parents fe46807 + 6186100
commit 297f57e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -5,6 +5,13 @@ node_js:
   - "lts/*"
   - "node"
 
+env:
+  - FASTIFY_VERSION=^2.0.0
+  - FASTIFY_VERSION=2.9.0
+
+before_script:
+  - npm i --no-save fastify@$FASTIFY_VERSION
+
 script:
   - npm run lint-ci
   - npm run test-ci

diff --git a/plugin.js b/plugin.js
@@ -77,11 +77,13 @@ function plugin (fastify, options, next) {
       return done()
     }
 
-    if (reply.statusCode === 429) {
+    const statusCode = reply.statusCode || reply.res.statusCode
+
+    if (statusCode === 429) {
       segment.addThrottleFlag()
     }
 
-    const cause = AWSXRay.utils.getCauseTypeFromHttpStatus(reply.statusCode)
+    const cause = AWSXRay.utils.getCauseTypeFromHttpStatus(statusCode)
     if (cause) {
       segment[cause] = true
     }