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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/**
* Tests resource consumption metrics for retryable writes.
* Retryable writes persist transaction documents in "config.transaction" and all reads and writes
* to "config.transaction" should not trigger any operation metrics.
*
* @tags: [
* requires_replication,
* ]
*/
(function() {
"use strict";
load("jstests/libs/retryable_writes_util.js");
function setupReplicaSet() {
var rst = new ReplSetTest({
nodes: 2,
nodeOptions: {setParameter: {"aggregateOperationResourceConsumptionMetrics": true}}
});
rst.startSet();
rst.initiate();
return rst;
}
function clearMetrics(conn) {
conn.getDB('admin').aggregate([{$operationMetrics: {clearMetrics: true}}]);
}
function getMetrics(conn) {
const cursor = conn.getDB('admin').aggregate([{$operationMetrics: {}}]);
let allMetrics = {};
while (cursor.hasNext()) {
let doc = cursor.next();
allMetrics[doc.db] = doc;
}
return allMetrics;
}
function assertMetricsZeroRead(metrics, dbName) {
assert(metrics[dbName]);
assert.eq(metrics[dbName].primaryMetrics.docBytesRead, 0);
assert.eq(metrics[dbName].primaryMetrics.docUnitsRead, 0);
assert.eq(metrics[dbName].primaryMetrics.idxEntryBytesRead, 0);
assert.eq(metrics[dbName].primaryMetrics.idxEntryUnitsRead, 0);
}
function assertMetricsWritten(metrics, dbName, expectedWritten) {
assert(metrics[dbName]);
assert.eq(metrics[dbName].docBytesWritten, expectedWritten.docBytesWritten);
assert.eq(metrics[dbName].docUnitsWritten, expectedWritten.docUnitsWritten);
assert.eq(metrics[dbName].idxEntryBytesWritten, expectedWritten.idxEntryBytesWritten);
assert.eq(metrics[dbName].idxEntryUnitsWritten, expectedWritten.idxEntryUnitsWritten);
assert.eq(metrics[dbName].totalUnitsWritten, expectedWritten.totalUnitsWritten);
}
function runRetryableTransaction(sessionDb, txnNumber, cmdObj) {
let cmd = Object.assign(
{}, cmdObj, {txnNumber: NumberLong(txnNumber), startTransaction: true, autocommit: false});
assert.commandWorked(sessionDb.runCommand(cmd));
assert.commandWorked(sessionDb.adminCommand({
commitTransaction: 1,
txnNumber: NumberLong(txnNumber),
autocommit: false,
writeConcern: {w: "majority"}
}));
}
function runRetryableWriteCmd(sessionDb, txnNumber, cmdObj) {
let cmd = Object.assign({}, cmdObj, {txnNumber: NumberLong(txnNumber)});
jsTest.log("run retryable write with command :" + tojson(cmd));
assert.commandWorked(sessionDb.runCommand(cmd));
}
const rst = setupReplicaSet();
const primary = rst.getPrimary();
const kDbName = "testDb";
const kCollName = "testColl";
const testDb = primary.getDB(kDbName);
assert.commandWorked(testDb.createCollection(kCollName));
const session = testDb.getMongo().startSession();
let sessionDB = session.getDatabase(kDbName);
let txnNumber = 0;
let makeInsertCmdObj = (docs) => {
return {insert: kCollName, documents: docs, ordered: false};
};
let makeDocs = (fromId, toId) => {
let docs = [];
for (let i = fromId; i <= toId; i++) {
docs.push({_id: i, number: i});
}
return docs;
};
let expectedWritten;
jsTest.log("Tests non-retryable insert comamnd which has no transaction number.");
{
clearMetrics(primary);
let cmdObj = makeInsertCmdObj(makeDocs(1, 3));
assert.commandWorked(testDb.runCommand(cmdObj));
let metrics = getMetrics(primary);
assertMetricsZeroRead(metrics, kDbName);
// Init the expected doc, index and total data written. The following retryable writes
// should have the same output as the size of inserted user data is same.
expectedWritten = {
docBytesWritten: metrics[kDbName].docBytesWritten,
docUnitsWritten: metrics[kDbName].docUnitsWritten,
idxEntryBytesWritten: metrics[kDbName].idxEntryBytesWritten,
idxEntryUnitsWritten: metrics[kDbName].idxEntryUnitsWritten,
totalUnitsWritten: metrics[kDbName].totalUnitsWritten
};
}
jsTest.log("Tests retryable commitTransaction command which inserts documents.");
{
clearMetrics(primary);
let cmdObj = makeInsertCmdObj(makeDocs(4, 6));
runRetryableTransaction(sessionDB, txnNumber, cmdObj);
let metrics = getMetrics(primary);
assertMetricsZeroRead(metrics, kDbName);
assertMetricsWritten(metrics, kDbName, expectedWritten);
}
txnNumber++;
jsTest.log("Tests retryable insert command.");
{
clearMetrics(primary);
let cmdObj = makeInsertCmdObj(makeDocs(7, 9));
runRetryableWriteCmd(sessionDB, txnNumber, cmdObj);
let metrics = getMetrics(primary);
assertMetricsZeroRead(metrics, kDbName);
assertMetricsWritten(metrics, kDbName, expectedWritten);
}
session.endSession();
rst.stopSet();
}());
|
