chore(NODE-5771): benchmark new connection

Author: nbbeeken

modern.cjs

@@ -0,0 +1,42 @@
+const totalStart = performance.now();
+/* eslint-disable no-console */
+/* eslint-disable @typescript-eslint/no-var-requires */
+const process = require('node:process');
+const { MongoClient } = require('./lib/index.js');
+const { ModernConnection } = require('./lib/cmap/connection.js');
+
+const tweet = require('./test/benchmarks/driverBench/spec/single_and_multi_document/tweet.json');
+
+const client = new MongoClient(process.env.MONGODB_URI, { connectionType: ModernConnection });
+
+async function main() {
+  console.log('modern connection');
+
+  const db = client.db('test');
+  let collection = db.collection('test');
+  await collection.drop().catch(() => null);
+  collection = await db.createCollection('test');
+  await collection.insertOne(tweet);
+
+  const total = 10_000;
+
+  for (let i = 0; i < total; i++) {
+    await collection.findOne();
+  }
+
+  const start = performance.now() - totalStart;
+  for (let i = 0; i < total; i++) {
+    await collection.findOne();
+  }
+  const end = performance.now() - totalStart;
+
+  console.log(
+    `end - start = ms time for 10k findOne calls (script boot: ${totalStart.toFixed(3)})`
+  );
+  console.log(`${end.toFixed(3)} - ${start.toFixed(3)} = ${(end - start).toFixed(4)}`);
+  console.log(`avg findOne: ${((end - start) / total).toFixed(3)} ms`);
+
+  await client.close();
+}
+
+main().catch(console.error);

old.cjs

@@ -0,0 +1,42 @@
+const totalStart = performance.now();
+/* eslint-disable no-console */
+/* eslint-disable @typescript-eslint/no-var-requires */
+const process = require('node:process');
+const { MongoClient } = require('./lib/index.js');
+const { Connection } = require('./lib/cmap/connection.js');
+
+const tweet = require('./test/benchmarks/driverBench/spec/single_and_multi_document/tweet.json');
+
+const client = new MongoClient(process.env.MONGODB_URI, { connectionType: Connection });
+
+async function main() {
+  console.log('old connection');
+
+  const db = client.db('test');
+  let collection = db.collection('test');
+  await collection.drop().catch(() => null);
+  collection = await db.createCollection('test');
+  await collection.insertOne(tweet);
+
+  const total = 10_000;
+
+  for (let i = 0; i < total; i++) {
+    await collection.findOne();
+  }
+
+  const start = performance.now() - totalStart;
+  for (let i = 0; i < total; i++) {
+    await collection.findOne();
+  }
+  const end = performance.now() - totalStart;
+
+  console.log(
+    `end - start = ms time for 10k findOne calls (script boot: ${totalStart.toFixed(3)})`
+  );
+  console.log(`${end.toFixed(3)} - ${start.toFixed(3)} = ${(end - start).toFixed(4)}`);
+  console.log(`avg findOne: ${((end - start) / total).toFixed(3)} ms`);
+
+  await client.close();
+}
+
+main().catch(console.error);

test/benchmarks/driverBench/common.js

@@ -6,6 +6,8 @@ const { Readable } = require('stream');
 const { pipeline } = require('stream/promises');
 const { MongoClient } = require('../../..');
 const { GridFSBucket } = require('../../..');
+// eslint-disable-next-line no-restricted-modules
+const { ModernConnection, Connection } = require('../../../lib/cmap/connection');
 
 // eslint-disable-next-line no-restricted-modules
 const { MONGODB_ERROR_CODES } = require('../../../lib/error');
@@ -25,7 +27,9 @@ function loadSpecString(filePath) {
 }
 
 function makeClient() {
-  this.client = new MongoClient(process.env.MONGODB_URI || 'mongodb://localhost:27017');
+  this.client = new MongoClient(process.env.MONGODB_URI || 'mongodb://127.0.0.1:27017', {
+    connectionType: Connection
+  });
 }
 
 function connectClient() {

test/benchmarks/driverBench/index.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const MongoBench = require('../mongoBench');
+const os = require('node:os');
 
 const Runner = MongoBench.Runner;
 
@@ -11,6 +12,21 @@ const { inspect } = require('util');
 const { writeFile } = require('fs/promises');
 const { makeParallelBenchmarks, makeSingleBench, makeMultiBench } = require('../mongoBench/suites');
 
+const hw = os.cpus();
+const ram = os.totalmem() / 1024 ** 3;
+const platform = { name: hw[0].model, cores: hw.length, ram: `${ram}GB` };
+
+const systemInfo = () =>
+  [
+    `Connection`,
+    `\n- cpu: ${platform.name}`,
+    `- cores: ${platform.cores}`,
+    `- arch: ${os.arch()}`,
+    `- os: ${process.platform} (${os.release()})`,
+    `- ram: ${platform.ram}`
+  ].join('\n');
+console.log(systemInfo());
+
 function average(arr) {
   return arr.reduce((x, y) => x + y, 0) / arr.length;
 }