taist 0.1.13 → 0.1.15

package/lib/trace-context.js

@@ -5,22 +5,35 @@
  * across function calls without explicit parameter passing.
  *
  * This is the same approach used by OpenTelemetry, Datadog, and other APM tools.
+ *
+ * IMPORTANT: Uses globalThis to share context across bundled modules.
+ * Without this, each bundle would have its own AsyncLocalStorage instance,
+ * breaking context propagation across bundle boundaries.
  */
 
 import { AsyncLocalStorage } from 'async_hooks';
 
-// Global storage for trace context
-export const traceContext = new AsyncLocalStorage();
+// Use globalThis to share context across all bundles/modules
+// This ensures trace context propagates even when code is bundled separately
+const TAIST_CONTEXT_KEY = '__taist_trace_context__';
+const TAIST_COUNTER_KEY = '__taist_id_counter__';
+
+if (!globalThis[TAIST_CONTEXT_KEY]) {
+  globalThis[TAIST_CONTEXT_KEY] = new AsyncLocalStorage();
+}
+if (globalThis[TAIST_COUNTER_KEY] === undefined) {
+  globalThis[TAIST_COUNTER_KEY] = 0;
+}
 
-// Counter for generating unique IDs
-let idCounter = 0;
+// Global storage for trace context (shared via globalThis)
+export const traceContext = globalThis[TAIST_CONTEXT_KEY];
 
 /**
  * Generate a unique trace/span ID
  * @returns {string} Unique identifier
  */
 export function generateId() {
-  return `__${++idCounter}_${Date.now()}`;
+  return `__${++globalThis[TAIST_COUNTER_KEY]}_${Date.now()}`;
 }
 
 /**
@@ -76,5 +89,5 @@ export function startTrace(fn) {
  * Reset the ID counter (for testing)
  */
 export function resetIdCounter() {
-  idCounter = 0;
+  globalThis[TAIST_COUNTER_KEY] = 0;
 }

package/lib/trace-reporter.js

@@ -21,6 +21,10 @@ export class TraceReporter extends EventEmitter {
     this.flushInterval = options.flushInterval || 1000;
     this.maxRetries = options.maxRetries || 3;
     this.retryDelay = options.retryDelay || 100;
+    // Immediate flush mode - sends each trace immediately instead of buffering
+    // This is the default to prevent trace loss on process exit
+    // Set TAIST_BUFFER_TRACES=true to enable batching for high-throughput scenarios
+    this.flushImmediate = options.flushImmediate ?? (process.env.TAIST_BUFFER_TRACES !== 'true');
 
     this.buffer = [];
     this.socket = null;
@@ -32,7 +36,7 @@ export class TraceReporter extends EventEmitter {
     this.closed = false;
     this.shuttingDown = false; // Prevents race between shutdown signal and SIGTERM
 
-    logger.debug("[reporter] Created with socketPath:", this.socketPath);
+    logger.debug("[reporter] Created with socketPath:", this.socketPath, "flushImmediate:", this.flushImmediate);
 
     // Auto-setup if socket path is available
     if (this.socketPath) {
@@ -254,8 +258,11 @@ export class TraceReporter extends EventEmitter {
       });
     }
 
-    // Auto-flush when batch size reached
-    if (this.buffer.length >= this.batchSize) {
+    // Flush immediately if configured (for spawned processes with unpredictable exit)
+    // Otherwise auto-flush when batch size reached
+    if (this.flushImmediate) {
+      this.flush().catch(() => {});
+    } else if (this.buffer.length >= this.batchSize) {
       this.flush().catch(() => {});
     }
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "taist",
-  "version": "0.1.13",
+  "version": "0.1.15",
   "description": "Token-Optimized Testing Framework for AI-Assisted Development",
   "main": "index.js",
   "type": "module",