-
Notifications
You must be signed in to change notification settings - Fork 5.6k
/
array_shard_key.js
127 lines (91 loc) · 3.5 KB
/
array_shard_key.js
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
// 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
var checkError = function( shouldError ){
var error = coll.getDB().getLastError()
if( error != null ) printjson( error )
if( error == null && ! shouldError ) return
if( error != null && shouldError ) return
if( error == null ) print( "No error detected!" )
else print( "Unexpected error!" )
assert( false )
}
// Insert an object with invalid array key
coll.insert({ i : [ 1, 2 ] })
checkError( true )
// Insert an object with valid array key
coll.insert({ i : 1 })
checkError( false )
// Update the value with valid other field
value = coll.findOne({ i : 1 })
coll.update( value, { $set : { j : 2 } } )
checkError( false )
// Update the value with invalid other fields
value = coll.findOne({ i : 1 })
coll.update( value, Object.merge( value, { i : [ 3 ] } ) )
checkError( true )
// Multi-update the value with invalid other fields
value = coll.findOne({ i : 1 })
coll.update( value, Object.merge( value, { i : [ 3, 4 ] } ), false, true)
checkError( true )
// Single update the value with valid other fields
value = coll.findOne({ i : 1 })
coll.update( Object.merge( value, { i : [ 3, 4 ] } ), value )
checkError( true )
// Multi-update the value with other fields (won't work, but no error)
value = coll.findOne({ i : 1 })
coll.update( Object.merge( value, { i : [ 1, 1 ] } ), { $set : { k : 4 } }, false, true)
checkError( false )
// 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()
checkError( false )
// Can't remove using multikey, but shouldn't error
value = coll.findOne({ i : 1 })
coll.remove( Object.extend( value, { i : [ 1, 2, 3, 4 ] } ) )
checkError( false )
// Can't remove using multikey, but shouldn't error
value = coll.findOne({ i : 1 })
coll.remove( Object.extend( value, { i : [ 1, 2, 3, 4, 5 ] } ) )
error = coll.getDB().getLastError()
assert.eq( error, null )
assert.eq( coll.find().itcount(), 1 )
value = coll.findOne({ i : 1 })
coll.remove( Object.extend( value, { i : 1 } ) )
error = coll.getDB().getLastError()
assert.eq( error, null )
assert.eq( coll.find().itcount(), 0 )
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 ]
coll.insert({ i : [ i, i + 1 ] })
checkError( false )
}
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 ]
coll.insert({ i : i })
checkError( false )
}
coll.ensureIndex({ _id : 1, i : 1 })
st.shardColl( coll, { _id : 1, i : 1 }, { _id : ObjectId(), i : 1 } )
st.printShardingStatus()
// Finish
st.stop()