@mastra/observability 1.2.0 → 1.2.1

package/dist/index.js

@@ -5556,6 +5556,8 @@ var DefaultExporter = class extends BaseExporter {
   #resolvedStrategy;
   buffer;
   #flushTimer = null;
+  #isInitializing = false;
+  #initPromises = /* @__PURE__ */ new Set();
   // Track all spans that have been created, persists across flushes
   allCreatedSpans = /* @__PURE__ */ new Set();
   constructor(config = {}) {
@@ -5589,17 +5591,28 @@ var DefaultExporter = class extends BaseExporter {
    * Initialize the exporter (called after all dependencies are ready)
    */
   async init(options) {
-    this.#storage = options.mastra?.getStorage();
-    if (!this.#storage) {
-      this.logger.warn("DefaultExporter disabled: Storage not available. Traces will not be persisted.");
-      return;
-    }
-    this.#observability = await this.#storage.getStore("observability");
-    if (!this.#observability) {
-      this.logger.warn("DefaultExporter disabled: Observability storage not available. Traces will not be persisted.");
-      return;
+    try {
+      this.#isInitializing = true;
+      this.#storage = options.mastra?.getStorage();
+      if (!this.#storage) {
+        this.logger.warn("DefaultExporter disabled: Storage not available. Traces will not be persisted.");
+        return;
+      }
+      this.#observability = await this.#storage.getStore("observability");
+      if (!this.#observability) {
+        this.logger.warn(
+          "DefaultExporter disabled: Observability storage not available. Traces will not be persisted."
+        );
+        return;
+      }
+      this.initializeStrategy(this.#observability, this.#storage.constructor.name);
+    } finally {
+      this.#isInitializing = false;
+      this.#initPromises.forEach((resolve) => {
+        resolve();
+      });
+      this.#initPromises.clear();
     }
-    this.initializeStrategy(this.#observability, this.#storage.constructor.name);
   }
   /**
    * Initialize the resolved strategy once observability store is available
@@ -6004,6 +6017,7 @@ var DefaultExporter = class extends BaseExporter {
     }
   }
   async _exportTracingEvent(event) {
+    await this.waitForInit();
     if (!this.#observability) {
       this.logger.debug("Cannot store traces. Observability storage is not initialized");
       return;
@@ -6023,6 +6037,17 @@ var DefaultExporter = class extends BaseExporter {
         break;
     }
   }
+  /**
+   * Resolves when an ongoing init call is finished
+   * Doesn't wait for the caller to call init
+   * @returns
+   */
+  async waitForInit() {
+    if (!this.#isInitializing) return;
+    return new Promise((resolve) => {
+      this.#initPromises.add(resolve);
+    });
+  }
   /**
    * Force flush any buffered spans without shutting down the exporter.
    * This is useful in serverless environments where you need to ensure spans
@@ -7451,6 +7476,7 @@ function compressJsonSchema(schema, depth = 0) {
 }
 function deepClean(value, options = DEFAULT_DEEP_CLEAN_OPTIONS) {
   const { keysToStrip, maxDepth, maxStringLength, maxArrayLength, maxObjectKeys } = options;
+  const stripSet = keysToStrip instanceof Set ? keysToStrip : new Set(Array.isArray(keysToStrip) ? keysToStrip : Object.keys(keysToStrip));
   const seen = /* @__PURE__ */ new WeakSet();
   function helper(val, depth) {
     if (depth > maxDepth) {
@@ -7521,7 +7547,7 @@ function deepClean(value, options = DEFAULT_DEEP_CLEAN_OPTIONS) {
     const entries = Object.entries(val);
     let keyCount = 0;
     for (const [key, v] of entries) {
-      if (keysToStrip.has(key)) {
+      if (stripSet.has(key)) {
         continue;
       }
       if (keyCount >= maxObjectKeys) {