mongo/jstests/sharding/inserts_consistent.js

// Test write re-routing on version mismatch.

var st = new ShardingTest({ shards : 2, mongos : 2, verbose : 2 })

jsTest.log( "Doing test setup..." )

// Stop balancer, since it'll just get in the way of this
st.stopBalancer()

var mongos = st.s
var admin = mongos.getDB("admin")
var config = mongos.getDB("config")
var coll = st.s.getCollection( jsTest.name() + ".coll" )

st.shardColl( coll, { _id : 1 }, { _id : 0 }, false )

jsTest.log( "Refreshing second mongos..." )

var mongosB = st.s1
var adminB = mongosB.getDB("admin")
var collB = mongosB.getCollection( coll + "" )

// Make sure mongosB knows about the coll
assert.eq( 0, collB.find().itcount() )
// printjson( adminB.runCommand({ flushRouterConfig : 1 }) )

jsTest.log( "Moving chunk to create stale mongos..." )

var otherShard = config.chunks.findOne({ _id : sh._collRE( coll ) }).shard
for( var i = 0; i < st._shardNames.length; i++ ){
    if( otherShard != st._shardNames[i] ){
        otherShard = st._shardNames[i]
        break
    }
}

print( "Other shard : " + otherShard )

printjson( admin.runCommand({ moveChunk : coll + "", find : { _id : 0 }, to : otherShard }) )

jsTest.log( "Inserting docs that needs to be retried..." )

var nextId = -1
for( var i = 0; i < 2; i++ ){
    printjson( "Inserting " + nextId )
    assert.writeOK(collB.insert({ _id : nextId--, hello : "world" }));
}

jsTest.log( "Inserting doc which successfully goes through..." )

// Do second write
assert.writeOK(collB.insert({ _id : nextId--, goodbye : "world" }))

// Assert that write went through
assert.eq( coll.find().itcount(), 3 )

jsTest.log( "Now try moving the actual chunk we're writing to..." )

// Now move the actual chunk we're writing to
printjson( admin.runCommand({ moveChunk : coll + "", find : { _id : -1 }, to : otherShard }) )

jsTest.log( "Inserting second docs to get written back..." )

// Will fail entirely if too many of these, waiting for write to get applied can get too long.
for( var i = 0; i < 2; i++ ){
    collB.insert({ _id : nextId--, hello : "world" })
}

// Refresh server
printjson( adminB.runCommand({ flushRouterConfig : 1 }) )

jsTest.log( "Inserting second doc which successfully goes through..." )

// Do second write
assert.writeOK(collB.insert({ _id : nextId--, goodbye : "world" }));

jsTest.log( "All docs written this time!" )

// Assert that writes went through.
assert.eq( coll.find().itcount(), 6 )

jsTest.log( "DONE" )

st.stop()