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
|
/**
* Tests running the update command on a time-series collection closes any in-memory buckets that
* were updated.
*
* @tags: [
* # This test depends on certain writes ending up in the same bucket. Stepdowns may result in
* # writes splitting between two primaries, and thus different buckets.
* does_not_support_stepdowns,
* # Test explicitly relies on multi-updates.
* requires_multi_updates,
* # We need a timeseries collection.
* requires_timeseries,
* # This test depends on stats read from the primary node in replica sets.
* assumes_read_preference_unchanged,
* ]
*/
(function() {
"use strict";
load("jstests/core/timeseries/libs/timeseries.js");
load("jstests/libs/fixture_helpers.js");
if (!TimeseriesTest.timeseriesUpdatesAndDeletesEnabled(db.getMongo())) {
jsTestLog("Skipping test because the time-series updates and deletes feature flag is disabled");
return;
}
if (FixtureHelpers.isMongos(db) &&
!TimeseriesTest.shardedtimeseriesCollectionsEnabled(db.getMongo())) {
jsTestLog("Skipping test because the time-series updates and deletes feature flag is disabled");
return;
}
if (FixtureHelpers.isMongos(db) &&
!TimeseriesTest.shardedTimeseriesUpdatesAndDeletesEnabled(db.getMongo())) {
jsTestLog(
"Skipping test because the sharded time-series updates and deletes feature flag is disabled");
return;
}
TimeseriesTest.run((insert) => {
const testDB = db.getSiblingDB(jsTestName());
assert.commandWorked(testDB.dropDatabase());
const coll = testDB["t"];
const bucketsColl = testDB["system.buckets." + coll.getName()];
const timeFieldName = "time";
const metaFieldName = "m";
const docs = [
{[timeFieldName]: ISODate("2021-01-01T01:00:00Z"), [metaFieldName]: "a"},
{[timeFieldName]: ISODate("2021-01-01T01:01:00Z"), [metaFieldName]: "a"},
{[timeFieldName]: ISODate("2021-01-01T01:02:00Z"), [metaFieldName]: "b"},
];
assert.commandWorked(testDB.createCollection(
coll.getName(), {timeseries: {timeField: timeFieldName, metaField: metaFieldName}}));
assert.commandWorked(insert(coll, docs[0]));
assert.commandWorked(testDB.runCommand({
update: coll.getName(),
updates: [{q: {[metaFieldName]: "a"}, u: {$set: {[metaFieldName]: "b"}}, multi: true}]
}));
docs[0][metaFieldName] = "b";
let stats = assert.commandWorked(coll.stats());
assert(stats.timeseries);
const expectedNumBucketsReopened = stats.timeseries['numBucketsReopened'] + 1;
assert.commandWorked(insert(coll, docs.slice(1)));
assert.docEq(coll.find({}, {_id: 0}).sort({[timeFieldName]: 1}).toArray(), docs);
if (TimeseriesTest.timeseriesScalabilityImprovementsEnabled(testDB)) {
assert.eq(bucketsColl.find().itcount(), 2, bucketsColl.find().toArray());
stats = assert.commandWorked(coll.stats());
assert(stats.timeseries);
assert.eq(stats.timeseries['bucketCount'], 2);
assert.eq(stats.timeseries['numBucketsReopened'], expectedNumBucketsReopened);
} else {
// All three documents should be in their own buckets.
assert.eq(bucketsColl.find().itcount(), 3, bucketsColl.find().toArray());
}
});
})();
|