feat(mongodb): support v5

plugins/node/opentelemetry-instrumentation-mongodb/.tav.yml

@@ -1,9 +1,11 @@
 mongodb:
   jobs:
     - versions: ">=3.3 <4"
-      commands: npm run test
+      commands: npm run test-v3
     - versions: ">=4 <5"
-      commands: npm run test-new-versions
+      commands: npm run test-v4
+    - versions: ">=5 <6"
+      commands: npm run test-v5
 
   # Fix missing `contrib-test-utils` package
   pretest: npm run --prefix ../../../ lerna:link

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

@@ -7,8 +7,10 @@
   "repository": "open-telemetry/opentelemetry-js-contrib",
   "scripts": {
     "docker:start": "docker run -e MONGODB_DB=opentelemetry-tests -e MONGODB_PORT=27017 -e MONGODB_HOST=127.0.0.1 -p 27017:27017 --rm mongo",
-    "test": "nyc ts-mocha -p tsconfig.json --require '@opentelemetry/contrib-test-utils' 'test/**/mongodb-v3.test.ts'",
-    "test-new-versions": "nyc ts-mocha -p tsconfig.json --require '@opentelemetry/contrib-test-utils' 'test/mongodb-v4**.test.ts'",
+    "test": "npm run test-v3",
+    "test-v3": "nyc ts-mocha -p tsconfig.json --require '@opentelemetry/contrib-test-utils' 'test/**/mongodb-v3.test.ts'",
+    "test-v4": "nyc ts-mocha -p tsconfig.json --require '@opentelemetry/contrib-test-utils' 'test/mongodb-v4-v5.metrics.test.ts' 'test/**/mongodb-v4.test.ts'",
+    "test-v5": "nyc ts-mocha -p tsconfig.json --require '@opentelemetry/contrib-test-utils' 'test/mongodb-v4-v5.metrics.test.ts' 'test/**/mongodb-v5.test.ts'",
     "test-all-versions": "tav",
     "tdd": "npm run test -- --watch-extensions ts --watch",
     "clean": "rimraf build/*",
@@ -51,16 +53,16 @@
   },
   "devDependencies": {
     "@opentelemetry/api": "^1.3.0",
-    "@opentelemetry/contrib-test-utils": "^0.33.3",
     "@opentelemetry/context-async-hooks": "^1.8.0",
+    "@opentelemetry/contrib-test-utils": "^0.33.3",
     "@opentelemetry/sdk-trace-base": "^1.8.0",
     "@opentelemetry/sdk-trace-node": "^1.8.0",
+    "@types/bson": "4.0.5",
     "@types/mocha": "7.0.2",
+    "@types/mongodb": "3.6.20",
     "@types/node": "18.11.7",
     "mocha": "7.2.0",
-    "mongodb": "3.6.11",
-    "@types/mongodb": "3.6.20",
-    "@types/bson": "4.0.5",
+    "mongodb": "4",
     "nyc": "15.1.0",
     "rimraf": "5.0.0",
     "test-all-versions": "5.0.1",
@@ -73,4 +75,4 @@
     "@opentelemetry/semantic-conventions": "^1.0.0"
   },
   "homepage": "https://github.com/open-telemetry/opentelemetry-js-contrib/tree/main/plugins/node/opentelemetry-instrumentation-mongodb#readme"
-}
+}

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

