Merge pull request newrelic#175 from newrelic/fix-paramware

jsumners-nr · Mar 22, 2024 · b2068d2 · b2068d2
2 parents 60d5f1a + 168fdf5
commit b2068d2
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/lib/instrumentation/koa/lib/router-instrumentation.js b/lib/instrumentation/koa/lib/router-instrumentation.js
@@ -27,15 +27,18 @@ module.exports = function instrumentRouter(shim, Router) {
 }
 
 function wrapParamware(shim, paramware, fnName, route) {
-  return shim.recordParamware(paramware, {
-    name: route,
-    next: shim.LAST,
-    promise: true,
-    appendPath: false,
-    req: function getReq(shim, fn, _fnName, args) {
-      return args[1] && args[1].req
-    }
-  })
+  return shim.recordParamware(
+    paramware,
+    new shim.specs.MiddlewareSpec({
+      name: route,
+      next: shim.LAST,
+      promise: true,
+      appendPath: false,
+      req: function getReq(shim, fn, _fnName, args) {
+        return args[1] && args[1].req
+      }
+    })
+  )
 }
 
 function wrapMiddleware(shim, fn, name, layer) {