repro-nest 0.0.207 → 0.0.208

package/dist/index.d.ts

@@ -29,6 +29,11 @@ type TracerApi = {
     init?: (opts: any) => void;
     tracer?: {
         on: (fn: (ev: any) => void) => () => void;
+        getCurrentTraceContext?: () => {
+            traceId: any;
+            spanId: any;
+            depth?: number;
+        };
     };
     getCurrentTraceId?: () => string | null;
     patchHttp?: () => void;

package/dist/index.js

@@ -439,6 +439,13 @@ function initReproTracing(opts) {
         const initOpts = { ...defaultTracerInitOpts(), ...rest };
         tracerPkg.init?.(initOpts);
         tracerPkg.patchHttp?.();
+        // Ensure tracer exposes current trace context on the on-event API when available.
+        try {
+            if (!tracerPkg.tracer?.getCurrentTraceContext && tracerPkg.getCurrentTraceContext) {
+                tracerPkg.tracer.getCurrentTraceContext = tracerPkg.getCurrentTraceContext;
+            }
+        }
+        catch { }
         applyTraceLogPreference(tracerPkg);
         __TRACER_READY = true;
         patchAllKnownMongooseInstances();
@@ -800,7 +807,8 @@ const SESSION_DRAIN_TIMEOUT_MS = (() => {
     if (Number.isFinite(env) && env >= 0)
         return env;
     // Bound wait for draining sessions to avoid lost flushes when a request hangs.
-    return 10000;
+    // Default to 1 minute; caller can override via env.
+    return 60000;
 })();
 function isThenable(value) {
     return value != null && typeof value === 'object' && typeof value.then === 'function';
@@ -1891,6 +1899,12 @@ function dehydrateComplexValue(value) {
 function emitDbQuery(cfg, sid, aid, payload) {
     if (!sid)
         return;
+    let traceCtx = null;
+    try {
+        const getCtx = __TRACER__?.tracer?.getCurrentTraceContext;
+        traceCtx = typeof getCtx === 'function' ? getCtx() : null;
+    }
+    catch { }
     post(cfg.apiBase, cfg.tenantId, cfg.appId, cfg.appSecret, sid, {
         entries: [{
                 actionId: aid ?? null,
@@ -1900,6 +1914,8 @@ function emitDbQuery(cfg, sid, aid, payload) {
                         query: payload.query ?? undefined,
                         resultMeta: payload.resultMeta ?? undefined,
                         durMs: payload.durMs ?? undefined,
+                        traceId: traceCtx?.traceId ?? null,
+                        spanId: traceCtx?.spanId ?? null,
                         pk: null, before: null, after: null,
                         error: payload.error ?? undefined,
                     }],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "repro-nest",
-  "version": "0.0.207",
+  "version": "0.0.208",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -174,7 +174,10 @@ function patchAllKnownMongooseInstances() {
 // ====== tiny, safe tracer auto-init (no node_modules patches) ======
 type TracerApi = {
     init?: (opts: any) => void;
-    tracer?: { on: (fn: (ev: any) => void) => () => void };
+    tracer?: {
+        on: (fn: (ev: any) => void) => () => void;
+        getCurrentTraceContext?: () => { traceId: any; spanId: any; depth?: number };
+    };
     getCurrentTraceId?: () => string | null;
     patchHttp?: () => void; // optional in your tracer
     setFunctionLogsEnabled?: (enabled: boolean) => void;
@@ -623,6 +626,12 @@ export function initReproTracing(opts?: ReproTracingInitOptions) {
         const initOpts = { ...defaultTracerInitOpts(), ...(rest as TracerInitOpts) };
         tracerPkg.init?.(initOpts);
         tracerPkg.patchHttp?.();
+        // Ensure tracer exposes current trace context on the on-event API when available.
+        try {
+            if (!tracerPkg.tracer?.getCurrentTraceContext && (tracerPkg as any).getCurrentTraceContext) {
+                (tracerPkg.tracer as any).getCurrentTraceContext = (tracerPkg as any).getCurrentTraceContext;
+            }
+        } catch {}
         applyTraceLogPreference(tracerPkg);
         __TRACER_READY = true;
         patchAllKnownMongooseInstances();
@@ -1024,7 +1033,8 @@ const SESSION_DRAIN_TIMEOUT_MS = (() => {
     const env = Number(process.env.SESSION_DRAIN_TIMEOUT_MS);
     if (Number.isFinite(env) && env >= 0) return env;
     // Bound wait for draining sessions to avoid lost flushes when a request hangs.
-    return 10000;
+    // Default to 1 minute; caller can override via env.
+    return 60000;
 })();
 
 function isThenable(value: any): value is PromiseLike<any> {
@@ -2104,6 +2114,12 @@ function dehydrateComplexValue(value: any) {
 
 function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
     if (!sid) return;
+    let traceCtx: { traceId: any; spanId: any } | null = null;
+    try {
+        const getCtx = __TRACER__?.tracer?.getCurrentTraceContext;
+        traceCtx = typeof getCtx === 'function' ? getCtx() : null;
+    } catch {}
+
     post(cfg.apiBase, cfg.tenantId, cfg.appId, cfg.appSecret, sid, {
         entries: [{
             actionId: aid ?? null,
@@ -2113,6 +2129,8 @@ function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
                 query: payload.query ?? undefined,
                 resultMeta: payload.resultMeta ?? undefined,
                 durMs: payload.durMs ?? undefined,
+                traceId: traceCtx?.traceId ?? null,
+                spanId: traceCtx?.spanId ?? null,
                 pk: null, before: null, after: null,
                 error: payload.error ?? undefined,
             }],

package/tracer/runtime.js

@@ -100,6 +100,17 @@ function popSpan(ctx) {
 const trace = {
     on(fn){ listeners.add(fn); return () => listeners.delete(fn); },
     withTrace(id, fn, depth = 0){ return als.run({ traceId: id, depth }, fn); },
+    getCurrentTraceContext(){
+        const store = als.getStore();
+        if (!store) return { traceId: null, spanId: null, depth: 0 };
+        const spanStack = Array.isArray(store.__repro_span_stack) ? store.__repro_span_stack : [];
+        const top = spanStack.length ? spanStack[spanStack.length - 1] : null;
+        return {
+            traceId: store.traceId || null,
+            spanId: top ? top.id ?? null : null,
+            depth: typeof store.depth === 'number' ? store.depth : (top?.depth ?? 0),
+        };
+    },
     enter(fn, meta, detail){
         const parentSpanIdOverride = meta && Object.prototype.hasOwnProperty.call(meta, 'parentSpanId')
             ? meta.parentSpanId