summaryrefslogtreecommitdiff
path: root/jstests/sharding/coll_epoch_test1.js
blob: fa24a035da75266d67be4a082088d361bec0b417 (plain)
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
// Tests various cases of dropping and recreating collections in the same namespace with multiple mongoses

var st = new ShardingTest({ shards : 3, mongos : 3, verbose : 1 });
// Balancer is by default stopped, thus it will not interfere

// Use separate mongoses for admin, inserting data, and validating results, so no
// single-mongos tricks will work
var insertMongos = st.s2;
var staleMongos = st.s1;

var config = st.s.getDB( "config" );
var admin = st.s.getDB( "admin" );
var coll = st.s.getCollection( "foo.bar" );

insertMongos.getDB( "admin" ).runCommand({ setParameter : 1, traceExceptions : true });

var shards = {};
config.shards.find().forEach( function( doc ){ 
    shards[ doc._id ] = new Mongo( doc.host );
});

//
// Test that inserts and queries go to the correct shard even when the collection has been sharded
// in the background
//

jsTest.log( "Enabling sharding for the first time..." );

admin.runCommand({ enableSharding : coll.getDB() + "" });
st.ensurePrimaryShard(coll.getDB().getName(), 'shard0001');
admin.runCommand({ shardCollection : coll  + "", key : { _id : 1 } });

var bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ ) {
    bulk.insert({ _id : i, test : "a" });
}
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "a" }).itcount() );

coll.drop();

//
// Test that inserts and queries go to the correct shard even when the collection has been 
// re-sharded in the background
//

jsTest.log( "Re-enabling sharding with a different key..." );

admin.runCommand({ enableSharding : coll.getDB() + "" });
st.ensurePrimaryShard(coll.getDB().getName(), 'shard0001');
coll.ensureIndex({ notId : 1 });
admin.runCommand({ shardCollection : coll  + "", key : { notId : 1 } });

bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ ) {
    bulk.insert({ notId : i, test : "b" });
}
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "b" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a" ] } }).itcount() );

coll.drop();

//
// Test that inserts and queries go to the correct shard even when the collection has been 
// unsharded and moved to a different primary
//

jsTest.log( "Re-creating unsharded collection from a sharded collection on different primary..." );

var getOtherShard = function( shard ){
    for( id in shards ){
        if( id != shard ) return id;
    }
};

var otherShard = getOtherShard(config.databases.findOne({_id: coll.getDB() + ""}).primary);
assert.commandWorked(admin.runCommand({movePrimary: coll.getDB() + "", to: otherShard}));
if (st.configRS) {
    // If we are in CSRS mode need to make sure that staleMongos will actually get
    // the most recent config data.
    st.configRS.awaitLastOpCommitted();
}
jsTest.log( "moved primary..." );

bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ )
    bulk.insert({ test : "c" });
assert.writeOK( bulk.execute() );

assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "c" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a", "b" ] } }).itcount() );

coll.drop();

//
// Test that inserts and queries go to correct shard even when the collection has been unsharded, 
// resharded, and moved to a different primary
//

jsTest.log( "Re-creating sharded collection with different primary..." );

admin.runCommand({ enableSharding : coll.getDB() + "" });
admin.runCommand({ movePrimary : coll.getDB() + "", 
                   to : getOtherShard( config.databases.findOne({ _id : coll.getDB() + "" }).primary ) });
admin.runCommand({ shardCollection : coll  + "", key : { _id : 1 } });

bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ )
    bulk.insert({ test : "d" });
assert.writeOK( bulk.execute() );

assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "d" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a", "b", "c" ] } }).itcount() );

coll.drop();

jsTest.log( "Done!" );

st.stop();