summaryrefslogtreecommitdiff
path: root/jstests/sharding/inserts_consistent.js
blob: 86db9ae4e829005a81f761b58dd1bc2141f00923 (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
// Test write re-routing on version mismatch.
(function() {
    'use strict';

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

    var mongos = st.s;
    var admin = mongos.getDB("admin");
    var config = mongos.getDB("config");
    var coll = st.s.getCollection('TestDB.coll');

    assert.commandWorked(mongos.adminCommand({enableSharding: 'TestDB'}));
    st.ensurePrimaryShard('TestDB', st.shard0.shardName);
    assert.commandWorked(mongos.adminCommand({shardCollection: 'TestDB.coll', key: {_id: 1}}));

    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());

    jsTest.log("Moving chunk to create stale mongos...");
    assert.commandWorked(
        admin.runCommand({moveChunk: coll + "", find: {_id: 0}, to: st.shard1.shardName}));

    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: st.shard1.shardName}));

    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);

    st.stop();
})();