chore(NODE-6721): migrate singlebench tests

Author: nbbeeken

test/benchmarks/driver_bench/src/driver.mts

@@ -135,6 +135,9 @@ export function metrics(test_name: string, result: number, count: number) {
  * For use in setup/teardown mostly.
  */
 export class DriverTester {
+  readonly DB_NAME = DB_NAME;
+  readonly COLLECTION_NAME = COLLECTION_NAME;
+
   public client: mongodb.MongoClient;
   constructor() {
     this.client = new MongoClient(MONGODB_URI, MONGODB_CLIENT_OPTIONS);

test/benchmarks/driver_bench/src/main.mts

@@ -28,15 +28,6 @@ export const alphabetically = (a: unknown, b: unknown) => {
   return res < 0 ? -1 : res > 0 ? 1 : 0;
 };
 
-export const alphabetically = (a: unknown, b: unknown) => {
-  const res = `${a}`.localeCompare(`${b}`, 'en-US', {
-    usage: 'sort',
-    numeric: true,
-    ignorePunctuation: false
-  });
-  return res < 0 ? -1 : res > 0 ? 1 : 0;
-};
-
 /** Find every mjs file in the suites folder */
 async function getBenchmarks(): Promise<{
   tests: Record<string, Record<string, string>>;

test/benchmarks/driver_bench/src/suites/single_bench/find_one.mts

@@ -0,0 +1,26 @@
+import { driver, type mongodb } from '../../driver.mjs';
+
+export const taskSize = 16.22;
+
+let collection: mongodb.Collection<{ _id: number }>;
+
+export async function before() {
+  await driver.drop();
+  await driver.create();
+
+  const tweet = await driver.load('single_and_multi_document/tweet.json', 'json');
+  await driver.insertManyOf(tweet, 10000, true);
+
+  collection = driver.client.db(driver.DB_NAME).collection(driver.COLLECTION_NAME);
+}
+
+export async function run() {
+  for (let _id = 0; _id < 10000; ++_id) {
+    await collection.findOne({ _id });
+  }
+}
+
+export async function after() {
+  await driver.drop();
+  await driver.close();
+}

test/benchmarks/driver_bench/src/suites/single_bench/large_doc_insert_one.mts

@@ -0,0 +1,32 @@
+import { driver, type mongodb } from '../../driver.mjs';
+
+export const taskSize = 27.31;
+
+let collection: mongodb.Collection;
+let documents: Record<string, any>[];
+let largeDoc: Record<string, any>;
+
+export async function before() {
+  largeDoc = await driver.load('single_and_multi_document/large_doc.json', 'json');
+}
+
+export async function beforeEach() {
+  await driver.drop();
+  await driver.create();
+
+  // Make new "documents" so the _id field is not carried over from the last run
+  documents = Array.from({ length: 10 }, () => ({ ...largeDoc }));
+
+  collection = driver.collection;
+}
+
+export async function run() {
+  for (const doc of documents) {
+    await collection.insertOne(doc);
+  }
+}
+
+export async function after() {
+  await driver.drop();
+  await driver.close();
+}

test/benchmarks/driver_bench/src/suites/single_bench/ping.mts

@@ -0,0 +1,24 @@
+import { driver, type mongodb } from '../../driver.mjs';
+
+// { ping: 1 } is 15 bytes of BSON x 10,000 iterations
+export const taskSize = 0.15;
+
+let db: mongodb.Db;
+
+export async function before() {
+  await driver.drop();
+  await driver.create();
+
+  db = driver.db;
+}
+
+export async function run() {
+  for (let i = 0; i < 10000; ++i) {
+    await db.command({ ping: 1 });
+  }
+}
+
+export async function after() {
+  await driver.drop();
+  await driver.close();
+}

test/benchmarks/driver_bench/src/suites/single_bench/primes.mts

@@ -0,0 +1,46 @@
+import assert from 'node:assert/strict';
+
+const findPrimesBelow = 290_000;
+const expectedPrimes = 25_224;
+
+// should never change
+export const taskSize = 1.00896;
+
+// expected primes * 4 bytes (int32) in MB
+assert.equal(taskSize, expectedPrimes * 4 * 10e-6);
+
+/** @see https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes */
+function sieveOfEratosthenes(n: number) {
+  // Create a boolean array "prime[0..n]" and initialize
+  // all entries as true. A value in prime[i] will
+  // become false if i is Not a prime
+  const prime = Array.from({ length: n + 1 }, () => true);
+
+  // We know 0 and 1 are not prime
+  prime[0] = false;
+  prime[1] = false;
+
+  for (let p = 2; p * p <= n; p++) {
+    // If prime[p] is not changed, then it is a prime
+    if (prime[p] === true) {
+      // Update all multiples of p as false
+      for (let i = p * p; i <= n; i += p) {
+        prime[i] = false;
+      }
+    }
+  }
+
+  // Collecting all prime numbers
+  const primes = [];
+  for (let i = 2; i <= n; i++) {
+    if (prime[i] === true) {
+      primes.push(i);
+    }
+  }
+
+  return primes;
+}
+
+export async function run() {
+  assert.equal(sieveOfEratosthenes(findPrimesBelow).length, expectedPrimes);
+}

test/benchmarks/driver_bench/src/suites/single_bench/run_command.mts

@@ -0,0 +1,24 @@
+import { driver, type mongodb } from '../../driver.mjs';
+
+// { hello: true } is 13 bytes of BSON x 10,000 iterations
+export const taskSize = 0.13;
+
+let db: mongodb.Db;
+
+export async function before() {
+  await driver.drop();
+  await driver.create();
+
+  db = driver.db;
+}
+
+export async function run() {
+  for (let i = 0; i < 10000; ++i) {
+    await db.command({ hello: true });
+  }
+}
+
+export async function after() {
+  await driver.drop();
+  await driver.close();
+}

test/benchmarks/driver_bench/src/suites/single_bench/small_doc_insert_one.mts

@@ -0,0 +1,32 @@
+import { driver, type mongodb } from '../../driver.mjs';
+
+export const taskSize = 2.75;
+
+let collection: mongodb.Collection;
+let documents: Record<string, any>[];
+let smallDoc: Record<string, any>;
+
+export async function before() {
+  smallDoc = await driver.load('single_and_multi_document/small_doc.json', 'json');
+}
+
+export async function beforeEach() {
+  await driver.drop();
+  await driver.create();
+
+  // Make new "documents" so the _id field is not carried over from the last run
+  documents = Array.from({ length: 10000 }, () => ({ ...smallDoc }));
+
+  collection = driver.collection;
+}
+
+export async function run() {
+  for (const doc of documents) {
+    await collection.insertOne(doc);
+  }
+}
+
+export async function after() {
+  await driver.drop();
+  await driver.close();
+}