@reproapp/node-sdk 0.0.8 → 0.0.9

package/dist/index.d.ts

@@ -283,6 +283,8 @@ type InlinePrivacyMaterializationResult = {
     };
 };
 declare function materializeInlinePrivacyValueAsync(target: RuntimePrivacySurface, value: any, cfg: ReproMiddlewareConfig, req: MaskRequestContext, trace: TraceEventForFilter | null, masking: NormalizedMaskingConfig | null, privacy: NormalizedRuntimePrivacyPolicy | null, db?: RuntimePrivacyDbContext | null): Promise<InlinePrivacyMaterializationResult>;
+declare function estimateTraceBatchSerializedBytes(batch: TraceEventRecord[], requestRid: string, actionId: string | null | undefined, batchIndex: number): number;
+declare function chunkTraceEventsForTransport(events: TraceEventRecord[], requestRid: string, actionId: string | null | undefined): TraceEventRecord[][];
 export type ReproMiddlewareConfig = IngestClientConfig & {
     /** Configure header capture/masking. Defaults to capturing with sensitive headers masked. */
     captureHeaders?: boolean | HeaderCaptureOptions;
@@ -348,5 +350,7 @@ export declare const __reproTestHooks: {
     materializeInlinePrivacyValueAsyncForTest: typeof materializeInlinePrivacyValueAsync;
     patchKafkaProducerInstanceForTest: typeof patchKafkaProducerInstance;
     wrapKafkaEachMessageHandlerForTest: typeof wrapKafkaEachMessageHandler;
+    chunkTraceEventsForTransportForTest: typeof chunkTraceEventsForTransport;
+    estimateTraceBatchSerializedBytesForTest: typeof estimateTraceBatchSerializedBytes;
 };
 export {};

package/dist/index.js

@@ -1657,6 +1657,12 @@ const TRACE_INLINE_VALUE_MAX_SERIALIZED_CHARS = (() => {
 })();
 const TRACE_VALUE_SIZE_EXCEEDED = Symbol('trace-value-size-exceeded');
 const TRACE_BATCH_SIZE = 100;
+const TRACE_BATCH_MAX_SERIALIZED_BYTES = (() => {
+    const env = Number(process.env.REPRO_TRACE_BATCH_MAX_SERIALIZED_BYTES);
+    if (Number.isFinite(env) && env > 0)
+        return Math.trunc(env);
+    return 512 * 1024;
+})();
 const TRACE_FLUSH_DELAY_MS = 20;
 // Choose how to order trace events in payloads.
 //  - "chronological" (default): preserve event arrival order (no reshuffle).
@@ -3526,6 +3532,51 @@ function collectBatchTraceValueEntries(batch, batchIndex) {
     });
     return collected;
 }
