@inference-net/otel-cf-workers 2.5.0 → 2.5.1

package/dist/index.d.ts

@@ -549,7 +549,7 @@ declare class TraceState {
     addSpan(span: Span_2): void;
     endSpan(span: ReadableSpan): void;
     sample(): void;
-    flush(): Promise<void>;
+    flush(): Promise<boolean>;
     private isSpanInProgress;
     private exportSpans;
 }

package/dist/index.js

@@ -96,7 +96,7 @@ function passthroughGet(target, prop, thisArg) {
   }
 }
 
-const PACKAGE_VERSION = "2.5.0";
+const PACKAGE_VERSION = "2.5.1";
 const PACKAGE_NAME = "@inference-net/otel-cf-workers";
 const ATTR_CLOUDFLARE_COLO = "cloudflare.colo";
 const ATTR_CLOUDFLARE_RAY_ID = "cloudflare.ray_id";
@@ -358,9 +358,11 @@ class TraceState {
       this.exportPromises.push(this.exportSpans(this.unexportedSpans));
       this.unexportedSpans = [];
     }
-    if (this.exportPromises.length > 0) {
+    const exportCount = this.exportPromises.length;
+    if (exportCount > 0) {
       await Promise.allSettled(this.exportPromises);
     }
+    return this.inprogressSpans.size === 0 && this.unexportedSpans.length === 0 && this.exportPromises.length === exportCount;
   }
   isSpanInProgress(span) {
     return this.inprogressSpans.has(span.spanContext().spanId);
@@ -399,11 +401,13 @@ class BatchTraceSpanProcessor {
     this.getTraceState(traceId).endSpan(span);
   }
   async forceFlush(traceId) {
-    if (traceId) {
-      await this.getTraceState(traceId).flush();
-    } else {
-      const promises = Object.values(this.traces).map((traceState) => traceState.flush);
-      await Promise.allSettled(promises);
+    if (!traceId) {
+      await Promise.allSettled(Object.keys(this.traces).map((id) => this.forceFlush(id)));
+      return;
+    }
+    const state = this.traces[traceId];
+    if (state && await state.flush() && this.traces[traceId] === state) {
+      delete this.traces[traceId];
     }
   }
   async shutdown() {