@@ -95,25 +95,25 @@ export class MongoDBInstrumentation extends InstrumentationBase {
       ),
       new InstrumentationNodeModuleDefinition<any>(
         'mongodb',
-        ['4.*'],
+        ['4.*', '5.*'],
         undefined,
         undefined,
         [
           new InstrumentationNodeModuleFile<V4Connection>(
             'mongodb/lib/cmap/connection.js',
-            ['4.*'],
+            ['4.*', '5.*'],
             v4PatchConnection,
             v4UnpatchConnection
           ),
           new InstrumentationNodeModuleFile<V4Connect>(
             'mongodb/lib/cmap/connect.js',
-            ['4.*'],
+            ['4.*', '5.*'],
             v4PatchConnect,
             v4UnpatchConnect
           ),
           new InstrumentationNodeModuleFile<V4Session>(
             'mongodb/lib/sessions.js',
-            ['4.*'],
+            ['4.*', '5.*'],
             v4PatchSessions,
             v4UnpatchSessions
           ),

plugins/node/opentelemetry-instrumentation-mongodb/test/mongodb-v3.test.ts

@@ -35,7 +35,7 @@ import * as mongodb from 'mongodb';
 import { assertSpans, accessCollection, DEFAULT_MONGO_HOST } from './utils';
 import { SemanticAttributes } from '@opentelemetry/semantic-conventions';
 
-describe('MongoDBInstrumentation', () => {
+describe('MongoDBInstrumentation-Tracing-v3', () => {
   function create(config: MongoDBInstrumentationConfig = {}) {
     instrumentation.setConfig(config);
   }

plugins/node/opentelemetry-instrumentation-mongodb/test/mongodb-v4.metrics.test.ts → plugins/node/opentelemetry-instrumentation-mongodb/test/mongodb-v4-v5.metrics.test.ts

@@ -28,8 +28,6 @@ import {
   ResourceMetrics,
 } from '@opentelemetry/sdk-metrics';
 
-import * as mongodb from 'mongodb';
-
 const otelTestingMeterProvider = new MeterProvider();
 const inMemoryMetricsExporter = new InMemoryMetricExporter(
   AggregationTemporality.CUMULATIVE
@@ -40,17 +38,13 @@ const metricReader = new PeriodicExportingMetricReader({
   exportTimeoutMillis: 100,
 });
 
-otelTestingMeterProvider.addMetricReader(metricReader);
-
 import { registerInstrumentationTesting } from '@opentelemetry/contrib-test-utils';
 const instrumentation = registerInstrumentationTesting(
   new MongoDBInstrumentation()
 );
 
-instrumentation.setMeterProvider(otelTestingMeterProvider);
-
 import { accessCollection, DEFAULT_MONGO_HOST } from './utils';
-
+import * as mongodb from 'mongodb';
 import * as assert from 'assert';
 
 async function waitForNumberOfExports(
@@ -86,28 +80,33 @@ describe('MongoDBInstrumentation-Metrics', () => {
   const DB_NAME = process.env.MONGODB_DB || 'opentelemetry-tests-metrics';
   const COLLECTION_NAME = 'test-metrics';
   const URL = `mongodb://${HOST}:${PORT}/${DB_NAME}`;
-
   let client: mongodb.MongoClient;
-  let collection: mongodb.Collection;
+
+  before(() => {
+    otelTestingMeterProvider.addMetricReader(metricReader);
+    instrumentation?.setMeterProvider(otelTestingMeterProvider);
+  });
+
   beforeEach(function mongoBeforeEach(done) {
     // Skipping all tests in beforeEach() is a workaround. Mocha does not work
     // properly when skipping tests in before() on nested describe() calls.
     // https://github.com/mochajs/mocha/issues/2819
     if (!shouldTest) {
       this.skip();
     }
+
     inMemoryMetricsExporter.reset();
     done();
   });
 
   it('Should add connection usage metrics', async () => {
     const result = await accessCollection(URL, DB_NAME, COLLECTION_NAME);
     client = result.client;
-    collection = result.collection;
+    const collection = result.collection;
     const insertData = [{ a: 1 }, { a: 2 }, { a: 3 }];
     await collection.insertMany(insertData);
     await collection.deleteMany({});
-    let exportedMetrics = await waitForNumberOfExports(
+    const exportedMetrics = await waitForNumberOfExports(
       inMemoryMetricsExporter,
       1
     );
@@ -126,43 +125,55 @@ describe('MongoDBInstrumentation-Metrics', () => {
       metrics[0].descriptor.name,
       'db.client.connections.usage'
     );
-    assert.strictEqual(metrics[0].dataPoints.length, 2);
-    assert.strictEqual(metrics[0].dataPoints[0].value, 0);
-    assert.strictEqual(metrics[0].dataPoints[0].attributes['state'], 'used');
+
+    // Checking dataPoints
+    const dataPoints = metrics[0].dataPoints;
+    assert.strictEqual(dataPoints.length, 2);
+    assert.strictEqual(dataPoints[0].value, 0);
+    assert.strictEqual(dataPoints[0].attributes['state'], 'used');
     assert.strictEqual(
-      metrics[0].dataPoints[0].attributes['pool.name'],
+      dataPoints[0].attributes['pool.name'],
       `mongodb://${HOST}:${PORT}/${DB_NAME}`
     );
 
-    assert.strictEqual(metrics[0].dataPoints[1].value, 1);
-    assert.strictEqual(metrics[0].dataPoints[1].attributes['state'], 'idle');
+    assert.strictEqual(dataPoints[1].value, 1);
+    assert.strictEqual(dataPoints[1].attributes['state'], 'idle');
     assert.strictEqual(
-      metrics[0].dataPoints[1].attributes['pool.name'],
+      dataPoints[1].attributes['pool.name'],
       `mongodb://${HOST}:${PORT}/${DB_NAME}`
     );
+  });
+
+  it('Should add disconnection usage metrics', async () => {
     await client.close();
 
-    exportedMetrics = await waitForNumberOfExports(inMemoryMetricsExporter, 2);
+    const exportedMetrics = await waitForNumberOfExports(
+      inMemoryMetricsExporter,
+      2
+    );
     assert.strictEqual(exportedMetrics.length, 2);
-    metrics = exportedMetrics[1].scopeMetrics[0].metrics;
+    const metrics = exportedMetrics[1].scopeMetrics[0].metrics;
     assert.strictEqual(metrics.length, 1);
     assert.strictEqual(metrics[0].dataPointType, DataPointType.SUM);
 
     assert.strictEqual(
       metrics[0].descriptor.description,
       'The number of connections that are currently in state described by the state attribute.'
     );
-    assert.strictEqual(metrics[0].dataPoints.length, 2);
-    assert.strictEqual(metrics[0].dataPoints[0].value, 0);
-    assert.strictEqual(metrics[0].dataPoints[0].attributes['state'], 'used');
+
+    // Checking dataPoints
+    const dataPoints = metrics[0].dataPoints;
+    assert.strictEqual(dataPoints.length, 2);
+    assert.strictEqual(dataPoints[0].value, 0);
+    assert.strictEqual(dataPoints[0].attributes['state'], 'used');
     assert.strictEqual(
-      metrics[0].dataPoints[0].attributes['pool.name'],
+      dataPoints[0].attributes['pool.name'],
       `mongodb://${HOST}:${PORT}/${DB_NAME}`
     );
-    assert.strictEqual(metrics[0].dataPoints[1].value, 0);
-    assert.strictEqual(metrics[0].dataPoints[1].attributes['state'], 'idle');
+    assert.strictEqual(dataPoints[1].value, 0);
+    assert.strictEqual(dataPoints[1].attributes['state'], 'idle');
     assert.strictEqual(
-      metrics[0].dataPoints[1].attributes['pool.name'],
+      dataPoints[1].attributes['pool.name'],
       `mongodb://${HOST}:${PORT}/${DB_NAME}`
     );
   });