+function serializedByteLength(value) {
+    try {
+        return Buffer.byteLength(JSON.stringify(value), 'utf8');
+    }
+    catch {
+        return Number.MAX_SAFE_INTEGER;
+    }
+}
+function estimateTraceBatchSerializedBytes(batch, requestRid, actionId, batchIndex) {
+    const traceValues = collectBatchTraceValueEntries(batch, batchIndex);
+    return serializedByteLength({
+        actionId: actionId ?? null,
+        trace: batch,
+        traceValues: traceValues.length ? traceValues : undefined,
+        traceBatch: {
+            rid: requestRid,
+            index: batchIndex,
+            total: 0,
+        },
+        t: 0,
+    });
+}
+function chunkTraceEventsForTransport(events, requestRid, actionId) {
+    if (!Array.isArray(events) || events.length === 0)
+        return [];
+    const countBatches = chunkArray(events, TRACE_BATCH_SIZE);
+    const sizedBatches = [];
+    let current = [];
+    for (const event of countBatches.flat()) {
+        const candidate = current.concat(event);
+        const batchIndex = sizedBatches.length;
+        const estimatedBytes = estimateTraceBatchSerializedBytes(candidate, requestRid, actionId, batchIndex);
+        if (current.length > 0 &&
+            estimatedBytes > TRACE_BATCH_MAX_SERIALIZED_BYTES) {
+            sizedBatches.push(current);
+            current = [event];
+            continue;
+        }
+        current = candidate;
+    }
+    if (current.length > 0) {
+        sizedBatches.push(current);
+    }
+    return sizedBatches;
+}
 function createCapturedValueEntry(params) {
     if (!__FULL_VALUE_CAPTURE_ENABLED)
         return undefined;
@@ -4109,7 +4160,7 @@ function reproMiddleware(cfg) {
                             const orderedEvents = TRACE_ORDER_MODE === 'tree'
                                 ? reorderTraceEvents(baseEvents)
                                 : sortTraceEventsChronologically(baseEvents);
-                            const traceBatches = chunkArray(orderedEvents, TRACE_BATCH_SIZE);
+                            const traceBatches = chunkTraceEventsForTransport(orderedEvents, rid, aid);
                             if (traceBatches.length) {
                                 for (let i = 0; i < traceBatches.length; i++) {
                                     const batch = traceBatches[i];
@@ -5312,7 +5363,7 @@ function buildKafkaTraceEntries(actionId, requestRid, events) {
     const orderedEvents = TRACE_ORDER_MODE === 'tree'
         ? reorderTraceEvents(baseEvents)
         : sortTraceEventsChronologically(baseEvents);
-    const batches = chunkArray(orderedEvents, TRACE_BATCH_SIZE);
+    const batches = chunkTraceEventsForTransport(orderedEvents, requestRid, actionId);
     return batches.map((batch, index) => ({
         actionId: actionId ?? null,
         trace: batch,
@@ -6626,4 +6677,6 @@ exports.__reproTestHooks = {
     materializeInlinePrivacyValueAsyncForTest: materializeInlinePrivacyValueAsync,
     patchKafkaProducerInstanceForTest: patchKafkaProducerInstance,
     wrapKafkaEachMessageHandlerForTest: wrapKafkaEachMessageHandler,
+    chunkTraceEventsForTransportForTest: chunkTraceEventsForTransport,
+    estimateTraceBatchSerializedBytesForTest: estimateTraceBatchSerializedBytes,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "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 -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 -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

@@ -2028,6 +2028,11 @@ const TRACE_INLINE_VALUE_MAX_SERIALIZED_CHARS = (() => {
 })();
 const TRACE_VALUE_SIZE_EXCEEDED = Symbol('trace-value-size-exceeded');
 const TRACE_BATCH_SIZE = 100;
+const TRACE_BATCH_MAX_SERIALIZED_BYTES = (() => {
+    const env = Number(process.env.REPRO_TRACE_BATCH_MAX_SERIALIZED_BYTES);
+    if (Number.isFinite(env) && env > 0) return Math.trunc(env);
+    return 512 * 1024;
+})();
 const TRACE_FLUSH_DELAY_MS = 20;
 // Choose how to order trace events in payloads.
 //  - "chronological" (default): preserve event arrival order (no reshuffle).
@@ -4215,6 +4220,74 @@ function collectBatchTraceValueEntries(batch: TraceEventRecord[], batchIndex: nu
     return collected;
 }
 
+function serializedByteLength(value: any): number {
+    try {
+        return Buffer.byteLength(JSON.stringify(value), 'utf8');
+    } catch {
+        return Number.MAX_SAFE_INTEGER;
+    }
+}
+
+function estimateTraceBatchSerializedBytes(
+    batch: TraceEventRecord[],
+    requestRid: string,
+    actionId: string | null | undefined,
+    batchIndex: number,
+): number {
+    const traceValues = collectBatchTraceValueEntries(batch, batchIndex);
+    return serializedByteLength({
+        actionId: actionId ?? null,
+        trace: batch,
+        traceValues: traceValues.length ? traceValues : undefined,
+        traceBatch: {
+            rid: requestRid,
+            index: batchIndex,
+            total: 0,
+        },
+        t: 0,
+    });
+}
+
+function chunkTraceEventsForTransport(
+    events: TraceEventRecord[],
+    requestRid: string,
+    actionId: string | null | undefined,
+): TraceEventRecord[][] {
+    if (!Array.isArray(events) || events.length === 0) return [];
+
+    const countBatches = chunkArray(events, TRACE_BATCH_SIZE);
+    const sizedBatches: TraceEventRecord[][] = [];
+    let current: TraceEventRecord[] = [];
+
+    for (const event of countBatches.flat()) {
+        const candidate = current.concat(event);
+        const batchIndex = sizedBatches.length;
+        const estimatedBytes = estimateTraceBatchSerializedBytes(
+            candidate,
+            requestRid,
+            actionId,
+            batchIndex,
+        );
+
+        if (
+            current.length > 0 &&
+            estimatedBytes > TRACE_BATCH_MAX_SERIALIZED_BYTES
+        ) {
+            sizedBatches.push(current);
+            current = [event];
+            continue;
+        }
+
+        current = candidate;
+    }
+
+    if (current.length > 0) {
+        sizedBatches.push(current);
+    }
+
+    return sizedBatches;
+}
+
 function createCapturedValueEntry(
     params: {
         target:
@@ -4953,7 +5026,11 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
                                 const orderedEvents = TRACE_ORDER_MODE === 'tree'
                                     ? reorderTraceEvents(baseEvents)
                                     : sortTraceEventsChronologically(baseEvents);
-                                const traceBatches = chunkArray(orderedEvents, TRACE_BATCH_SIZE);
+                                const traceBatches = chunkTraceEventsForTransport(
+                                    orderedEvents,
+                                    rid,
+                                    aid,
+                                );
 
                                 if (traceBatches.length) {
                                     for (let i = 0; i < traceBatches.length; i++) {
@@ -6253,7 +6330,11 @@ function buildKafkaTraceEntries(
         TRACE_ORDER_MODE === 'tree'
             ? reorderTraceEvents(baseEvents)
             : sortTraceEventsChronologically(baseEvents);
-    const batches = chunkArray(orderedEvents, TRACE_BATCH_SIZE);
+    const batches = chunkTraceEventsForTransport(
+        orderedEvents,
+        requestRid,
+        actionId,
+    );
     return batches.map((batch, index) => ({
         actionId: actionId ?? null,
         trace: batch,
@@ -7618,4 +7699,6 @@ export const __reproTestHooks = {
     materializeInlinePrivacyValueAsyncForTest: materializeInlinePrivacyValueAsync,
     patchKafkaProducerInstanceForTest: patchKafkaProducerInstance,
     wrapKafkaEachMessageHandlerForTest: wrapKafkaEachMessageHandler,
+    chunkTraceEventsForTransportForTest: chunkTraceEventsForTransport,
+    estimateTraceBatchSerializedBytesForTest: estimateTraceBatchSerializedBytes,
 };

package/test/trace-batch-size.test.js

@@ -0,0 +1,58 @@
+const assert = require('assert');
+const { __reproTestHooks } = require('../dist');
+
+function makeEvent(index, payload) {
+    return {
+        t: index,
+        type: 'enter',
+        fn: `fn${index}`,
+        file: '/app/src/controllers/subjects/index.ts',
+        line: index + 1,
+        depth: 1,
+        spanId: index + 1,
+        parentSpanId: null,
+        args: [payload],
+    };
+}
+
+function main() {
+    const payload = {
+        __kind: 'http-request',
+        method: 'POST',
+        url: '/api/v1/subject_visits/697001aaac70cc3d60f21273/subjects/createNewSubject?tenantId=tgtherapeutics',
+        headers: {
+            x_http_user: 'x'.repeat(12000),
+            authorization: '[dropped]',
+        },
+        body: {
+            value: 'y'.repeat(12000),
+        },
+    };
+
+    const events = Array.from({ length: 120 }, (_, index) => makeEvent(index, payload));
+    const batches = __reproTestHooks.chunkTraceEventsForTransportForTest(
+        events,
+        'RID_test_trace_batch_size',
+        'A_test_trace_batch_size',
+    );
+
+    assert(batches.length > 1, `expected more than one batch, got ${batches.length}`);
+
+    batches.forEach((batch, index) => {
+        const size = __reproTestHooks.estimateTraceBatchSerializedBytesForTest(
+            batch,
+            'RID_test_trace_batch_size',
+            'A_test_trace_batch_size',
+            index,
+        );
+        assert(
+            size <= 512 * 1024,
+            `batch ${index} too large: ${size}`,
+        );
+    });
+
+    // eslint-disable-next-line no-console
+    console.log('trace batch size chunking OK');
+}
+
+main();