@reproapp/node-sdk 0.0.34 → 0.0.35

package/dist/index.js

@@ -3003,9 +3003,12 @@ function limitRawInlinePrivacyValue(target, value) {
     return limited.skipped ? limited : null;
 }
 function materializeInlinePrivacyValue(target, value, cfg, req, trace, masking, privacy, db) {
-    if (value === undefined || shouldCapturePrivacyRaw(cfg)) {
+    if (value === undefined) {
         return { value };
     }
+    if (shouldCapturePrivacyRaw(cfg)) {
+        return { value: limitMaterializedTraceInlineValue(value) };
+    }
     try {
         const valueAfterPrivacy = applyPrivacyThenMask(target, value, cfg, req, trace, masking, privacy, db);
         return { value: valueAfterPrivacy };
@@ -3022,9 +3025,12 @@ function materializeInlinePrivacyValue(target, value, cfg, req, trace, masking,
     }
 }
 async function materializeInlinePrivacyValueAsync(target, value, cfg, req, trace, masking, privacy, db) {
-    if (value === undefined || shouldCapturePrivacyRaw(cfg)) {
+    if (value === undefined) {
         return { value };
     }
+    if (shouldCapturePrivacyRaw(cfg)) {
+        return { value: limitMaterializedTraceInlineValue(value) };
+    }
     try {
         const valueAfterPrivacy = await applyPrivacyThenMaskAsync(target, value, cfg, req, trace, masking, privacy, db);
         return { value: valueAfterPrivacy };
@@ -5197,9 +5203,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,
@@ -5213,7 +5216,7 @@ function emitDbQuery(cfg, sid, aid, payload) {
         await als.run(runCtx, async () => {
             await emitDbQueryAsync(cfg, sid, aid, payload).catch(() => undefined);
         });
-    });
+    }, { priority: 'high' });
 }
 async function emitDbQueryAsync(cfg, sid, aid, payload) {
     if (!sid)
@@ -5327,6 +5330,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/package.json

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

package/src/index.ts

@@ -3610,9 +3610,12 @@ function materializeInlinePrivacyValue(
     privacy: NormalizedRuntimePrivacyPolicy | null,
     db?: RuntimePrivacyDbContext | null,
 ): InlinePrivacyMaterializationResult {
-    if (value === undefined || shouldCapturePrivacyRaw(cfg)) {
+    if (value === undefined) {
         return { value };
     }
+    if (shouldCapturePrivacyRaw(cfg)) {
+        return { value: limitMaterializedTraceInlineValue(value) };
+    }
 
     try {
         const valueAfterPrivacy = applyPrivacyThenMask(target, value, cfg, req, trace, masking, privacy, db);
@@ -3639,9 +3642,12 @@ async function materializeInlinePrivacyValueAsync(
     privacy: NormalizedRuntimePrivacyPolicy | null,
     db?: RuntimePrivacyDbContext | null,
 ): Promise<InlinePrivacyMaterializationResult> {
-    if (value === undefined || shouldCapturePrivacyRaw(cfg)) {
+    if (value === undefined) {
         return { value };
     }
+    if (shouldCapturePrivacyRaw(cfg)) {
+        return { value: limitMaterializedTraceInlineValue(value) };
+    }
 
     try {
         const valueAfterPrivacy = await applyPrivacyThenMaskAsync(target, value, cfg, req, trace, masking, privacy, db);
@@ -6154,9 +6160,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,
@@ -6170,7 +6173,7 @@ function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
         await als.run(runCtx, async () => {
             await emitDbQueryAsync(cfg, sid, aid, payload).catch(() => undefined);
         });
-    });
+    }, { priority: 'high' });
 }
 
 async function emitDbQueryAsync(cfg: any, sid?: string, aid?: string, payload?: any) {
@@ -6283,6 +6286,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/test/mongodb-integration.test.js

@@ -389,6 +389,68 @@ async function testNativeInsertOneIsCaptured() {
     });
 }
 
+async function testCaptureRawNativeInsertOneUsesJsonSafeTransportValues() {
+    await withCapturedFetch(async (capturedBodies) => {
+        __reproTestHooks.resetSdkBackgroundQueuesForTest();
+        const sessionId = 'S_mongo_insert_raw_complex';
+        const actionId = 'A_mongo_insert_raw_complex';
+        const cfg = {
+            ...makeConfig(sessionId, actionId),
+            privacy: { environment: 'dev', captureRaw: true },
+        };
+
+        class FakeCollection {
+            constructor() {
+                this.collectionName = 'subjects';
+                this.dbName = 'clinical';
+            }
+
+            async insertOne(doc, options) {
+                assert.strictEqual(options.session.id, 'native-raw-complex-session-1');
+                doc._id = doc._id ?? 'subject-raw-complex-1';
+                return { acknowledged: true, insertedId: doc._id };
+            }
+        }
+
+        __reproTestHooks.patchMongoCollectionPrototypeForTest(FakeCollection.prototype, cfg);
+        const collection = new FakeCollection();
+        const doc = {
+            subjectNumber: '10013039',
+            status: 'screened',
+            counter: BigInt(42),
+        };
+        doc.self = doc;
+
+        await runCapturedRequest(cfg, sessionId, actionId, async () => {
+            await collection.insertOne(doc, {
+                session: {
+                    id: 'native-raw-complex-session-1',
+                    hasEnded: false,
+                    inTransaction: () => true,
+                },
+            });
+        });
+
+        await waitFor(() => dbChanges(flattenEvents(capturedBodies)).length === 1);
+        await waitFor(() => requestCaptureManifests(flattenEvents(capturedBodies)).length === 1);
+        await drainSdk();
+
+        const [change] = dbChanges(flattenEvents(capturedBodies));
+        const [manifest] = requestCaptureManifests(flattenEvents(capturedBodies));
+        assert.strictEqual(change.type, 'mongo');
+        assert.strictEqual(change.collection, 'subjects');
+        assert.strictEqual(change.op, 'insertOne');
+        assert.strictEqual(change.query.driver, 'mongodb');
+        assert.strictEqual(change.query.doc.subjectNumber, '10013039');
+        assert.strictEqual(change.query.doc.counter, '42');
+        assert.deepStrictEqual(change.query.doc.self, { __reproCircularRef: '$.doc' });
+        assert.strictEqual(change.after.subjectNumber, '10013039');
+        assert.strictEqual(change.after.counter, '42');
+        assert.deepStrictEqual(change.after.self, { __reproCircularRef: '$' });
+        assert.strictEqual(manifest.expected.dbChange, 1);
+    });
+}
+
 async function testNativeUpdateOneUpsertIsCaptured() {
     await withCapturedFetch(async (capturedBodies) => {
         __reproTestHooks.resetSdkBackgroundQueuesForTest();
@@ -535,6 +597,7 @@ async function testNativeOperationFailureIsCapturedAndRethrown() {
     await testReproMiddlewarePatchesPreloadedNativeMongoClientInsertOne();
     await testMongoClientDbCollectionChainIsPatchedWithoutCollectionExport();
     await testNativeInsertOneIsCaptured();
+    await testCaptureRawNativeInsertOneUsesJsonSafeTransportValues();
     await testNativeUpdateOneUpsertIsCaptured();
     await testNativeFindCursorToArrayIsCaptured();
     await testNativeOperationFailureIsCapturedAndRethrown();