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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/**
* Tests that prepared transactions can survive state transitions from SECONDARY to RECOVERING and
* RECOVERING to SECONDARY.
*
* @tags: [uses_transactions, uses_prepare_transaction]
*/
(function() {
"use strict";
load("jstests/core/txns/libs/prepare_helpers.js");
const replSet = new ReplSetTest({nodes: 2});
replSet.startSet();
replSet.initiate();
const primary = replSet.getPrimary();
const secondary = replSet.getSecondary();
const dbName = "test";
const collName = "prepare_transaction_survives_state_transitions_to_and_from_recovering";
const testDB = primary.getDB(dbName);
const testColl = testDB.getCollection(collName);
assert.commandWorked(testDB.runCommand({create: collName}));
assert.commandWorked(testColl.insert({_id: 1}));
const session1 = primary.startSession({causalConsistency: false});
const sessionDB1 = session1.getDatabase(dbName);
const sessionColl1 = sessionDB1.getCollection(collName);
const session2 = primary.startSession({causalConsistency: false});
const sessionDB2 = session2.getDatabase(dbName);
const sessionColl2 = sessionDB2.getCollection(collName);
jsTestLog("Starting a transaction and putting it into prepare");
session1.startTransaction();
assert.commandWorked(sessionColl1.update({_id: 1}, {_id: 1, a: 1}));
const prepareTimestamp1 = PrepareHelpers.prepareTransaction(session1);
jsTestLog("Starting a second transaction and putting it into prepare");
session2.startTransaction();
assert.commandWorked(sessionColl2.insert({_id: 2}));
const prepareTimestamp2 = PrepareHelpers.prepareTransaction(session2);
replSet.awaitReplication();
jsTestLog("Putting secondary in maintenance mode so it will transition to RECOVERING");
assert.commandWorked(secondary.adminCommand({replSetMaintenance: 1}));
replSet.waitForState(secondary, ReplSetTest.State.RECOVERING);
jsTestLog("Commiting the second prepared transaction while a node is in the RECOVERING state");
assert.commandWorked(PrepareHelpers.commitTransaction(session2, prepareTimestamp2));
replSet.awaitReplication();
jsTestLog("Taking secondary out of maintenance mode so it will transition back to SECONDARY");
assert.commandWorked(secondary.adminCommand({replSetMaintenance: 0}));
replSet.waitForState(secondary, ReplSetTest.State.SECONDARY);
jsTestLog("Stepping up the secondary");
replSet.stepUp(secondary);
replSet.waitForState(secondary, ReplSetTest.State.PRIMARY);
const newPrimary = replSet.getPrimary();
const newPrimaryDB = newPrimary.getDB(dbName);
// Create a proxy session to reuse the session state of the old primary.
const newSession = new _DelegatingDriverSession(newPrimary, session1);
jsTestLog("Make sure that the transaction is properly prepared");
// Make sure that we can't read changes to the document from the second transaction after
// recovery.
assert.eq(newPrimaryDB.getCollection(collName).find().toArray(), [{_id: 1}, {_id: 2}]);
// Make sure that another write on the same document from the second transaction causes a write
// conflict.
assert.commandFailedWithCode(
newPrimaryDB.runCommand(
{update: collName, updates: [{q: {_id: 1}, u: {$set: {a: 1}}}], maxTimeMS: 5 * 1000}),
ErrorCodes.MaxTimeMSExpired);
// Make sure that we cannot add other operations to the second transaction since it is prepared.
assert.commandFailedWithCode(
newSession.getDatabase(dbName).getCollection(collName).insert({_id: 3}),
ErrorCodes.PreparedTransactionInProgress);
jsTestLog("Verify that the locks from the prepared transaction are still held");
assert.commandFailedWithCode(newPrimaryDB.runCommand({drop: collName, maxTimeMS: 5 * 1000}),
ErrorCodes.MaxTimeMSExpired);
jsTestLog("Committing transaction");
assert.commandWorked(PrepareHelpers.commitTransaction(newSession, prepareTimestamp1));
replSet.awaitReplication();
replSet.stopSet();
}());
|
