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
|
// Ensure you can't shard on an array key
var st = new ShardingTest({name: jsTestName(), shards: 3});
var mongos = st.s0;
var coll = mongos.getCollection(jsTestName() + ".foo");
st.shardColl(coll, {_id: 1, i: 1}, {_id: ObjectId(), i: 1});
printjson(mongos.getDB("config").chunks.find().toArray());
st.printShardingStatus();
print("1: insert some invalid data");
var value = null;
// Insert an object with invalid array key
assert.writeError(coll.insert({i: [1, 2]}));
// Insert an object with all the right fields, but an invalid array val for _id
assert.writeError(coll.insert({_id: [1, 2], i: 3}));
// Insert an object with valid array key
assert.writeOK(coll.insert({i: 1}));
// Update the value with valid other field
value = coll.findOne({i: 1});
assert.writeOK(coll.update(value, {$set: {j: 2}}));
// Update the value with invalid other fields
value = coll.findOne({i: 1});
assert.writeError(coll.update(value, Object.merge(value, {i: [3]})));
// Multi-update the value with invalid other fields
value = coll.findOne({i: 1});
assert.writeError(coll.update(value, Object.merge(value, {i: [3, 4]}), false, true));
// Multi-update the value with other fields (won't work, but no error)
value = coll.findOne({i: 1});
assert.writeOK(coll.update(Object.merge(value, {i: [1, 1]}), {$set: {k: 4}}, false, true));
// 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.writeOK(coll.remove(Object.extend(value, {i: [1, 2, 3, 4, 5]})));
assert.eq(coll.find().itcount(), 1);
value = coll.findOne({i: 1});
assert.writeOK(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.writeOK(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.writeOK(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.writeOK(coll.update({_id: 1, i: 1}, {_id: 1, i: 1, j: [1, 2]}, true));
assert.eq(coll.find().itcount(), 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
var coll = mongos.getCollection("" + coll + "2");
for (var i = 0; i < 10; i++) {
// TODO : does not check weird cases like [ i, i ]
assert.writeOK(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
var coll = mongos.getCollection("" + coll + "3");
for (var i = 0; i < 10; i++) {
// TODO : does not check weird cases like [ i, i ]
assert.writeOK(coll.insert({i: i}));
}
coll.ensureIndex({_id: 1, i: 1});
st.shardColl(coll, {_id: 1, i: 1}, {_id: ObjectId(), i: 1});
st.printShardingStatus();
// Finish
st.stop();
|