@reproapp/node-sdk 0.0.9 → 0.0.11

package/dist/index.d.ts

@@ -254,6 +254,8 @@ export type ReproTracingInitOptions = TracerInitOpts & {
 export declare function initReproTracing(opts?: ReproTracingInitOptions): TracerApi | null;
 /** Optional helper if users want to check it. */
 export declare function isReproTracingEnabled(): boolean;
+declare function sanitizeTraceValueForPrivacy(value: any): any;
+declare function sanitizeMaterializedTraceValue(value: any): any;
 export declare function __materializePendingTraceEventsForWorker(payload: TraceMaterializationWorkerPayload): Promise<PendingTraceEventRecord[]>;
 type NormalizedMaskRule = {
     when?: ReproMaskWhen;
@@ -344,6 +346,12 @@ export declare function initRepro(cfg: ReproInitConfig): Promise<void>;
 /** @internal Test hooks for runtime privacy policy wiring. */
 export declare const __reproTestHooks: {
     shareRuntimePrivacyStateForTest: typeof shareRuntimePrivacyState;
+    setFullValueCaptureEnabledForTest(enabled: boolean): void;
+    scheduleSdkBackgroundWorkForTest(work: () => Promise<void> | void, options?: {
+        priority?: 'high' | 'normal';
+    }): void;
+    drainSdkBackgroundQueueForTest(): Promise<void>;
+    resetSdkBackgroundQueuesForTest(): void;
     getRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig): NormalizedRuntimePrivacyPolicy | null;
     setRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig, policy: NormalizedRuntimePrivacyPolicy | null): void;
     recordKafkaTraceEventAsyncForTest: typeof recordKafkaTraceEventAsync;
@@ -352,5 +360,7 @@ export declare const __reproTestHooks: {
     wrapKafkaEachMessageHandlerForTest: typeof wrapKafkaEachMessageHandler;
     chunkTraceEventsForTransportForTest: typeof chunkTraceEventsForTransport;
     estimateTraceBatchSerializedBytesForTest: typeof estimateTraceBatchSerializedBytes;
+    sanitizeTraceValueForPrivacyForTest: typeof sanitizeTraceValueForPrivacy;
+    sanitizeMaterializedTraceValueForTest: typeof sanitizeMaterializedTraceValue;
 };
 export {};

package/dist/index.js

@@ -863,6 +863,7 @@ let activeClientRequestCount = 0;
 let oldestActiveClientRequestAt = null;
 let lastClientActivityAt = 0;
 let sdkBackgroundQuietUntil = 0;
+const sdkBackgroundHighQueue = [];
 const sdkBackgroundQueue = [];
 let sdkBackgroundTimer = null;
 let sdkBackgroundDraining = false;
@@ -1155,8 +1156,13 @@ function endClientRequest() {
         scheduleSdkBackgroundDrain();
     }
 }
