feat(tinybench-plugin): add dynamic startInstruments function

CodSpeedHQ · Jan 12, 2024 · 676473d · 676473d
1 parent e0fa17e
commit 676473d
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/packages/tinybench-plugin/src/index.ts b/packages/tinybench-plugin/src/index.ts
@@ -4,6 +4,8 @@ import {
   mongoMeasurement,
   optimizeFunction,
   setupCore,
+  StartInstrumentsRequestBody,
+  StartInstrumentsResponse,
   teardownCore,
   tryIntrospect,
 } from "@codspeed/core";
@@ -101,3 +103,18 @@ function getCallingFile(): string {
   }
   return path.relative(gitDir, callingFile);
 }
+
+/**
+ * Dynamically starts the CodSpeed instruments.
+ */
+export async function startInstruments(
+  body: StartInstrumentsRequestBody
+): Promise<StartInstrumentsResponse> {
+  if (!Measurement.isInstrumented()) {
+    console.warn("[CodSpeed] No instrumentation found, using default mongoUrl");
+
+    return { remoteAddr: body.mongoUrl };
+  }
+
+  return await mongoMeasurement.startInstruments(body);
+}