1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
class BasicServerlessTest {
constructor({recipientTagName, recipientSetName, nodeOptions}) {
this.donor = new ReplSetTest({name: "donor", nodes: 3, serverless: true, nodeOptions});
this.donor.startSet();
this.donor.initiate();
this.recipientTagName = recipientTagName;
this.recipientSetName = recipientSetName;
this.recipientNodes = [];
}
stop() {
this.removeAndStopRecipientNodes();
// If we validate, it will try to list all collections and the migrated collections will
// return a TenantMigrationCommitted error.
this.donor.stopSet(undefined /* signal */, false /* forRestart */, {skipValidation: 1});
}
addRecipientNodes(numNodes) {
numNodes = numNodes || 3; // default to three nodes
if (this.recipientNodes.lengh > 0) {
throw new Error("Recipient nodes may only be added once");
}
jsTestLog(`Adding ${numNodes} non-voting recipient nodes to donor`);
const donor = this.donor;
for (let i = 0; i < numNodes; ++i) {
this.recipientNodes.push(donor.add());
}
const primary = donor.getPrimary();
const admin = primary.getDB('admin');
const config = donor.getReplSetConfigFromNode();
config.version++;
// ensure recipient nodes are added as non-voting members
this.recipientNodes.forEach(node => {
config.members.push({
host: node.host,
votes: 0,
priority: 0,
tags: {[this.recipientTagName]: ObjectId().valueOf()}
});
});
// reindex all members from 0
config.members = config.members.map((member, idx) => {
member._id = idx;
return member;
});
assert.commandWorked(admin.runCommand({replSetReconfig: config}));
this.recipientNodes.forEach(node => donor.waitForState(node, ReplSetTest.State.SECONDARY));
}
removeAndStopRecipientNodes() {
print("Removing and stopping recipient nodes");
this.recipientNodes.forEach(node => this.donor.remove(node));
}
/**
* Crafts a tenant database name.
*/
tenantDB(tenantId, dbName) {
return `${tenantId}_${dbName}`;
}
}
function findMigration(primary, uuid) {
const donorsCollection = primary.getDB("config").getCollection("tenantSplitDonors");
return donorsCollection.findOne({"_id": uuid});
}
function cleanupMigrationDocument(primary, uuid) {
const donorsCollection = primary.getDB("config").getCollection("tenantSplitDonors");
return donorsCollection.deleteOne({"_id": uuid}, {w: "majority"});
}
function assertMigrationState(primary, uuid, state) {
const migrationDoc = findMigration(primary, uuid);
assert(migrationDoc);
if (migrationDoc.state === 'aborted') {
print(tojson(migrationDoc));
}
assert.eq(migrationDoc.state, state);
}
|
