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
|
// Ensure you can't shard on an array key
// @tags: [
// uses_multi_shard_transaction,
// uses_transactions,
// ]
(function() {
'use strict';
var st = new ShardingTest({shards: 3});
var mongos = st.s0;
const kDbName = 'TestDB';
const kCollName = 'Foo';
const kFullNs = kDbName + '.' + kCollName;
const session = st.s.startSession({retryWrites: true});
const sessionDB = session.getDatabase(kDbName);
let coll = mongos.getCollection(kFullNs);
st.shardColl(coll, {_id: 1, i: 1}, {_id: ObjectId(), i: 1});
printjson(mongos.getDB("config").chunks.find().toArray());
print("1: insert some invalid data");
var value = null;
// Insert an object with invalid array key
assert.commandFailedWithCode(coll.insert({i: [1, 2]}), ErrorCodes.ShardKeyNotFound);
// Insert an object with all the right fields, but an invalid array val for _id
assert.commandFailedWithCode(coll.insert({_id: [1, 2], i: 3}), ErrorCodes.ShardKeyNotFound);
// Insert an object with valid array key
assert.commandWorked(coll.insert({i: 1}));
// Update the value with valid other field
value = coll.findOne({i: 1});
assert.commandWorked(coll.update(value, {$set: {j: 2}}));
// Update the value with invalid other fields
value = coll.findOne({i: 1});
assert.commandFailedWithCode(sessionDB[kCollName].update(value, Object.merge(value, {i: [3]})),
ErrorCodes.NotSingleValueField);
// Multi-update the value with invalid other fields
value = coll.findOne({i: 1});
assert.commandFailedWithCode(coll.update(value, Object.merge(value, {i: [3, 4]}), false, true),
ErrorCodes.InvalidOptions);
// Multi-update the value with other fields (won't work, but no error)
value = coll.findOne({i: 1});
assert.commandWorked(coll.update(Object.merge(value, {i: [1, 1]}), {$set: {k: 4}}, false, true));
assert.docEq(coll.findOne({i: 1}, {_id: 0}), {i: 1, j: 2});
// Query the value with other fields (won't work, but no error)
value = coll.findOne({i: 1});
coll.find(Object.merge(value, {i: [1, 1]})).toArray();
// Can't remove using multikey, but shouldn't error
value = coll.findOne({i: 1});
coll.remove(Object.extend(value, {i: [1, 2, 3, 4]}));
// Can't remove using multikey, but shouldn't error
value = coll.findOne({i: 1});
assert.commandWorked(coll.remove(Object.extend(value, {i: [1, 2, 3, 4, 5]})));
assert.eq(coll.find().itcount(), 1);
value = coll.findOne({i: 1});
assert.commandWorked(coll.remove(Object.extend(value, {i: 1})));
assert.eq(coll.find().itcount(), 0);
coll.ensureIndex({_id: 1, i: 1, j: 1});
// Can insert document that will make index into a multi-key as long as it's not part of shard
// key.
coll.remove({});
assert.commandWorked(coll.insert({i: 1, j: [1, 2]}));
assert.eq(coll.find().itcount(), 1);
// Same is true for updates.
coll.remove({});
coll.insert({_id: 1, i: 1});
assert.commandWorked(coll.update({_id: 1, i: 1}, {_id: 1, i: 1, j: [1, 2]}));
assert.eq(coll.find().itcount(), 1);
// Same for upserts.
coll.remove({});
assert.commandWorked(coll.update({_id: 1, i: 1}, {_id: 1, i: 1, j: [1, 2]}, true));
assert.eq(coll.find().itcount(), 1);
jsTestLog("Testing dotted path shard key with array value inserts.");
const kDottedCollName = 'collDotted';
const kDottedFullNs = kDbName + '.' + kDottedCollName;
const dottedColl = mongos.getCollection(kDottedFullNs);
assert.commandWorked(mongos.adminCommand({shardCollection: kDottedFullNs, key: {"a.b": 1}}));
assert.commandFailedWithCode(dottedColl.insert({a: [{b: -5}, {b: 5}]}),
ErrorCodes.ShardKeyNotFound);
assert.eq(null, dottedColl.findOne({"a.b": 5}));
assert.eq(null, dottedColl.findOne({"a.b": -5}));
jsTestLog("Testing dotted path shard key with array value updates.");
assert.commandWorked(dottedColl.insert({a: {b: 1}}));
assert.commandFailedWithCode(dottedColl.update({a: {b: 1}}, {a: [{b: -5}, {b: 5}]}),
ErrorCodes.NotSingleValueField);
assert.commandWorked(sessionDB[kDottedCollName].update({a: {b: 1}}, {}));
printjson("Sharding-then-inserting-multikey tested, now trying inserting-then-sharding-multikey");
// Insert a bunch of data then shard over key which is an array
coll = mongos.getCollection("" + coll + "2");
for (var i = 0; i < 10; i++) {
// TODO : does not check weird cases like [ i, i ]
assert.commandWorked(coll.insert({i: [i, i + 1]}));
}
coll.ensureIndex({_id: 1, i: 1});
try {
st.shardColl(coll, {_id: 1, i: 1}, {_id: ObjectId(), i: 1});
} catch (e) {
print("Correctly threw error on sharding with multikey index.");
}
st.printShardingStatus();
// Insert a bunch of data then shard over key which is not an array
coll = mongos.getCollection("" + coll + "3");
for (var i = 0; i < 10; i++) {
// TODO : does not check weird cases like [ i, i ]
assert.commandWorked(coll.insert({i: i}));
}
coll.ensureIndex({_id: 1, i: 1});
st.shardColl(coll, {_id: 1, i: 1}, {_id: ObjectId(), i: 1});
st.printShardingStatus();
st.stop();
})();
|
