@mastra/observability 1.17.1 → 1.17.2-alpha.0

package/dist/index.js

@@ -7098,8 +7098,8 @@ const FLUSH_SUCCEEDED = { failed: false };
 var MastraStorageExporter = class extends BaseExporter {
 	name = "mastra-storage-exporter";
 	#config;
-	#isInitializing = false;
-	#initPromises = /* @__PURE__ */ new Set();
+	#initializationPromise;
+	#wasUnavailable = false;
 	#eventBuffer;
 	#storage;
 	#observabilityStorage;
@@ -7124,41 +7124,56 @@ var MastraStorageExporter = class extends BaseExporter {
 	* Initialize the exporter (called after all dependencies are ready)
 	*/
 	async init(options) {
+		this.#emitDropEvent = options.emitDropEvent;
+		this.#storage = options.mastra?.getStorage();
+		if (!this.#storage) {
+			this.logger.warn("MastraStorageExporter disabled: Storage not available. Traces will not be persisted.");
+			return;
+		}
+		await this.ensureInitialized();
+	}
+	warnUnavailable(message, error) {
+		if (this.#wasUnavailable) return;
+		this.#wasUnavailable = true;
+		if (error === void 0) {
+			this.logger.warn(message);
+			return;
+		}
+		this.logger.warn(message, { error: error instanceof Error ? error.message : String(error) });
+	}
+	async initializeStorage() {
+		const storage = this.#storage;
+		if (!storage) return;
 		try {
-			this.#isInitializing = true;
-			this.#emitDropEvent = options.emitDropEvent;
-			this.#storage = options.mastra?.getStorage();
-			if (!this.#storage) {
-				this.logger.warn("MastraStorageExporter disabled: Storage not available. Traces will not be persisted.");
-				return;
-			}
-			this.#observabilityStorage = await this.#storage.getStore("observability");
-			if (!this.#observabilityStorage) {
-				this.logger.warn("MastraStorageExporter disabled: Observability storage not available. Traces will not be persisted.");
-				return;
-			}
-			if (!this.#resolvedStrategy) {
-				this.#resolvedStrategy = resolveTracingStorageStrategy(this.#config, this.#observabilityStorage, this.#storage.constructor.name, this.logger);
-				this.logger.debug("tracing storage exporter initialized", {
-					strategy: this.#resolvedStrategy,
-					source: this.#config.strategy !== "auto" ? "user" : "auto",
-					storageAdapter: this.#storage.constructor.name,
-					maxBatchSize: this.#config.maxBatchSize,
-					maxBatchWaitMs: this.#config.maxBatchWaitMs
-				});
-			}
-			if (this.#resolvedStrategy) this.#eventBuffer.init({ strategy: this.#resolvedStrategy });
-		} finally {
-			this.#isInitializing = false;
-			/**
-			* Assumes caller waits until export of a parent span is completed before calling
-			* export for child spans , order is not relevant for resolve
-			*/
-			this.#initPromises.forEach((resolve) => {
-				resolve();
-			});
-			this.#initPromises.clear();
+			this.#observabilityStorage = await storage.getStore("observability");
+		} catch (error) {
+			this.warnUnavailable("MastraStorageExporter unavailable: Failed to initialize observability storage. Traces will not be persisted until storage becomes available.", error);
+			return;
 		}
+		if (!this.#observabilityStorage) {
+			this.warnUnavailable("MastraStorageExporter unavailable: Observability storage not available. Traces will not be persisted until storage becomes available.");
+			return;
+		}
+		this.#resolvedStrategy = resolveTracingStorageStrategy(this.#config, this.#observabilityStorage, storage.constructor.name, this.logger);
+		this.#eventBuffer.init({ strategy: this.#resolvedStrategy });
+		this.logger.debug("tracing storage exporter initialized", {
+			strategy: this.#resolvedStrategy,
+			source: this.#config.strategy !== "auto" ? "user" : "auto",
+			storageAdapter: storage.constructor.name,
+			maxBatchSize: this.#config.maxBatchSize,
+			maxBatchWaitMs: this.#config.maxBatchWaitMs
+		});
+		if (this.#wasUnavailable) {
+			this.#wasUnavailable = false;
+			this.logger.info("MastraStorageExporter recovered: Observability storage is available. Traces will be persisted.");
+		}
+	}
+	async ensureInitialized() {
+		if (this.#observabilityStorage || !this.#storage) return;
+		this.#initializationPromise ??= this.initializeStorage().finally(() => {
+			this.#initializationPromise = void 0;
+		});
+		await this.#initializationPromise;
 	}
 	/**
 	* Checks if buffer should be flushed based on size or time triggers
@@ -7410,7 +7425,7 @@ var MastraStorageExporter = class extends BaseExporter {
 		}
 	}
 	async _exportTracingEvent(event) {
-		await this.waitForInit();
+		await this.ensureInitialized();
 		if (!this.#observabilityStorage) {
 			this.logger.debug("Cannot store traces. Observability storage is not initialized");
 			return;
@@ -7419,21 +7434,10 @@ var MastraStorageExporter = class extends BaseExporter {
 		await this.handleBatchedFlush();
 	}
 	/**
-	* 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);
-		});
-	}
-	/**
 	* Handle metric events — buffer for batch flush.
 	*/
 	async onMetricEvent(event) {
-		await this.waitForInit();
+		await this.ensureInitialized();
 		if (!this.#observabilityStorage) return;
 		this.#eventBuffer.addEvent(event);
 		await this.handleBatchedFlush();
@@ -7442,7 +7446,7 @@ var MastraStorageExporter = class extends BaseExporter {
 	* Handle log events — buffer for batch flush.
 	*/
 	async onLogEvent(event) {
-		await this.waitForInit();
+		await this.ensureInitialized();
 		if (!this.#observabilityStorage) return;
 		this.#eventBuffer.addEvent(event);
 		await this.handleBatchedFlush();
@@ -7451,7 +7455,7 @@ var MastraStorageExporter = class extends BaseExporter {
 	* Handle score events — buffer for batch flush.
 	*/
 	async onScoreEvent(event) {
-		await this.waitForInit();
+		await this.ensureInitialized();
 		if (!this.#observabilityStorage) return;
 		this.#eventBuffer.addEvent(event);
 		await this.handleBatchedFlush();
@@ -7460,7 +7464,7 @@ var MastraStorageExporter = class extends BaseExporter {
 	* Handle feedback events — buffer for batch flush.
 	*/
 	async onFeedbackEvent(event) {
-		await this.waitForInit();
+		await this.ensureInitialized();
 		if (!this.#observabilityStorage) return;
 		this.#eventBuffer.addEvent(event);
 		await this.handleBatchedFlush();