fix: Updated cassandra-driver instrumentation to properly trace promi…

…se based executions (#2351)
newrelic · Jul 17, 2024 · bab9a8b · bab9a8b
1 parent d212b15
commit bab9a8b
Show file tree

Hide file tree

Showing 2 changed files with 208 additions and 136 deletions.
diff --git a/lib/instrumentation/cassandra-driver.js b/lib/instrumentation/cassandra-driver.js
@@ -29,7 +29,7 @@ module.exports = function initialize(_agent, cassandra, _moduleName, shim) {
     ClientProto,
     ['connect', 'shutdown'],
     function operationSpec(shim, _fn, name) {
-      return new OperationSpec({ callback: shim.LAST, name })
+      return new OperationSpec({ callback: shim.LAST, name, promise: true })
     }
   )
 
@@ -39,7 +39,8 @@ module.exports = function initialize(_agent, cassandra, _moduleName, shim) {
       '_execute',
       new QuerySpec({
         query: shim.FIRST,
-        callback: shim.LAST
+        callback: shim.LAST,
+        promise: true
       })
     )
 
@@ -64,7 +65,8 @@ module.exports = function initialize(_agent, cassandra, _moduleName, shim) {
       '_innerExecute',
       new QuerySpec({
         query: shim.FIRST,
-        callback: shim.LAST
+        callback: shim.LAST,
+        promise: true
       })
     )
 
@@ -109,7 +111,8 @@ module.exports = function initialize(_agent, cassandra, _moduleName, shim) {
     'batch',
     new QuerySpec({
       query: findBatchQueryArg,
-      callback: shim.LAST
+      callback: shim.LAST,
+      promise: true
     })
   )
 }