From 6cf79ef81a9f569a0cb65bbc59bb73e517ebbb7b Mon Sep 17 00:00:00 2001
From: Michele Azzolari <michele@azzolari.it>
Date: Sat, 7 Dec 2024 17:41:21 +0100
Subject: [PATCH] fixup! fix(instrumentation-mysql2)!: do not include
 parameterized values to db.statement span attribute (#1758)

---
 .../src/instrumentation.ts                    | 14 ++++++++++---
 .../test/mysql.test.ts                        | 21 +++++++++++++++++++
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/plugins/node/opentelemetry-instrumentation-mysql2/src/instrumentation.ts b/plugins/node/opentelemetry-instrumentation-mysql2/src/instrumentation.ts
index ef5562b115..6a80fd5983 100644
--- a/plugins/node/opentelemetry-instrumentation-mysql2/src/instrumentation.ts
+++ b/plugins/node/opentelemetry-instrumentation-mysql2/src/instrumentation.ts
@@ -154,8 +154,16 @@ export class MySQL2Instrumentation extends InstrumentationBase<MySQL2Instrumenta
           }
           span.end();
         });
-
-        if (typeof arguments[arguments.length - 1] !== 'function') {
+        function findCallback(_arguments: IArguments) {
+          for (let i = 0; i < _arguments.length; i++) {
+            if (typeof _arguments[i] === 'function') {
+              return i;
+            }
+          }
+          return -1;
+        }
+        const cb = findCallback(arguments);
+        if (cb < 0) {
           if (typeof (query as any).onResult === 'function') {
             thisPlugin._wrap(
               query as any,
@@ -182,7 +190,7 @@ export class MySQL2Instrumentation extends InstrumentationBase<MySQL2Instrumenta
         } else {
           thisPlugin._wrap(
             arguments,
-            arguments.length - 1,
+            cb,
             thisPlugin._patchCallbackQuery(endSpan)
           );
         }
diff --git a/plugins/node/opentelemetry-instrumentation-mysql2/test/mysql.test.ts b/plugins/node/opentelemetry-instrumentation-mysql2/test/mysql.test.ts
index 517270f791..69736ffff4 100644
--- a/plugins/node/opentelemetry-instrumentation-mysql2/test/mysql.test.ts
+++ b/plugins/node/opentelemetry-instrumentation-mysql2/test/mysql.test.ts
@@ -722,6 +722,27 @@ describe('mysql2', () => {
       });
     });
 
+    it('should intercept pool.query(text: options, callback, null)', done => {
+      const span = provider.getTracer('default').startSpan('test span');
+      context.with(trace.setSpan(context.active(), span), () => {
+        const sql = 'SELECT 1+1 as solution';
+        // @ts-ignore
+        pool.query(
+          { sql },
+          (err, res: mysqlTypes.RowDataPacket[]) => {
+            assert.ifError(err);
+            assert.ok(res);
+            assert.strictEqual(res[0].solution, 2);
+            const spans = memoryExporter.getFinishedSpans();
+            assert.strictEqual(spans.length, 1);
+            assertSpan(spans[0], sql);
+            done();
+          },
+          null
+        );
+      });
+    });
+
     it('should intercept pool.query(text: string, values: [], callback)', done => {
       const span = provider.getTracer('default').startSpan('test span');
       context.with(trace.setSpan(context.active(), span), () => {