fix: use context API to bind connection checkOut callback (open-telem…

…etry#1766) * fix: use context API to bind connection checkOut callback * chore: fix lint --------- Co-authored-by: Amir Blum <[email protected]>
jmcdo29 · Nov 13, 2023 · 02c7cae · 02c7cae
1 parent 6186b8b
commit 02c7cae
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/plugins/node/opentelemetry-instrumentation-mongodb/src/instrumentation.ts b/plugins/node/opentelemetry-instrumentation-mongodb/src/instrumentation.ts
@@ -13,8 +13,6 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { AsyncResource } from 'async_hooks';
-
 import {
   context,
   diag,
@@ -332,7 +330,7 @@ export class MongoDBInstrumentation extends InstrumentationBase {
   private _getV4ConnectionPoolCheckOut() {
     return (original: V4ConnectionPool['checkOut']) => {
       return function patchedCheckout(this: unknown, callback: any) {
-        const patchedCallback = AsyncResource.bind(callback);
+        const patchedCallback = context.bind(context.active(), callback);
         return original.call(this, patchedCallback);
       };
     };