taist 0.1.12 → 0.1.14

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/lib/transform.js

@@ -145,13 +145,16 @@ export function transformSource(source, moduleNameOrOptions, tracerImportPath) {
 import { getGlobalReporter as __taist_getReporter } from "${reporterPath}";
 import { getContext as __taist_getContext, runWithContext as __taist_runWithContext, generateId as __taist_generateId } from "${traceContextPath}";
 const __taist_reporter = __taist_getReporter();
+const __taist_debug = process.env.TAIST_DEBUG === 'true';
 // Eagerly connect to collector if socket path is set (build-time instrumentation)
 if (process.env.TAIST_COLLECTOR_SOCKET && !__taist_reporter.isConnected()) {
   __taist_reporter.connectEager();
 }
 const __taist_wrap = (fn, name) => {
   if (typeof fn !== 'function') return fn;
+  if (__taist_debug) console.log('[taist] wrapping:', name);
   const wrapped = function(...args) {
+    if (__taist_debug) console.log('[taist] CALLED:', name);
     const parentCtx = __taist_getContext();
     const id = __taist_generateId();
     const depth = parentCtx.depth;
@@ -218,10 +221,12 @@ const __taist_instrumentClass = (cls, name) => {
 };
 const __taist_instrumentObject = (obj, name, visited = new WeakSet()) => {
   if (!obj || typeof obj !== 'object' || visited.has(obj)) return obj;
+  if (__taist_debug) console.log('[taist] instrumentObject:', name, Object.keys(obj));
   visited.add(obj);
   for (const key of Object.keys(obj)) {
     const value = obj[key];
     if (typeof value === 'function') {
+      if (__taist_debug) console.log('[taist] found method:', name + '.' + key);
       obj[key] = __taist_wrap(value, name + '.' + key);
     } else if (value && typeof value === 'object' && !Array.isArray(value)) {
       __taist_instrumentObject(value, name + '.' + key, visited);

package/package.json

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