File: 4_upgrade_cluster.js

package info (click to toggle)
mongodb 1%3A2.4.10-5
  • links: PTS, VCS
  • area: main
  • in suites: jessie-kfreebsd
  • size: 82,464 kB
  • sloc: cpp: 740,225; ansic: 152,098; sh: 13,820; python: 11,864; makefile: 1,012; perl: 922; pascal: 617; java: 452; lisp: 222; asm: 174
file content (150 lines) | stat: -rw-r--r-- 4,081 bytes parent folder | download | duplicates (3)
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
//
// Upgrades a cluster to a newer version
//

load( './jstests/multiVersion/libs/multi_rs.js' )
load( './jstests/multiVersion/libs/multi_cluster.js' )
load( './jstests/libs/test_background_ops.js' )

var oldVersion = "2.0"
var newVersion = "2.2"

// BIG OUTER LOOP, RS CLUSTER OR NOT!
for( var test = 0; test < 1; test++ ){
    
// TODO: RS Test messes up here
var isRSCluster = test == 1



jsTest.log( "Starting " + ( isRSCluster ? "(replica set)" : "" ) + " cluster..." )

var options = {
    
    mongosOptions : { binVersion : oldVersion },
    configOptions : { binVersion : oldVersion },
    shardOptions : { binVersion : oldVersion },
    
    separateConfig : true,
    sync : true,
    rs : isRSCluster
}

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


jsTest.log( "Starting parallel operations during upgrade..." )

var insertNS = "test.foo"
var shardedInsertNS = "test.bar"
    
var admin = st.s.getDB( "admin" )
var shards = st.s.getDB( "config" ).shards.find().toArray()

printjson( admin.runCommand({ enableSharding : shardedInsertNS }) )
printjson( admin.runCommand({ movePrimary : shardedInsertNS, to : shards[0]._id }) )
printjson( admin.runCommand({ shardCollection : shardedInsertNS, key : { _id : 1 } }) )

st.stopBalancer()

for( var i = 0; i < 5; i++ ){
    printjson( admin.runCommand({ split : shardedInsertNS, middle : { _id : i * 50 } }) )
    printjson( admin.runCommand({ moveChunk : shardedInsertNS, 
                                  find : { _id : i * 50 }, 
                                  to : shards[ i % shards.length ]._id }) )
}

function findAndInsert( mongosURL, ns ){
    
    var coll = null
    
    // Make sure we can eventually connect to the mongos
    assert.soon( function(){
        try{ 
            coll = new Mongo( mongosURL ).getCollection( ns + "" )
            return true
        }
        catch( e ){
            printjson( e )
            return false
        }
    })
    
    var count = 0
    
    jsTest.log( "Starting finds and inserts..." )
    
    while( ! isFinished() ){
        
        try{
            
            coll.insert({ _id : count, hello : "world" })
            assert.eq( null, coll.getDB().getLastError() )
            assert.neq( null, coll.findOne({ _id : count }) )
        }
        catch( e ){
            printjson( e )
        }
        
        count++
    }
    
    jsTest.log( "Finished finds and inserts..." )
    return count
}

var staticMongod = MongoRunner.runMongod({})

printjson( staticMongod )

var joinFindInsert = 
    startParallelOps( staticMongod, // The connection where the test info is passed and stored
                      findAndInsert,
                      [ st.s0.host, insertNS ] )
                      
var joinShardedFindInsert = 
    startParallelOps( staticMongod, // The connection where the test info is passed and stored
                      findAndInsert,
                      [ st.s1.host, shardedInsertNS ] )


jsTest.log( "Upgrading cluster..." )

var startTime = new Date();

st.upgradeCluster( newVersion )

var endTime = new Date();

jsTest.log( "Cluster upgraded." )

st.printShardingStatus()

jsTest.log( "Cluster upgrade took " + ((endTime - startTime) / 1000) + 
            " secs, sleeping for valid inserts" )

// Allow enough time for more valid writes to go through.
sleep((endTime - startTime) * 3)

joinFindInsert()
joinShardedFindInsert()

var totalInserts = st.s.getCollection( insertNS ).find().sort({ _id : -1 }).next()._id + 1
var dataFound = st.s.getCollection( insertNS ).count()

jsTest.log( "Found " + dataFound + " docs out of " + totalInserts + " inserted." )

assert.gt( dataFound / totalInserts, 0.5 )

var totalInserts = st.s.getCollection( shardedInsertNS ).find().sort({ _id : -1 }).next()._id + 1
var dataFound = st.s.getCollection( shardedInsertNS ).find().itcount()

jsTest.log( "Found " + dataFound + " sharded docs out of " + tojson( totalInserts ) + " inserted." )

assert.gt( dataFound / totalInserts, 0.5 )

jsTest.log( "DONE!" )

st.stop()

} // END OUTER LOOP FOR RS CLUSTER