feat: integrate CodSpeed Mongo Tracer

Author: art049

packages/benchmark.js-plugin/src/index.ts

@@ -1,5 +1,6 @@
 import {
   Measurement,
+  mongoMeasurement,
   optimizeFunction,
   optimizeFunctionSync,
   setupCore,
@@ -183,6 +184,7 @@ async function runBenchmarks({
       benchPayload = bench.fn as CallableFunction;
     }
 
+    await mongoMeasurement.start(uri);
     if (isAsync) {
       await optimizeFunction(benchPayload);
       await (async function __codspeed_root_frame__() {
@@ -198,6 +200,8 @@ async function runBenchmarks({
         Measurement.stopInstrumentation(uri);
       })();
     }
+    await mongoMeasurement.stop(uri);
+
     console.log(`    ✔ Measured ${uri}`);
     benchmarkCompletedListeners.forEach((listener) => listener());
     teardownCore();

packages/core/.gitignore

@@ -1,2 +1,3 @@
 build
-prebuilds
+prebuilds
+generated

packages/core/moon.yml

@@ -2,6 +2,7 @@ tasks:
   clean:
     args:
       - build
+      - generated/openapi
   build:
     deps:
       - "build-native-addon"
@@ -14,3 +15,10 @@ tasks:
       - "binding.gyp"
     outputs:
       - "prebuilds"
+
+  build-tracer-client:
+    inputs:
+      - "./tracer.spec.json"
+    outputs:
+      - "src/generated/openapi"
+    command: openapi --client axios --input ./tracer.spec.json --name MongoTracer --output ./src/generated/openapi

packages/core/package.json

@@ -22,9 +22,11 @@
   "devDependencies": {
     "node-addon-api": "^5.1.0",
     "node-gyp": "^9.3.1",
+    "openapi-typescript-codegen": "^0.23.0",
     "prebuildify": "^5.0.1"
   },
   "dependencies": {
+    "axios": "^1.4.0",
     "node-gyp-build": "^4.6.0"
   }
 }

packages/core/src/index.ts

@@ -1,4 +1,5 @@
 import { tryIntrospect } from "./introspection";
+import { MongoMeasurement } from "./mongoMeasurement";
 tryIntrospect();
 
 import native_core from "./native_core";
@@ -10,12 +11,15 @@ const linuxPerf = new native_core.LinuxPerf();
 
 export const isBound = native_core.isBound;
 
+export let mongoMeasurement: MongoMeasurement;
+
 export const setupCore = () => {
   initOptimization();
   native_core.Measurement.stopInstrumentation(
     `Metadata: codspeed-node ${__VERSION__}`
   );
   linuxPerf.start();
+  mongoMeasurement = new MongoMeasurement();
 };
 
 export const teardownCore = () => {

packages/core/src/mongoMeasurement.ts

@@ -0,0 +1,42 @@
+import { MongoTracer } from "./generated/openapi";
+
+export class MongoMeasurement {
+  private tracerClient: MongoTracer | undefined;
+
+  constructor() {
+    const serverUrl = process.env.CODSPEED_MONGO_INSTR_SERVER_ADDRESS;
+    // TODO
+    const mongoUriEnvName = process.env.CODSPEED_MONGO_INSTR_URI_ENV_NAME;
+    if (mongoUriEnvName === undefined) {
+      throw new Error("CODSPEED_MONGO_INSTR_URI_ENV_NAME is not defined");
+    }
+    const mongoUri = process.env[mongoUriEnvName];
+    if (mongoUri === undefined) {
+      throw new Error(`Environment variable ${mongoUriEnvName} is not defined`);
+    }
+    process.env[mongoUriEnvName] =
+      "mongodb://localhost:27018?directConnection=true";
+
+    if (serverUrl !== undefined) {
+      this.tracerClient = new MongoTracer({
+        BASE: serverUrl,
+      });
+    }
+  }
+
+  public async start(uri: string) {
+    if (this.tracerClient !== undefined) {
+      await this.tracerClient.instrumentation.start({
+        uri,
+      });
+    }
+  }
+
+  public async stop(uri: string) {
+    if (this.tracerClient !== undefined) {
+      await this.tracerClient.instrumentation.stop({
+        uri,
+      });
+    }
+  }
+}