MockReplicaSet::MockReplicaSet(const string& setName, size_t nodes) : _setName(setName) {
BSONObjBuilder configBuilder;
configBuilder.append("_id", setName);
configBuilder.append("version", 1);
BSONArrayBuilder membersBuilder(configBuilder.subarrayStart("members"));
for (size_t n = 0; n < nodes; n++) {
std::stringstream str;
str << "$" << setName << n << ":27017";
const string hostName(str.str());
if (n == 0) {
_primaryHost = hostName;
}
MockRemoteDBServer* mockServer = new MockRemoteDBServer(hostName);
_nodeMap[hostName] = mockServer;
MockConnRegistry::get()->addServer(mockServer);
membersBuilder.append(BSON("_id" << static_cast<int>(n) << "host" << hostName));
}
membersBuilder.done();
ReplicaSetConfig replConfig;
fassert(28566, replConfig.initialize(configBuilder.obj()));
fassert(28573, replConfig.validate());
setConfig(replConfig);
}
ReplicaSetConfig ReplCoordTest::assertMakeRSConfig(const BSONObj& configBson) {
ReplicaSetConfig config;
ASSERT_OK(config.initialize(configBson));
ASSERT_OK(config.validate());
return config;
}
Status LegacyReplicationCoordinator::processHeartbeat(const ReplSetHeartbeatArgs& args,
ReplSetHeartbeatResponse* response) {
if (args.getProtocolVersion() != 1) {
return Status(ErrorCodes::BadValue, "incompatible replset protocol version");
}
{
if (_settings.ourSetName() != args.getSetName()) {
log() << "replSet set names do not match, our cmdline: " << _settings.replSet
<< rsLog;
log() << "replSet s: " << args.getSetName() << rsLog;
response->noteMismatched();
return Status(ErrorCodes::BadValue, "repl set names do not match");
}
}
response->noteReplSet();
if( (theReplSet == 0) || (theReplSet->startupStatus == ReplSetImpl::LOADINGCONFIG) ) {
if (!args.getSenderHost().empty()) {
scoped_lock lck( _settings.discoveredSeeds_mx );
_settings.discoveredSeeds.insert(args.getSenderHost().toString());
}
response->setHbMsg("still initializing");
return Status::OK();
}
if (theReplSet->name() != args.getSetName()) {
response->noteMismatched();
return Status(ErrorCodes::BadValue, "repl set names do not match (2)");
}
response->setSetName(theReplSet->name());
MemberState currentState = theReplSet->state();
response->setState(currentState.s);
if (currentState == MemberState::RS_PRIMARY) {
response->setElectionTime(theReplSet->getElectionTime().asDate());
}
response->setElectable(theReplSet->iAmElectable());
response->setHbMsg(theReplSet->hbmsg());
response->setTime((long long) time(0));
response->setOpTime(theReplSet->lastOpTimeWritten.asDate());
const Member *syncTarget = BackgroundSync::get()->getSyncTarget();
if (syncTarget) {
response->setSyncingTo(syncTarget->fullName());
}
int v = theReplSet->config().version;
response->setVersion(v);
if (v > args.getConfigVersion()) {
ReplicaSetConfig config;
fassert(18635, config.initialize(theReplSet->config().asBson()));
response->setConfig(config);
}
Member* from = NULL;
if (v == args.getConfigVersion() && args.getSenderId() != -1) {
from = theReplSet->getMutableMember(args.getSenderId());
}
if (!from) {
from = theReplSet->findByName(args.getSenderHost().toString());
if (!from) {
return Status::OK();
}
}
// if we thought that this node is down, let it know
if (!from->hbinfo().up()) {
response->noteStateDisagreement();
}
// note that we got a heartbeat from this node
theReplSet->mgr->send(stdx::bind(&ReplSet::msgUpdateHBRecv,
theReplSet, from->hbinfo().id(), time(0)));
return Status::OK();
}
