mirror of https://github.com/mongodb/mongo
88 lines
2.5 KiB
JavaScript
88 lines
2.5 KiB
JavaScript
/**
|
|
* Tests for basic functionality of the abort move collection feature.
|
|
*
|
|
* @tags: [
|
|
* requires_fcv_80,
|
|
* featureFlagMoveCollection,
|
|
* assumes_balancer_off,
|
|
* ]
|
|
*/
|
|
|
|
import {configureFailPoint} from "jstests/libs/fail_point_util.js";
|
|
import {funWithArgs} from "jstests/libs/parallel_shell_helpers.js";
|
|
import {ShardingTest} from "jstests/libs/shardingtest.js";
|
|
|
|
(function () {
|
|
"use strict";
|
|
|
|
let st = new ShardingTest({mongos: 1, shards: 2});
|
|
|
|
const dbName = "db";
|
|
const collName = "foo";
|
|
const ns = dbName + "." + collName;
|
|
let mongos = st.s0;
|
|
let shard0 = st.shard0.shardName;
|
|
let shard1 = st.shard1.shardName;
|
|
|
|
assert.commandWorked(st.s.adminCommand({enableSharding: dbName, primaryShard: shard0}));
|
|
|
|
const coll = mongos.getDB(dbName)[collName];
|
|
for (let i = -5; i < 5; ++i) {
|
|
assert.commandWorked(coll.insert({oldKey: i}));
|
|
}
|
|
|
|
let failpoint = configureFailPoint(st.rs1.getPrimary(), "reshardingPauseRecipientDuringCloning");
|
|
|
|
// Starting the parallel shell for moveCollectionCmd
|
|
const awaitResult = startParallelShell(
|
|
funWithArgs(
|
|
function (ns, toShardId) {
|
|
assert.commandFailedWithCode(
|
|
db.adminCommand({moveCollection: ns, toShard: toShardId}),
|
|
ErrorCodes.ReshardCollectionAborted,
|
|
);
|
|
},
|
|
ns,
|
|
shard1,
|
|
),
|
|
st.s.port,
|
|
);
|
|
|
|
// Waiting to reach failpoint
|
|
failpoint.wait();
|
|
|
|
// Verify that the provenance field is appended to the currentOp
|
|
const filter = {
|
|
type: "op",
|
|
"originatingCommand.reshardCollection": ns,
|
|
"provenance": "moveCollection",
|
|
};
|
|
assert.soon(() => {
|
|
return (
|
|
st.s
|
|
.getDB("admin")
|
|
.aggregate([{$currentOp: {allUsers: true, localOps: false}}, {$match: filter}])
|
|
.toArray().length >= 1
|
|
);
|
|
});
|
|
|
|
// Calling abortMoveCollection
|
|
assert.commandWorked(mongos.adminCommand({abortMoveCollection: ns}));
|
|
|
|
// Waiting for parallel shell to be finished
|
|
failpoint.off();
|
|
awaitResult();
|
|
|
|
const metrics = st.config0.getDB("admin").serverStatus({}).shardingStatistics.moveCollection;
|
|
|
|
assert.eq(metrics.countStarted, 1);
|
|
assert.eq(metrics.countSucceeded, 0);
|
|
assert.eq(metrics.countFailed, 0);
|
|
assert.eq(metrics.countCanceled, 1);
|
|
|
|
assert.eq(0, st.rs1.getPrimary().getCollection(ns).countDocuments({}));
|
|
assert.eq(10, st.rs0.getPrimary().getCollection(ns).countDocuments({}));
|
|
|
|
st.stop();
|
|
})();
|