blob: e611e541c3f95d56a096f4dca3cd8264c2e4f594 (
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
121
122
|
//
// MultiVersion utility functions for clusters
//
/**
* Restarts the specified binaries in options with the specified binVersion.
* Note: this does not perform any upgrade operations.
*
* @param binVersion {string}
* @param options {Object} format:
*
* {
* upgradeShards: <bool>, // defaults to true
* upgradeConfigs: <bool>, // defaults to true
* upgradeMongos: <bool>, // defaults to true
* }
*/
ShardingTest.prototype.upgradeCluster = function(binVersion, options) {
options = options || {};
if (options.upgradeShards == undefined)
options.upgradeShards = true;
if (options.upgradeConfigs == undefined)
options.upgradeConfigs = true;
if (options.upgradeMongos == undefined)
options.upgradeMongos = true;
var upgradedSingleShards = [];
if (options.upgradeConfigs) {
// Upgrade config servers if they aren't already upgraded shards
var numConfigs = this._configServers.length;
for (var i = 0; i < numConfigs; i++) {
var configSvr = this._configServers[i];
if (configSvr.host in upgradedSingleShards) {
configSvr = upgradedSingleShards[configSvr.host];
} else {
MongoRunner.stopMongod(configSvr);
configSvr = MongoRunner.runMongod(
{restart: configSvr, binVersion: binVersion, appendOptions: true});
}
this["config" + i] = this["c" + i] = this._configServers[i] = configSvr;
}
}
if (options.upgradeShards) {
var numShards = this._connections.length;
// Upgrade shards
for (var i = 0; i < numShards; i++) {
if (this._rs && this._rs[i]) {
// Upgrade replica set
var rst = this._rs[i].test;
rst.upgradeSet({binVersion: binVersion});
} else {
// Upgrade shard
var shard = this._connections[i];
MongoRunner.stopMongod(shard);
shard = MongoRunner.runMongod(
{restart: shard, binVersion: binVersion, appendOptions: true});
upgradedSingleShards[shard.host] = shard;
this["shard" + i] = this["d" + i] = this._connections[i] = shard;
}
}
}
if (options.upgradeMongos) {
// Upgrade all mongos hosts if specified
var numMongoses = this._mongos.length;
for (var i = 0; i < numMongoses; i++) {
var mongos = this._mongos[i];
MongoRunner.stopMongos(mongos);
mongos = MongoRunner.runMongos(
{restart: mongos, binVersion: binVersion, appendOptions: true});
this["s" + i] = this._mongos[i] = mongos;
if (i == 0)
this.s = mongos;
}
this.config = this.s.getDB("config");
this.admin = this.s.getDB("admin");
}
};
ShardingTest.prototype.restartMongoses = function() {
var numMongoses = this._mongos.length;
for (var i = 0; i < numMongoses; i++) {
var mongos = this._mongos[i];
MongoRunner.stopMongos(mongos);
mongos = MongoRunner.runMongos({restart: mongos});
this["s" + i] = this._mongos[i] = mongos;
if (i == 0)
this.s = mongos;
}
this.config = this.s.getDB("config");
this.admin = this.s.getDB("admin");
};
ShardingTest.prototype.getMongosAtVersion = function(binVersion) {
var mongoses = this._mongos;
for (var i = 0; i < mongoses.length; i++) {
try {
var version = mongoses[i].getDB("admin").runCommand("serverStatus").version;
if (version.indexOf(binVersion) == 0) {
return mongoses[i];
}
} catch (e) {
printjson(e);
print(mongoses[i]);
}
}
};
|