-function scheduleSdkBackgroundWork(work) {
-    sdkBackgroundQueue.push(work);
+function scheduleSdkBackgroundWork(work, options = {}) {
+    if (options.priority === 'high') {
+        sdkBackgroundHighQueue.push(work);
+    }
+    else {
+        sdkBackgroundQueue.push(work);
+    }
     scheduleSdkBackgroundDrain();
 }
 function scheduleSdkBackgroundDrain(delayMs = 0) {
@@ -1209,12 +1215,12 @@ async function drainSdkBackgroundQueue() {
     }
     sdkBackgroundDraining = true;
     try {
-        while (sdkBackgroundQueue.length > 0) {
+        while (sdkBackgroundHighQueue.length > 0 || sdkBackgroundQueue.length > 0) {
             if (shouldDeferSdkBackgroundWork()) {
                 scheduleSdkBackgroundDrain();
                 return;
             }
-            const work = sdkBackgroundQueue.shift();
+            const work = sdkBackgroundHighQueue.shift() ?? sdkBackgroundQueue.shift();
             if (!work)
                 continue;
             try {
@@ -1228,7 +1234,7 @@ async function drainSdkBackgroundQueue() {
     }
     finally {
         sdkBackgroundDraining = false;
-        if (sdkBackgroundQueue.length > 0) {
+        if (sdkBackgroundHighQueue.length > 0 || sdkBackgroundQueue.length > 0) {
             scheduleSdkBackgroundDrain();
         }
         else {
@@ -2032,14 +2038,26 @@ function sanitizeTraceValue(value, depth = 0, seen = new WeakMap(), options = {}
         return circularReference(existingPath);
     seen.set(value, valuePath);
     if (!options.disableTruncation && depth >= TRACE_VALUE_MAX_DEPTH) {
-        const shallow = safeJson(value);
-        if (shallow !== undefined) {
-            return shallow;
+        if (Array.isArray(value)) {
+            return {
+                __type: 'Array',
+                length: value.length,
+                __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}`,
+            };
         }
         const ctor = value?.constructor?.name;
-        return ctor && ctor !== 'Object'
-            ? { __class: ctor, __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}` }
-            : { __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}` };
+        const keys = Object.keys(value);
+        const summary = {
+            __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}`,
+            __keys: keys.slice(0, TRACE_VALUE_MAX_KEYS),
+        };
+        if (keys.length > TRACE_VALUE_MAX_KEYS) {
+            summary.__truncatedKeys = keys.length - TRACE_VALUE_MAX_KEYS;
+        }
+        if (ctor && ctor !== 'Object') {
+            summary.__class = ctor;
+        }
+        return summary;
     }
     if (Array.isArray(value)) {
         const sourceItems = options.disableTruncation ? value : value.slice(0, TRACE_VALUE_MAX_ITEMS);
@@ -2105,6 +2123,9 @@ function sanitizeTraceArgs(values) {
 function sanitizeTraceValueForPrivacy(value) {
     return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
 }
+function sanitizeInlinePrivacyValue(value) {
+    return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
+}
 function sanitizeMaterializedTraceValue(value) {
     return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
 }
@@ -3860,7 +3881,7 @@ function reproMiddleware(cfg) {
                 }
                 catch { }
                 if (flushPayload) {
-                    const scheduleFlushPayload = () => scheduleSdkBackgroundWork(flushPayload);
+                    const scheduleFlushPayload = () => scheduleSdkBackgroundWork(flushPayload, { priority: 'high' });
                     if (sessionDrainWait) {
                         sessionDrainWait.then(scheduleFlushPayload).catch(scheduleFlushPayload);
                     }
@@ -5176,7 +5197,7 @@ function cloneKafkaTelemetryValue(value) {
     }
 }
 async function materializeKafkaRequestBody(cfg, maskReq, body) {
-    return materializeInlinePrivacyValueAsync('request.body', sanitizeTraceValueForPrivacy(body), cfg, maskReq, null, normalizeMaskingConfig(cfg.masking), getRuntimePrivacyState(cfg).policy ?? null);
+    return materializeInlinePrivacyValueAsync('request.body', sanitizeInlinePrivacyValue(body), cfg, maskReq, null, normalizeMaskingConfig(cfg.masking), getRuntimePrivacyState(cfg).policy ?? null);
 }
 function summarizeKafkaError(error) {
     const message = typeof error?.message === 'string'
@@ -5404,7 +5425,7 @@ function scheduleKafkaTraceFlush(cfg, kafkaCtx, actionId, requestRid, rawTraceEv
                 post(cfg, sid, { entries });
             }
         });
-    });
+    }, { priority: 'high' });
 }
 function patchKafkaProducerInstance(producer, cfg) {
     if (!producer || producer.__repro_kafka_producer_patched)
@@ -6667,6 +6688,31 @@ exports.initRepro = initRepro;
 /** @internal Test hooks for runtime privacy policy wiring. */
 exports.__reproTestHooks = {
     shareRuntimePrivacyStateForTest: shareRuntimePrivacyState,
+    setFullValueCaptureEnabledForTest(enabled) {
+        __FULL_VALUE_CAPTURE_ENABLED = enabled === true;
+    },
+    scheduleSdkBackgroundWorkForTest(work, options) {
+        scheduleSdkBackgroundWork(work, options);
+    },
+    async drainSdkBackgroundQueueForTest() {
+        await drainSdkBackgroundQueue();
+    },
+    resetSdkBackgroundQueuesForTest() {
+        sdkBackgroundHighQueue.length = 0;
+        sdkBackgroundQueue.length = 0;
+        if (sdkBackgroundTimer) {
+            try {
+                clearTimeout(sdkBackgroundTimer);
+            }
+            catch { }
+            sdkBackgroundTimer = null;
+        }
+        sdkBackgroundDraining = false;
+        activeClientRequestCount = 0;
+        oldestActiveClientRequestAt = null;
+        lastClientActivityAt = 0;
+        sdkBackgroundQuietUntil = 0;
+    },
     getRuntimePrivacyPolicyForTest(cfg) {
         return getRuntimePrivacyState(cfg).policy;
     },
@@ -6679,4 +6725,6 @@ exports.__reproTestHooks = {
     wrapKafkaEachMessageHandlerForTest: wrapKafkaEachMessageHandler,
     chunkTraceEventsForTransportForTest: chunkTraceEventsForTransport,
     estimateTraceBatchSerializedBytesForTest: estimateTraceBatchSerializedBytes,
+    sanitizeTraceValueForPrivacyForTest: sanitizeTraceValueForPrivacy,
+    sanitizeMaterializedTraceValueForTest: sanitizeMaterializedTraceValue,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.9",
+  "version": "0.0.11",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,7 +12,7 @@
     "build": "tsc -p tsconfig.json",
     "dev": "tsc -p tsconfig.json --watch --preserveWatchOutput",
     "prepublishOnly": "npm run build",
-    "test": "npm run build && node test/unawaited.test.js && node test/integration-unawaited.js && node test/request-flush-timing.test.js && node test/express-trace-http-args.test.js && node test/trace-batch-size.test.js && node -r ./tracer/register test/promise-map.test.js && node test/disable-subtree.test.js && node test/circular-capture.test.js && node test/wrap-plugin-arrow-args.test.js && node test/privacy-runtime-policy.test.js && node test/runtime-privacy-materialization.test.js && node test/kafka-runtime-privacy-policy.test.js"
+    "test": "npm run build && node test/unawaited.test.js && node test/integration-unawaited.js && node test/request-flush-timing.test.js && node test/express-trace-http-args.test.js && node test/trace-batch-size.test.js && node test/sdk-background-priority.test.js && node -r ./tracer/register test/promise-map.test.js && node test/disable-subtree.test.js && node test/circular-capture.test.js && node test/wrap-plugin-arrow-args.test.js && node test/privacy-runtime-policy.test.js && node test/runtime-privacy-materialization.test.js && node test/kafka-runtime-privacy-policy.test.js"
   },
   "peerDependencies": {
     "express": "^5.1.0",

package/src/index.ts

@@ -1231,6 +1231,7 @@ let activeClientRequestCount = 0;
 let oldestActiveClientRequestAt: number | null = null;
 let lastClientActivityAt = 0;
 let sdkBackgroundQuietUntil = 0;
+const sdkBackgroundHighQueue: Array<() => Promise<void> | void> = [];
 const sdkBackgroundQueue: Array<() => Promise<void> | void> = [];
 let sdkBackgroundTimer: NodeJS.Timeout | null = null;
 let sdkBackgroundDraining = false;
@@ -1499,8 +1500,15 @@ function endClientRequest(): void {
     }
 }
 
-function scheduleSdkBackgroundWork(work: () => Promise<void> | void): void {
-    sdkBackgroundQueue.push(work);
+function scheduleSdkBackgroundWork(
+    work: () => Promise<void> | void,
+    options: { priority?: 'high' | 'normal' } = {},
+): void {
+    if (options.priority === 'high') {
+        sdkBackgroundHighQueue.push(work);
+    } else {
+        sdkBackgroundQueue.push(work);
+    }
     scheduleSdkBackgroundDrain();
 }
 
@@ -1549,12 +1557,12 @@ async function drainSdkBackgroundQueue(): Promise<void> {
 
     sdkBackgroundDraining = true;
     try {
-        while (sdkBackgroundQueue.length > 0) {
+        while (sdkBackgroundHighQueue.length > 0 || sdkBackgroundQueue.length > 0) {
             if (shouldDeferSdkBackgroundWork()) {
                 scheduleSdkBackgroundDrain();
                 return;
             }
-            const work = sdkBackgroundQueue.shift();
+            const work = sdkBackgroundHighQueue.shift() ?? sdkBackgroundQueue.shift();
             if (!work) continue;
             try {
                 await drainQueuedIngestBeforeBackgroundWork();
@@ -1565,7 +1573,7 @@ async function drainSdkBackgroundQueue(): Promise<void> {
         }
     } finally {
         sdkBackgroundDraining = false;
-        if (sdkBackgroundQueue.length > 0) {
+        if (sdkBackgroundHighQueue.length > 0 || sdkBackgroundQueue.length > 0) {
             scheduleSdkBackgroundDrain();
         } else {
             kickIngestQueueDrain();
@@ -2429,14 +2437,26 @@ function sanitizeTraceValue(
     seen.set(value, valuePath);
 
     if (!options.disableTruncation && depth >= TRACE_VALUE_MAX_DEPTH) {
-        const shallow = safeJson(value);
-        if (shallow !== undefined) {
-            return shallow;
+        if (Array.isArray(value)) {
+            return {
+                __type: 'Array',
+                length: value.length,
+                __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}`,
+            };
         }
         const ctor = value?.constructor?.name;
-        return ctor && ctor !== 'Object'
-            ? { __class: ctor, __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}` }
-            : { __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}` };
+        const keys = Object.keys(value);
+        const summary: Record<string, any> = {
+            __truncated: `depth>${TRACE_VALUE_MAX_DEPTH}`,
+            __keys: keys.slice(0, TRACE_VALUE_MAX_KEYS),
+        };
+        if (keys.length > TRACE_VALUE_MAX_KEYS) {
+            summary.__truncatedKeys = keys.length - TRACE_VALUE_MAX_KEYS;
+        }
+        if (ctor && ctor !== 'Object') {
+            summary.__class = ctor;
+        }
+        return summary;
     }
 
     if (Array.isArray(value)) {
@@ -2505,6 +2525,10 @@ function sanitizeTraceValueForPrivacy(value: any): any {
     return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
 }
 
+function sanitizeInlinePrivacyValue(value: any): any {
+    return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
+}
+
 function sanitizeMaterializedTraceValue(value: any): any {
     return sanitizeTraceValue(value, 0, new WeakMap(), { preserveLongStrings: true, disableTruncation: true });
 }
@@ -4674,11 +4698,11 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
 	                        return;
 	                    }
 	                }
-	                flushed = true;
-	                clearTimers();
+                flushed = true;
+                clearTimers();
                 try { unsubscribe && unsubscribe(); } catch {}
                 if (flushPayload) {
-                    const scheduleFlushPayload = () => scheduleSdkBackgroundWork(flushPayload!);
+                    const scheduleFlushPayload = () => scheduleSdkBackgroundWork(flushPayload!, { priority: 'high' });
                     if (sessionDrainWait) {
                         sessionDrainWait.then(scheduleFlushPayload).catch(scheduleFlushPayload);
                     } else {
@@ -6083,7 +6107,7 @@ async function materializeKafkaRequestBody(
 ): Promise<InlinePrivacyMaterializationResult> {
     return materializeInlinePrivacyValueAsync(
         'request.body',
-        sanitizeTraceValueForPrivacy(body),
+        sanitizeInlinePrivacyValue(body),
         cfg,
         maskReq,
         null,
@@ -6381,7 +6405,7 @@ function scheduleKafkaTraceFlush(
                 post(cfg, sid, { entries });
             }
         });
-    });
+    }, { priority: 'high' });
 }
 
 function patchKafkaProducerInstance(producer: any, cfg: KafkaJsPatchConfig) {
@@ -7686,6 +7710,31 @@ export async function initRepro(cfg: ReproInitConfig): Promise<void> {
 /** @internal Test hooks for runtime privacy policy wiring. */
 export const __reproTestHooks = {
     shareRuntimePrivacyStateForTest: shareRuntimePrivacyState,
+    setFullValueCaptureEnabledForTest(enabled: boolean): void {
+        __FULL_VALUE_CAPTURE_ENABLED = enabled === true;
+    },
+    scheduleSdkBackgroundWorkForTest(
+        work: () => Promise<void> | void,
+        options?: { priority?: 'high' | 'normal' },
+    ): void {
+        scheduleSdkBackgroundWork(work, options);
+    },
+    async drainSdkBackgroundQueueForTest(): Promise<void> {
+        await drainSdkBackgroundQueue();
+    },
+    resetSdkBackgroundQueuesForTest(): void {
+        sdkBackgroundHighQueue.length = 0;
+        sdkBackgroundQueue.length = 0;
+        if (sdkBackgroundTimer) {
+            try { clearTimeout(sdkBackgroundTimer); } catch {}
+            sdkBackgroundTimer = null;
+        }
+        sdkBackgroundDraining = false;
+        activeClientRequestCount = 0;
+        oldestActiveClientRequestAt = null;
+        lastClientActivityAt = 0;
+        sdkBackgroundQuietUntil = 0;
+    },
     getRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig): NormalizedRuntimePrivacyPolicy | null {
         return getRuntimePrivacyState(cfg).policy;
     },
@@ -7701,4 +7750,6 @@ export const __reproTestHooks = {
     wrapKafkaEachMessageHandlerForTest: wrapKafkaEachMessageHandler,
     chunkTraceEventsForTransportForTest: chunkTraceEventsForTransport,
     estimateTraceBatchSerializedBytesForTest: estimateTraceBatchSerializedBytes,
+    sanitizeTraceValueForPrivacyForTest: sanitizeTraceValueForPrivacy,
+    sanitizeMaterializedTraceValueForTest: sanitizeMaterializedTraceValue,
 };

package/test/kafka-runtime-privacy-policy.test.js

@@ -567,6 +567,7 @@ async function testLargeStringTraceValuesAreCapturedInlineWithoutTruncation() {
     };
 
     const sink = [];
+    __reproTestHooks.setFullValueCaptureEnabledForTest(true);
     await __reproTestHooks.recordKafkaTraceEventAsyncForTest(
         {
             type: 'enter',
@@ -579,6 +580,7 @@ async function testLargeStringTraceValuesAreCapturedInlineWithoutTruncation() {
         cfg,
         maskReq,
     );
+    __reproTestHooks.setFullValueCaptureEnabledForTest(false);
 
     assert.strictEqual(sink.length, 1);
     assert.strictEqual(sink[0].fn, 'parseGatewayPayload');
@@ -622,6 +624,7 @@ async function testLargeStructuredTraceValuesAreCapturedInlineWithoutTruncation(
     };
 
     const sink = [];
+    __reproTestHooks.setFullValueCaptureEnabledForTest(true);
     await __reproTestHooks.recordKafkaTraceEventAsyncForTest(
         {
             type: 'enter',
@@ -634,6 +637,7 @@ async function testLargeStructuredTraceValuesAreCapturedInlineWithoutTruncation(
         cfg,
         maskReq,
     );
+    __reproTestHooks.setFullValueCaptureEnabledForTest(false);
 
     assert.strictEqual(sink.length, 1);
     assert.strictEqual(sink[0].fn, 'recordConsumedEvent');
@@ -677,6 +681,7 @@ async function testLargeTraceReturnValueIsCapturedInlineWithoutTruncation() {
     };
 
     const sink = [];
+    __reproTestHooks.setFullValueCaptureEnabledForTest(true);
     await __reproTestHooks.recordKafkaTraceEventAsyncForTest(
         {
             type: 'exit',
@@ -689,6 +694,7 @@ async function testLargeTraceReturnValueIsCapturedInlineWithoutTruncation() {
         cfg,
         maskReq,
     );
+    __reproTestHooks.setFullValueCaptureEnabledForTest(false);
 
     assert.strictEqual(sink.length, 1);
     assert.strictEqual(sink[0].fn, 'toDrillResponse');

package/test/sdk-background-priority.test.js

@@ -0,0 +1,35 @@
+const assert = require('node:assert/strict');
+
+const sdk = require('../dist/index.js');
+
+async function main() {
+    const hooks = sdk.__reproTestHooks;
+    hooks.resetSdkBackgroundQueuesForTest();
+
+    const order = [];
+
+    hooks.scheduleSdkBackgroundWorkForTest(() => {
+        order.push('normal-1');
+    });
+    hooks.scheduleSdkBackgroundWorkForTest(() => {
+        order.push('normal-2');
+    });
+    hooks.scheduleSdkBackgroundWorkForTest(() => {
+        order.push('high-1');
+    }, { priority: 'high' });
+    hooks.scheduleSdkBackgroundWorkForTest(() => {
+        order.push('high-2');
+    }, { priority: 'high' });
+
+    await hooks.drainSdkBackgroundQueueForTest();
+
+    assert.deepStrictEqual(order, ['high-1', 'high-2', 'normal-1', 'normal-2']);
+
+    hooks.resetSdkBackgroundQueuesForTest();
+    console.log('sdk background priority OK');
+}
+
+main().catch((error) => {
+    console.error(error);
+    process.exit(1);
+});