@reproapp/node-sdk 0.0.22 → 0.0.24

package/dist/index.js

@@ -1525,9 +1525,10 @@ function resolveCollectionOrWarn(source, type) {
 function post(cfg, sessionId, body, options) {
     try {
         const entries = Array.isArray(body?.entries) ? body.entries : [];
-        (0, client_1.enqueueIngestEntries)(cfg, sessionId, entries, options);
+        return (0, client_1.enqueueIngestEntries)(cfg, sessionId, entries, options);
     }
     catch { /* swallow in SDK */ }
+    return false;
 }
 function toIngestConfig(cfg) {
     return {
@@ -1712,6 +1713,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 +3779,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;
@@ -4897,9 +4919,6 @@ function emitDbQuery(cfg, sid, aid, payload) {
     const spanContext = payload?.spanContext ?? captureSpanContextFromTracer();
     if (!shouldCaptureDbSpan(spanContext))
         return;
-    if (ctx.captureScope) {
-        ctx.captureScope.dbChangeCount += 1;
-    }
     const runCtx = {
         sid: ctx.sid,
         aid: ctx.aid,
@@ -5018,7 +5037,7 @@ async function emitDbQueryAsync(cfg, sid, aid, payload) {
         resultMetaMaterialization: resultMetaMaterialization.skipped,
         errorMaterialization: errorMaterialization.skipped,
     }, spanContext);
-    post(cfg, sid, {
+    const enqueued = post(cfg, sid, {
         entries: [{
                 actionId: aid ?? null,
                 requestRid: ctx.rid ?? null,
@@ -5027,6 +5046,9 @@ async function emitDbQueryAsync(cfg, sid, aid, payload) {
                 t: payload.t,
             }]
     });
+    if (enqueued && ctx.captureScope) {
+        ctx.captureScope.dbChangeCount += 1;
+    }
 }
 function buildMinimalUpdate(before, after) {
     const set = {};

package/dist/ingest/client.d.ts

@@ -8,6 +8,6 @@ export declare const configureIngestQueueDrain: (options: {
 }) => void;
 export declare const kickIngestQueueDrain: () => void;
 export declare const postIngestEntries: (config: IngestClientConfig, sessionId: string, entries: LegacyEntry[]) => Promise<void>;
-export declare const enqueueIngestEntries: (config: IngestClientConfig, sessionId: string, entries: LegacyEntry[], options?: EnqueueIngestOptions) => void;
+export declare const enqueueIngestEntries: (config: IngestClientConfig, sessionId: string, entries: LegacyEntry[], options?: EnqueueIngestOptions) => boolean;
 export declare const flushIngestQueue: () => Promise<void>;
 export type { IngestClientConfig, LegacyEntry } from './types';

package/dist/ingest/client.js

@@ -173,7 +173,7 @@ const drainIngestQueue = async () => {
 };
 const enqueueIngestEntries = (config, sessionId, entries, options = {}) => {
     if (!sessionId || !Array.isArray(entries) || entries.length === 0) {
-        return;
+        return false;
     }
     const priority = options.priority === 'high' ? 'high' : 'normal';
     const batch = {
@@ -208,6 +208,7 @@ const enqueueIngestEntries = (config, sessionId, entries, options = {}) => {
         }
     }
     scheduleDrain(shouldDeferDrain?.() ? deferDrainMs : 0);
+    return ingestQueue.includes(batch);
 };
 exports.enqueueIngestEntries = enqueueIngestEntries;
 const flushIngestQueue = async () => {

package/package.json

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

package/src/index.ts

@@ -1921,11 +1921,12 @@ function post(
     sessionId: string,
     body: any,
     options?: EnqueueIngestOptions,
-) {
+): boolean {
     try {
         const entries = Array.isArray(body?.entries) ? (body.entries as LegacyEntry[]) : [];
-        enqueueIngestEntries(cfg, sessionId, entries, options);
+        return enqueueIngestEntries(cfg, sessionId, entries, options);
     } catch { /* swallow in SDK */ }
+    return false;
 }
 
 function toIngestConfig(cfg: IngestClientConfig): IngestClientConfig {
@@ -2116,6 +2117,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 +4510,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 +4533,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 +4544,10 @@ function chunkTraceEventsForTransport(
         current = candidate;
     }
 
-    if (current.length > 0) {
+    if (
+        current.length > 0 &&
+        sizedBatches.length < TRACE_MAX_BATCHES_PER_REQUEST
+    ) {
         sizedBatches.push(current);
     }
 
@@ -5837,9 +5859,6 @@ function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
     const ctx = getCtx() as Ctx;
     const spanContext = payload?.spanContext ?? captureSpanContextFromTracer();
     if (!shouldCaptureDbSpan(spanContext)) return;
-    if (ctx.captureScope) {
-        ctx.captureScope.dbChangeCount += 1;
-    }
     const runCtx: Ctx = {
         sid: ctx.sid,
         aid: ctx.aid,
@@ -5957,7 +5976,7 @@ async function emitDbQueryAsync(cfg: any, sid?: string, aid?: string, payload?:
         resultMetaMaterialization: resultMetaMaterialization.skipped,
         errorMaterialization: errorMaterialization.skipped,
     }, spanContext);
-    post(cfg, sid, {
+    const enqueued = post(cfg, sid, {
         entries: [{
             actionId: aid ?? null,
             requestRid: ctx.rid ?? null,
@@ -5966,6 +5985,9 @@ async function emitDbQueryAsync(cfg: any, sid?: string, aid?: string, payload?:
             t: payload.t,
         }]
     });
+    if (enqueued && ctx.captureScope) {
+        ctx.captureScope.dbChangeCount += 1;
+    }
 }
 
 function buildMinimalUpdate(before: any, after: any) {

package/src/ingest/client.ts

@@ -211,9 +211,9 @@ export const enqueueIngestEntries = (
     sessionId: string,
     entries: LegacyEntry[],
     options: EnqueueIngestOptions = {},
-): void => {
+): boolean => {
     if (!sessionId || !Array.isArray(entries) || entries.length === 0) {
-        return;
+        return false;
     }
     const priority = options.priority === 'high' ? 'high' : 'normal';
     const batch: QueuedIngestBatch = {
@@ -246,6 +246,7 @@ export const enqueueIngestEntries = (
         }
     }
     scheduleDrain(shouldDeferDrain?.() ? deferDrainMs : 0);
+    return ingestQueue.includes(batch);
 };
 
 export const flushIngestQueue = async (): Promise<void> => {

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');
 }