SERVER-112121 Test that async oplog sampling doesn't block rollback-to-stable (#42550)

GitOrigin-RevId: dfd824bdd5503faa1649ab18cde53d6e2e4e9912

jstests/noPassthrough/oplog/rollback_async_oplog_sampling.js

@@ -0,0 +1,87 @@
+/**
+ * Tests that rollback interrupts the oplog cap maintainer thread, and that it restarts after
+ * rollback is complete.
+ *
+ * @tags: [
+ *      requires_persistence,
+ *      requires_wiredtiger,
+ * ]
+ */
+import {kDefaultWaitForFailPointTimeout} from "jstests/libs/fail_point_util.js";
+import {ReplSetTest} from "jstests/libs/replsettest.js";
+import {RollbackTest} from "jstests/replsets/libs/rollback_test.js";
+
+const rst = new ReplSetTest({
+    name: jsTestName(),
+    nodes: [{}, {}, {rsConfig: {priority: 0}}],
+    useBridge: true,
+    nodeOptions: {
+        setParameter: {
+            "oplogSamplingAsyncEnabled": true,
+            "collectionSamplingLogIntervalSeconds": 1,
+            "failpoint.hangBeforeOplogSampling": tojson({mode: "alwaysOn"}),
+        },
+    },
+    settings: {chainingAllowed: false},
+});
+rst.startSet();
+rst.initiate();
+
+const rollbackTest = new RollbackTest(jsTestName(), rst);
+
+const primary = rollbackTest.getPrimary();
+const testDb = primary.getDB("test");
+const coll = testDb.getCollection(jsTestName());
+
+// Wait for the oplog cap maintainer thread to start and hang.
+assert.commandWorked(
+    primary.adminCommand({
+        waitForFailPoint: "hangBeforeOplogSampling",
+        timesEntered: 1,
+        maxTimeMS: kDefaultWaitForFailPointTimeout,
+    }),
+);
+
+// Populate the nodes with some data.
+const bulk = coll.initializeUnorderedBulkOp();
+for (let i = 0; i < 1000; i++) {
+    bulk.insert({x: i});
+}
+assert.commandWorked(bulk.execute());
+
+rollbackTest.awaitLastOpCommitted();
+rollbackTest.transitionToRollbackOperations();
+
+let interruptCount = assert.commandWorked(primary.getDB("admin").runCommand({serverStatus: 1}))
+                         .oplogTruncation.interruptCount;
+assert(interruptCount == 0);
+
+assert.commandWorked(coll.insert({x: 1000}));
+
+rollbackTest.transitionToSyncSourceOperationsBeforeRollback();
+rollbackTest.transitionToSyncSourceOperationsDuringRollback();
+
+// OplogCapMaintainerThread interrupted.
+checkLog.containsJson(primary, 11212201);
+
+// OplogCapMaintainerThread shutdown.
+checkLog.containsJson(primary, 7474901);
+
+interruptCount = assert.commandWorked(primary.getDB("admin").runCommand({serverStatus: 1}))
+                     .oplogTruncation.interruptCount;
+assert(interruptCount == 1);
+
+// OplogCapMaintainerThread never finished scanning up to this point.
+assert(checkLog.checkContainsWithCountJson(primary, 22382, {}, /*expectedCount=*/ 0));
+
+// The OplogCapMaintainerThread gets restarted at the end of FCBIS, so turn off the fail point.
+assert.commandWorked(primary.getDB("admin").adminCommand(
+    {configureFailPoint: "hangBeforeOplogSampling", mode: "off"}));
+
+// OplogCapMaintainerThread finishes scanning the oplog exactly once.
+assert.soon(() => {
+    return checkLog.checkContainsWithCountJson(primary, 22382, {}, /*expectedCount=*/ 1);
+});
+
+rollbackTest.transitionToSteadyStateOperations();
+rollbackTest.stop();