@reproapp/node-sdk 0.0.22 → 0.0.23

package/dist/index.js

@@ -1712,6 +1712,18 @@ const TRACE_BATCH_MAX_SERIALIZED_BYTES = (() => {
         return Math.trunc(env);
     return 512 * 1024;
 })();
+const TRACE_MAX_EVENTS_PER_REQUEST = (() => {
+    const env = Number(process.env.REPRO_TRACE_MAX_EVENTS_PER_REQUEST);
+    if (Number.isFinite(env) && env > 0)
+        return Math.trunc(env);
+    return 1000;
+})();
+const TRACE_MAX_BATCHES_PER_REQUEST = (() => {
+    const env = Number(process.env.REPRO_TRACE_MAX_BATCHES_PER_REQUEST);
+    if (Number.isFinite(env) && env > 0)
+        return Math.trunc(env);
+    return 8;
+})();
 const TRACE_FLUSH_DELAY_MS = 20;
 // Choose how to order trace events in payloads.
 //  - "chronological" (default): preserve event arrival order (no reshuffle).
@@ -3766,22 +3778,31 @@ function estimateTraceBatchSerializedBytes(batch, requestRid, actionId, batchInd
 function chunkTraceEventsForTransport(events, requestRid, actionId) {
     if (!Array.isArray(events) || events.length === 0)
         return [];
-    const countBatches = chunkArray(events, TRACE_BATCH_SIZE);
+    const boundedEvents = events.slice(0, TRACE_MAX_EVENTS_PER_REQUEST);
+    const countBatches = chunkArray(boundedEvents, TRACE_BATCH_SIZE);
     const sizedBatches = [];
     let current = [];
     for (const event of countBatches.flat()) {
+        if (sizedBatches.length >= TRACE_MAX_BATCHES_PER_REQUEST) {
+            break;
+        }
         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);
+            if (sizedBatches.length >= TRACE_MAX_BATCHES_PER_REQUEST) {
+                current = [];
+                break;
+            }
             current = [event];
             continue;
         }
         current = candidate;
     }
-    if (current.length > 0) {
+    if (current.length > 0 &&
+        sizedBatches.length < TRACE_MAX_BATCHES_PER_REQUEST) {
         sizedBatches.push(current);
     }
     return sizedBatches;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.22",
+  "version": "0.0.23",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -2116,6 +2116,16 @@ const TRACE_BATCH_MAX_SERIALIZED_BYTES = (() => {
     if (Number.isFinite(env) && env > 0) return Math.trunc(env);
     return 512 * 1024;
 })();
+const TRACE_MAX_EVENTS_PER_REQUEST = (() => {
+    const env = Number(process.env.REPRO_TRACE_MAX_EVENTS_PER_REQUEST);
+    if (Number.isFinite(env) && env > 0) return Math.trunc(env);
+    return 1000;
+})();
+const TRACE_MAX_BATCHES_PER_REQUEST = (() => {
+    const env = Number(process.env.REPRO_TRACE_MAX_BATCHES_PER_REQUEST);
+    if (Number.isFinite(env) && env > 0) return Math.trunc(env);
+    return 8;
+})();
 const TRACE_FLUSH_DELAY_MS = 20;
 // Choose how to order trace events in payloads.
 //  - "chronological" (default): preserve event arrival order (no reshuffle).
@@ -4499,11 +4509,15 @@ function chunkTraceEventsForTransport(
 ): TraceEventRecord[][] {
     if (!Array.isArray(events) || events.length === 0) return [];
 
-    const countBatches = chunkArray(events, TRACE_BATCH_SIZE);
+    const boundedEvents = events.slice(0, TRACE_MAX_EVENTS_PER_REQUEST);
+    const countBatches = chunkArray(boundedEvents, TRACE_BATCH_SIZE);
     const sizedBatches: TraceEventRecord[][] = [];
     let current: TraceEventRecord[] = [];
 
     for (const event of countBatches.flat()) {
+        if (sizedBatches.length >= TRACE_MAX_BATCHES_PER_REQUEST) {
+            break;
+        }
         const candidate = current.concat(event);
         const batchIndex = sizedBatches.length;
         const estimatedBytes = estimateTraceBatchSerializedBytes(
@@ -4518,6 +4532,10 @@ function chunkTraceEventsForTransport(
             estimatedBytes > TRACE_BATCH_MAX_SERIALIZED_BYTES
         ) {
             sizedBatches.push(current);
+            if (sizedBatches.length >= TRACE_MAX_BATCHES_PER_REQUEST) {
+                current = [];
+                break;
+            }
             current = [event];
             continue;
         }
@@ -4525,7 +4543,10 @@ function chunkTraceEventsForTransport(
         current = candidate;
     }
 
-    if (current.length > 0) {
+    if (
+        current.length > 0 &&
+        sizedBatches.length < TRACE_MAX_BATCHES_PER_REQUEST
+    ) {
         sizedBatches.push(current);
     }
 

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

@@ -37,6 +37,10 @@ function main() {
     );
 
     assert(batches.length > 1, `expected more than one batch, got ${batches.length}`);
+    assert(
+        batches.length <= 8,
+        `expected trace transport to cap batches per request, got ${batches.length}`,
+    );
 
     batches.forEach((batch, index) => {
         const size = __reproTestHooks.estimateTraceBatchSerializedBytesForTest(
@@ -51,6 +55,18 @@ function main() {
         );
     });
 
+    const manyEvents = Array.from({ length: 1500 }, (_, index) => makeEvent(index, 'tiny'));
+    const cappedBatches = __reproTestHooks.chunkTraceEventsForTransportForTest(
+        manyEvents,
+        'RID_test_trace_event_cap',
+        'A_test_trace_event_cap',
+    );
+    const cappedEventCount = cappedBatches.reduce((sum, batch) => sum + batch.length, 0);
+    assert(
+        cappedEventCount <= 1000,
+        `expected trace transport to cap events per request, got ${cappedEventCount}`,
+    );
+
     // eslint-disable-next-line no-console
     console.log('trace batch size chunking OK');
 }