fix(instr-mysql2): Fix mysql2 instrumentation connection prototype

plugins/node/opentelemetry-instrumentation-mysql2/package.json

@@ -50,7 +50,7 @@
     "@types/mocha": "7.0.2",
     "@types/node": "18.18.14",
     "@types/semver": "7.5.8",
-    "mysql2": "3.11.3",
+    "mysql2": "3.11.5",
     "nyc": "15.1.0",
     "rimraf": "5.0.10",
     "semver": "7.6.3",

plugins/node/opentelemetry-instrumentation-mysql2/src/instrumentation.ts

@@ -55,8 +55,9 @@
         'mysql2',
         ['>=1.4.2 <4'],
         (moduleExports: any) => {
-          const ConnectionPrototype: mysqlTypes.Connection =
-            moduleExports.Connection.prototype;
+          const ConnectionPrototype =
+            this._getConnectionPrototype(moduleExports);
+
           if (isWrapped(ConnectionPrototype.query)) {
             this._unwrap(ConnectionPrototype, 'query');
           }
@@ -79,8 +80,8 @@
         },
         (moduleExports: any) => {
           if (moduleExports === undefined) return;
-          const ConnectionPrototype: mysqlTypes.Connection =
-            moduleExports.Connection.prototype;
+          const ConnectionPrototype =
+            this._getConnectionPrototype(moduleExports);
           this._unwrap(ConnectionPrototype, 'query');
           this._unwrap(ConnectionPrototype, 'execute');
         }
@@ -210,4 +211,16 @@
       };
     };
   }
+
+  private _getConnectionPrototype(moduleExports: any): mysqlTypes.Connection {
+    const baseClass = Object.getPrototypeOf(moduleExports.Connection);
+
+    // return the base class prototype if it has the method
+    // we need to patch ([email protected]+)
+    if (typeof baseClass.prototype?.query === 'function') {
+      return baseClass.prototype;
+    }
+
+    return moduleExports.Connection.prototype;
+  }
 }