246 lines
9.0 KiB
C++
246 lines
9.0 KiB
C++
// rs_config.cpp
|
|
|
|
/**
|
|
* Copyright (C) 2008 10gen Inc.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
* as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "pch.h"
|
|
#include "replset.h"
|
|
#include "../../client/dbclient.h"
|
|
#include "../../util/hostandport.h"
|
|
#include "../dbhelpers.h"
|
|
#include "connections.h"
|
|
|
|
using namespace bson;
|
|
|
|
namespace mongo {
|
|
|
|
void ReplSetConfig::save() {
|
|
check();
|
|
log() << "replSet info saving a newer config version to local.system.replset" << rsLog;
|
|
MemoryMappedFile::flushAll(true);
|
|
{
|
|
writelock lk("admin.");
|
|
BSONObj o = asBson();
|
|
Helpers::putSingletonGod(rsConfigNs.c_str(), o, false/*logOp=false; local db so would work regardless...*/);
|
|
MemoryMappedFile::flushAll(true);
|
|
}
|
|
}
|
|
|
|
bo ReplSetConfig::MemberCfg::asBson() const {
|
|
bob b;
|
|
b << "_id" << _id;
|
|
b.append("host", h.toString());
|
|
if( votes != 1 ) b << "votes" << votes;
|
|
if( priority != 1.0 ) b << "priority" << priority;
|
|
if( arbiterOnly ) b << "arbiterOnly" << true;
|
|
return b.obj();
|
|
}
|
|
|
|
bo ReplSetConfig::asBson() const {
|
|
bob b;
|
|
b.append("_id", _id).append("version", version);
|
|
if( !ho.isDefault() || !getLastErrorDefaults.isEmpty() ) {
|
|
bob settings;
|
|
if( !ho.isDefault() )
|
|
settings << "heartbeatConnRetries " << ho.heartbeatConnRetries <<
|
|
"heartbeatSleep" << ho.heartbeatSleepMillis / 1000 <<
|
|
"heartbeatTimeout" << ho.heartbeatTimeoutMillis / 1000;
|
|
if( !getLastErrorDefaults.isEmpty() )
|
|
settings << "getLastErrorDefaults" << getLastErrorDefaults;
|
|
b << "settings" << settings.obj();
|
|
}
|
|
|
|
BSONArrayBuilder a;
|
|
for( unsigned i = 0; i < members.size(); i++ )
|
|
a.append( members[i].asBson() );
|
|
b.append("members", a.arr());
|
|
|
|
return b.obj();
|
|
}
|
|
|
|
static inline void mchk(bool expr) {
|
|
uassert(13126, "bad Member config", expr);
|
|
}
|
|
|
|
void ReplSetConfig::MemberCfg::check() const{
|
|
mchk(_id >= 0 && _id <= 255);
|
|
mchk(priority >= 0 && priority <= 1000);
|
|
mchk(votes >= 0 && votes <= 100);
|
|
}
|
|
|
|
void ReplSetConfig::clear() {
|
|
version = -5;
|
|
_ok = false;
|
|
}
|
|
|
|
void ReplSetConfig::check() const {
|
|
uassert(13132,
|
|
"nonmatching repl set name in _id field; check --replSet command line",
|
|
startsWith(cmdLine.replSet, _id + '/'));
|
|
uassert(13133,
|
|
"replSet config value is not valid",
|
|
members.size() >= 1 && members.size() <= 64 &&
|
|
version > 0);
|
|
|
|
}
|
|
|
|
void ReplSetConfig::from(BSONObj o) {
|
|
md5 = o.md5();
|
|
_id = o["_id"].String();
|
|
if( o["version"].ok() ) {
|
|
version = o["version"].numberInt();
|
|
uassert(13115, "bad " + rsConfigNs + " config: version", version > 0);
|
|
}
|
|
|
|
if( o["settings"].ok() ) {
|
|
BSONObj settings = o["settings"].Obj();
|
|
if( settings["heartbeatConnRetries "].ok() )
|
|
ho.heartbeatConnRetries = settings["heartbeatConnRetries "].numberInt();
|
|
if( settings["heartbeatSleep"].ok() )
|
|
ho.heartbeatSleepMillis = (unsigned) (settings["heartbeatSleep"].Number() * 1000);
|
|
if( settings["heartbeatTimeout"].ok() )
|
|
ho.heartbeatTimeoutMillis = (unsigned) (settings["heartbeatTimeout"].Number() * 1000);
|
|
ho.check();
|
|
try { getLastErrorDefaults = settings["getLastErrorDefaults"].Obj(); } catch(...) { }
|
|
}
|
|
|
|
set<string> hosts;
|
|
set<int> ords;
|
|
vector<BSONElement> members;
|
|
try {
|
|
members = o["members"].Array();
|
|
}
|
|
catch(...) {
|
|
uasserted(13131, "replSet error parsing (or missing) 'members' field in config object");
|
|
}
|
|
for( unsigned i = 0; i < members.size(); i++ ) {
|
|
BSONObj mobj = members[i].Obj();
|
|
MemberCfg m;
|
|
try {
|
|
try {
|
|
m._id = (int) mobj["_id"].Number();
|
|
} catch(...) { throw "_id must be numeric"; }
|
|
string s;
|
|
try {
|
|
s = mobj["host"].String();
|
|
m.h = HostAndPort(s);
|
|
}
|
|
catch(...) { throw "bad or missing host field?"; }
|
|
m.arbiterOnly = mobj.getBoolField("arbiterOnly");
|
|
try { m.priority = mobj["priority"].Number(); } catch(...) { }
|
|
try { m.votes = (unsigned) mobj["votes"].Number(); } catch(...) { }
|
|
m.check();
|
|
}
|
|
catch( const char * p ) {
|
|
log() << "replSet cfg parsing exception for members[" << i << "] " << p << rsLog;
|
|
stringstream ss;
|
|
ss << "replSet members[" << i << "] " << p;
|
|
uassert(13107, ss.str(), false);
|
|
}
|
|
catch(DBException& e) {
|
|
log() << "replSet cfg parsing exception for members[" << i << "] " << e.what() << rsLog;
|
|
stringstream ss;
|
|
ss << "replSet members[" << i << "] bad config object";
|
|
uassert(13135, ss.str(), false);
|
|
}
|
|
uassert(13108, "bad " + rsConfigNs + " config dups?", ords.count(m._id) == 0 && hosts.count(m.h.toString()) == 0);
|
|
hosts.insert(m.h.toString());
|
|
ords.insert(m._id);
|
|
this->members.push_back(m);
|
|
}
|
|
uassert(13117, "bad " + rsConfigNs + " config", !_id.empty());
|
|
}
|
|
|
|
static inline void configAssert(bool expr) {
|
|
uassert(13122, "bad " + rsConfigNs + " config", expr);
|
|
}
|
|
|
|
ReplSetConfig::ReplSetConfig(BSONObj cfg) {
|
|
clear();
|
|
from(cfg);
|
|
configAssert( version < 0 /*unspecified*/ || version == 1 );
|
|
version = 1;
|
|
_ok = true;
|
|
}
|
|
|
|
ReplSetConfig::ReplSetConfig(const HostAndPort& h) {
|
|
clear();
|
|
int level = 2;
|
|
DEV level = 0;
|
|
//log(0) << "replSet load config from: " << h.toString() << rsLog;
|
|
|
|
auto_ptr<DBClientCursor> c;
|
|
int v = -5;
|
|
try {
|
|
if( h.isSelf() ) {
|
|
;
|
|
}
|
|
else {
|
|
/* first, make sure other node is configured to be a replset. just to be safe. */
|
|
size_t sl = cmdLine.replSet.find('/');
|
|
assert( sl != string::npos );
|
|
string setname = cmdLine.replSet.substr(0, sl);
|
|
BSONObj cmd = BSON( "replSetHeartbeat" << setname );
|
|
int theirVersion;
|
|
BSONObj info;
|
|
bool ok = requestHeartbeat(setname, h.toString(), info, -2, theirVersion);
|
|
if( info["rs"].trueValue() ) {
|
|
// yes, it is a replicate set, although perhaps not yet initialized
|
|
}
|
|
else {
|
|
if( !ok ) {
|
|
log() << "replSet TEMP !ok heartbeating " << h.toString() << " on cfg load" << rsLog;
|
|
if( !info.isEmpty() ) log() << "replSet TEMP response was: " << info.toString() << rsLog;
|
|
return;
|
|
}
|
|
{
|
|
stringstream ss;
|
|
ss << "replSet error: member " << h.toString() << " is not in --replSet mode";
|
|
cout << "TEMP " << info.toString() << endl;
|
|
msgassertedNoTrace(13260, ss.str().c_str()); // not caught as not a user exception - we want it not caught
|
|
//for python err# checker: uassert(13260, "", false);
|
|
}
|
|
}
|
|
}
|
|
|
|
v = -4;
|
|
ScopedConn conn(h.toString());
|
|
v = -3;
|
|
c = conn->query(rsConfigNs);
|
|
if( c.get() == 0 ) {
|
|
version = v; return;
|
|
}
|
|
if( !c->more() ) {
|
|
version = EMPTYCONFIG;
|
|
return;
|
|
}
|
|
version = -1;
|
|
}
|
|
catch( DBException& e) {
|
|
version = v;
|
|
log(level) << "replSet load config couldn't load " << h.toString() << ' ' << e.what() << rsLog;
|
|
return;
|
|
}
|
|
|
|
BSONObj o = c->nextSafe();
|
|
uassert(13109, "multiple rows in " + rsConfigNs + " not supported", !c->more());
|
|
from(o);
|
|
_ok = true;
|
|
log(level) << "replSet load config ok " << h.toString() << rsLog;
|
|
}
|
|
|
|
}
|