fix(mysql): fixing callback wrapping in some case, adding debug line (#…

…55)
epsagon · Feb 17, 2019 · 1be691f · 1be691f
1 parent 84b06cc
commit 1be691f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/events/mysql.js b/src/events/mysql.js
@@ -35,10 +35,10 @@ function mysqlQueryWrapper(wrappedFunction) {
             this.config,
             'mysql'
         );
-        if (typeof sql === 'string') {
-            callback = patchedCallback;
-        } else {
+        if (sql.onResult) {
             sql.onResult = patchedCallback; // eslint-disable-line
+        } else {
+            callback = patchedCallback;
         }
         return wrappedFunction.apply(this, [sql, params, callback]);
     };

diff --git a/src/events/sql.js b/src/events/sql.js
@@ -84,6 +84,7 @@ module.exports.wrapSqlQuery = function wrapSqlQuery(queryString, params, callbac
 
         const responsePromise = new Promise((resolve) => {
             patchedCallback = (err, res, fields) => {
+                utils.debugLog('SQL Patched callback was called.');
                 dbapiEvent.setDuration(utils.createDurationTimestamp(startTime));
 
                 if (err) {