bitfab 0.54.3 → 0.54.4

package/dist/{chunk-NI6B2BP4.js → chunk-KDLFM5FD.js}

@@ -1,3 +1,90 @@
+// src/traceCompletion.ts
+var TraceCompletion = class {
+  constructor() {
+    this.traces = /* @__PURE__ */ new Map();
+    this.closed = /* @__PURE__ */ new Map();
+  }
+  state(traceId) {
+    let state = this.traces.get(traceId);
+    if (state === void 0) {
+      state = { spanIds: /* @__PURE__ */ new Set(), active: /* @__PURE__ */ new Set() };
+      this.traces.set(traceId, state);
+    }
+    return state;
+  }
+  start(traceId, spanId) {
+    if (this.closed.has(traceId)) {
+      return;
+    }
+    const state = this.state(traceId);
+    state.spanIds.add(spanId);
+    state.active.add(spanId);
+  }
+  open(traceId) {
+    if (!this.closed.has(traceId)) {
+      this.state(traceId);
+    }
+  }
+  hold(traceId, key) {
+    this.state(traceId).active.add(key);
+  }
+  record(traceId, spanId) {
+    if (!this.closed.has(traceId)) {
+      this.state(traceId).spanIds.add(spanId);
+    }
+  }
+  end(traceId, spanId) {
+    const state = this.traces.get(traceId);
+    if (state !== void 0) {
+      state.active.delete(spanId);
+      this.drain(traceId, state);
+    }
+  }
+  abort(traceId, spanId) {
+    const state = this.traces.get(traceId);
+    if (state === void 0) {
+      return;
+    }
+    state.active.delete(spanId);
+    state.spanIds.delete(spanId);
+    if (state.active.size === 0 && state.spanIds.size === 0 && state.complete === void 0) {
+      this.traces.delete(traceId);
+      return;
+    }
+    this.drain(traceId, state);
+  }
+  close(traceId, complete, dropped = false) {
+    const count = this.closed.get(traceId);
+    if (count !== void 0) {
+      complete(count);
+      return;
+    }
+    const state = this.traces.get(traceId);
+    if (state === void 0) {
+      return;
+    }
+    state.complete = complete;
+    if (dropped) {
+      state.active.clear();
+    }
+    this.drain(traceId, state);
+  }
+  drain(traceId, state) {
+    if (state.active.size !== 0 || state.complete === void 0) {
+      return;
+    }
+    this.traces.delete(traceId);
+    this.closed.set(traceId, state.spanIds.size);
+    if (this.closed.size > 1024) {
+      const oldest = this.closed.keys().next().value;
+      if (oldest !== void 0) {
+        this.closed.delete(oldest);
+      }
+    }
+    state.complete(state.spanIds.size);
+  }
+};
+
 // src/readEnv.ts
 function readEnv(name) {
   if (typeof process !== "undefined" && process.env) {
@@ -94,7 +181,7 @@ function encodeRequestBody(body) {
 }
 
 // src/version.generated.ts
-var __version__ = "0.54.3";
+var __version__ = "0.54.4";
 var __packageName__ = "bitfab";
 
 // src/errors.ts
@@ -1540,6 +1627,7 @@ var HttpClient = class {
     // must not wait on another client's slow finalize: a false `close()` failure
     // caused by unrelated work is worse than no signal at all.
     this.deferredWork = /* @__PURE__ */ new Set();
+    this.traceCompletion = new TraceCompletion();
     this.closed = false;
     this.apiKey = config.apiKey;
     this.serviceUrl = config.serviceUrl;
@@ -1973,6 +2061,10 @@ var HttpClient = class {
     );
   }
   sendExternalSpan(payload) {
+    const ref = carrierRef(payload);
+    if (ref?.spanId !== void 0) {
+      this.traceCompletion.record(ref.traceId, ref.spanId);
+    }
     const gate = this.externalSpanTransform;
     if (gate === void 0) {
       this.submitExternalSpan(payload);
@@ -2004,13 +2096,28 @@ var HttpClient = class {
    */
   sendExternalTrace(rawPayload) {
     const payload = mergeCallerMetadataIntoTracePayload(rawPayload);
-    const gate = this.externalTraceTransform;
-    if (gate === void 0) {
-      this.submitExternalTrace(payload);
+    const traceId = sourceTraceIdOf(payload);
+    if (payload.completed === true && traceId !== void 0) {
+      this.traceCompletion.close(
+        traceId,
+        (expectedSpanCount) => this.sendCountedExternalTrace({ ...payload, expectedSpanCount }),
+        payload.dropped === true
+      );
       return;
     }
+    if (traceId !== void 0) {
+      this.traceCompletion.open(traceId);
+    }
+    this.sendCountedExternalTrace(payload);
+  }
+  sendCountedExternalTrace(payload) {
     try {
-      gate(payload, (ready) => this.submitExternalTrace(ready));
+      const gate = this.externalTraceTransform;
+      if (gate === void 0) {
+        this.submitExternalTrace(payload);
+      } else {
+        gate(payload, (ready) => this.submitExternalTrace(ready));
+      }
     } catch (error) {
       warnDroppedExternalTrace(error);
     }
@@ -2264,4 +2371,4 @@ export {
   parseRetryAfterMs,
   HttpClient
 };
-//# sourceMappingURL=chunk-NI6B2BP4.js.map
+//# sourceMappingURL=chunk-KDLFM5FD.js.map