mirrors/mongodb
mirrors/mongodb
0
0
Fork 0
mirror of https://github.com/mongodb/mongo.git synced 2024-12-01 09:32:32 +01:00
Code Issues Releases Activity

SERVER-22499 Mark mongos as inShutdown() as soon as shutdown starts

Browse Source
This commit is contained in:
Spencer T Brody 2016-02-05 16:21:04 -05:00
parent c3a232ef38
commit 50e97d0955
1 changed files with 43 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
src/mongo/s/server.cpp
View File

@ -108,10 +108,51 @@ ntservice::NtServiceDefaultStrings defaultServiceStrings = {
static ExitCode initService();
#endif
bool dbexitCalled = false;
static AtomicUInt32 shutdownInProgress(0);
bool inShutdown() {
return dbexitCalled;
return shutdownInProgress.loadRelaxed() != 0;
}
void signalShutdown() {
// Notify all threads shutdown has started
shutdownInProgress.fetchAndAdd(1);
}
void exitCleanly(ExitCode code) {
signalShutdown();
{
Client& client = cc();
ServiceContext::UniqueOperationContext uniqueTxn;
OperationContext* txn = client.getOperationContext();
if (!txn) {
uniqueTxn = client.makeOperationContext();
txn = uniqueTxn.get();
}
auto cursorManager = grid.getCursorManager();
cursorManager->shutdown();
grid.shardRegistry()->shutdown();
grid.catalogManager(txn)->shutDown(txn);
}
dbexit(code);
}
void dbexit(ExitCode rc, const char* why) {
audit::logShutdown(ClientBasic::getCurrent());
#if defined(_WIN32)
// Windows Service Controller wants to be told when we are done shutting down
// and call quickExit itself.
//
if (rc == EXIT_WINDOWS_SERVICE_STOP) {
log() << "dbexit: exiting because Windows service was stopped";
return;
}
#endif
log() << "dbexit: " << why << " rc:" << rc;
quickExit(rc);
}
static BSONObj buildErrReply(const DBException& ex) {
@ -442,46 +483,3 @@ int main(int argc, char* argv[], char** envp) {
quickExit(exitCode);
}
#endif
void mongo::signalShutdown() {
// Notify all threads shutdown has started
dbexitCalled = true;
}
void mongo::exitCleanly(ExitCode code) {
// TODO: do we need to add anything?
{
Client& client = cc();
ServiceContext::UniqueOperationContext uniqueTxn;
OperationContext* txn = client.getOperationContext();
if (!txn) {
uniqueTxn = client.makeOperationContext();
txn = uniqueTxn.get();
}
auto cursorManager = grid.getCursorManager();
cursorManager->shutdown();
grid.shardRegistry()->shutdown();
grid.catalogManager(txn)->shutDown(txn);
}
mongo::dbexit(code);
}
void mongo::dbexit(ExitCode rc, const char* why) {
dbexitCalled = true;
audit::logShutdown(ClientBasic::getCurrent());
#if defined(_WIN32)
// Windows Service Controller wants to be told when we are done shutting down
// and call quickExit itself.
//
if (rc == EXIT_WINDOWS_SERVICE_STOP) {
log() << "dbexit: exiting because Windows service was stopped";
return;
}
#endif
log() << "dbexit: " << why << " rc:" << rc;
quickExit(rc);
}