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
|
'use strict';
/**
* Represents a MongoDB cluster.
*/
var Cluster = function(options) {
if (!(this instanceof Cluster)) {
return new Cluster(options);
}
var allowedKeys = [
'masterSlave',
'replication',
'sameCollection',
'sameDB',
'seed',
'sharded'
];
Object.keys(options).forEach(function(option) {
assert.contains(option, allowedKeys,
'invalid option: ' + tojson(option) +
'; valid options are: ' + tojson(allowedKeys));
});
var conn;
var initialized = false;
this.setup = function setup() {
var verbosityLevel = 0;
if (initialized) {
throw new Error('cluster has already been initialized');
}
if (options.sharded) {
// TODO: allow 'options' to specify the number of shards
var shardConfig = {
shards: 2,
mongos: 1,
verbose: verbosityLevel
};
// TODO: allow 'options' to specify an 'rs' config
if (options.replication) {
shardConfig.rs = {
nodes: 3,
verbose: verbosityLevel
};
}
var st = new ShardingTest(shardConfig);
st.stopBalancer();
conn = st.s; // mongos
this.shardCollection = function() {
st.shardColl.apply(st, arguments);
};
this.teardown = function() {
st.stop();
};
} else if (options.replication) {
// TODO: allow 'options' to specify the number of nodes
var replSetConfig = {
nodes: 3,
nodeOptions: { verbose: verbosityLevel }
};
var rst = new ReplSetTest(replSetConfig);
rst.startSet();
// Send the replSetInitiate command and wait for initiation
rst.initiate();
rst.awaitSecondaryNodes();
conn = rst.getPrimary();
this.teardown = function() {
rst.stopSet();
};
} else if (options.masterSlave) {
var rt = new ReplTest('replTest');
var master = rt.start(true);
var slave = rt.start(false);
conn = master;
master.adminCommand({ setParameter: 1, logLevel: verbosityLevel });
slave.adminCommand({ setParameter: 1, logLevel: verbosityLevel });
this.teardown = function() {
rt.stop();
};
} else { // standalone server
conn = db.getMongo();
db.adminCommand({ setParameter: 1, logLevel: verbosityLevel });
}
initialized = true;
};
this.teardown = function teardown() { };
this.getDB = function getDB(dbName) {
if (!initialized) {
throw new Error('cluster has not been initialized yet');
}
return conn.getDB(dbName);
};
this.getHost = function getHost() {
if (!initialized) {
throw new Error('cluster has not been initialized yet');
}
return conn.host;
};
this.isSharded = function isSharded() {
return !!options.sharded;
};
this.shardCollection = function shardCollection() {
assert(this.isSharded(), 'cluster is not sharded');
throw new Error('cluster has not been initialized yet');
};
};
/**
* Returns true if 'clusterOptions' represents a standalone mongod,
* and false otherwise.
*/
Cluster.isStandalone = function isStandalone(clusterOptions) {
return !clusterOptions.sharded && !clusterOptions.replication && !clusterOptions.masterSlave;
};
|
