Update to reflect changes to the plugin API

googleapis · Feb 14, 2017 · d94d518 · d94d518
1 parent 9df68f7
commit d94d518
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/plugins/plugin-redis.js b/src/plugins/plugin-redis.js
@@ -79,7 +79,7 @@ function startSpanFromArguments(api, cmd, args, cb, send_command) {
 
 function internalSendCommandWrap(api, internal_send_command) {
   return function internal_send_command_trace(cmd, args, cb) {
-    var root = api.getTransaction();
+    var root = api.getRootSpan();
     if (!root) {
       return internal_send_command.call(this, cmd, args, cb);
     }
@@ -94,8 +94,8 @@ function internalSendCommandWrap(api, internal_send_command) {
 
 function sendCommandWrap(api, send_command) {
   return function send_command_trace(cmd, args, cb) {
-    var transaction = api.getTransaction();
-    if (!transaction) {
+    var root = api.getRootSpan();
+    if (!root) {
       return send_command.call(this, cmd, args, cb);
     }
     return startSpanFromArguments(api, cmd, args, cb, send_command.bind(this));