90 lines
3.0 KiB
C++
90 lines
3.0 KiB
C++
/**
|
|
* 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 "stdafx.h"
|
|
#include "../cmdline.h"
|
|
#include "replset.h"
|
|
#include "../../util/sock.h"
|
|
|
|
namespace mongo {
|
|
|
|
ReplSet *theReplSet = 0;
|
|
|
|
/** @param cfgString <setname>/<seedhost1>,<seedhost2> */
|
|
ReplSet::ReplSet(string cfgString) {
|
|
const char *p = cfgString.c_str();
|
|
const char *slash = strchr(p, '/');
|
|
uassert(13093, "bad --replSet config string format is: <setname>/<seedhost1>,<seedhost2>[,...]", slash != 0 && p != slash);
|
|
_name = string(p, slash-p);
|
|
log() << "replSet: " << cfgString << endl;
|
|
|
|
set<HostAndPort> temp;
|
|
vector<HostAndPort> *seeds = new vector<HostAndPort>;
|
|
p = slash + 1;
|
|
while( 1 ) {
|
|
const char *comma = strchr(p, ',');
|
|
if( comma == 0 ) comma = strchr(p,0);
|
|
uassert(13094, "bad --replSet config string", p != comma);
|
|
{
|
|
HostAndPort m;
|
|
try {
|
|
m = HostAndPort::fromString(string(p, comma-p));
|
|
}
|
|
catch(...) {
|
|
uassert(13114, "bad --replSet seed hostname", false);
|
|
}
|
|
uassert(13096, "bad --replSet config string - dups?", temp.count(m) == 0 );
|
|
temp.insert(m);
|
|
uassert(13101, "can't use localhost in replset host list", !m.isLocalHost());
|
|
if( m.isSelf() )
|
|
log() << "replSet: ignoring seed " << m.toString() << " (=self)" << endl;
|
|
else
|
|
seeds->push_back(m);
|
|
if( *comma == 0 )
|
|
break;
|
|
p = comma + 1;
|
|
}
|
|
}
|
|
|
|
_seeds = seeds;
|
|
for( vector<HostAndPort>::iterator i = seeds->begin(); i != seeds->end(); i++ )
|
|
addMemberIfMissing(*i);
|
|
|
|
startHealthThreads();
|
|
}
|
|
|
|
void ReplSet::addMemberIfMissing(const HostAndPort& h) {
|
|
MemberInfo *m = _members.head();
|
|
while( m ) {
|
|
if( h.host() == m->host && h.port() == m->port )
|
|
return;
|
|
m = m->next();
|
|
}
|
|
MemberInfo *nm = new MemberInfo(h.host(), h.port());
|
|
_members.push(nm);
|
|
}
|
|
|
|
/* called at initialization */
|
|
bool startReplSets() {
|
|
assert( theReplSet == 0 );
|
|
if( cmdLine.replSet.empty() )
|
|
return false;
|
|
theReplSet = new ReplSet(cmdLine.replSet);
|
|
return false;
|
|
}
|
|
|
|
}
|