mirror of
https://github.com/mongodb/mongo.git
synced 2024-12-01 09:32:32 +01:00
45886bc725
Ensures profiler data gets written.
209 lines
5.6 KiB
C++
209 lines
5.6 KiB
C++
// server.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 "stdafx.h"
|
|
#include "../util/message.h"
|
|
#include "../util/unittest.h"
|
|
#include "../client/connpool.h"
|
|
#include "../util/message_server.h"
|
|
|
|
#include "server.h"
|
|
#include "request.h"
|
|
#include "config.h"
|
|
#include "chunk.h"
|
|
|
|
namespace mongo {
|
|
|
|
Database *database = 0;
|
|
string ourHostname;
|
|
OID serverID;
|
|
bool dbexitCalled = false;
|
|
CmdLine cmdLine;
|
|
|
|
bool inShutdown(){
|
|
return dbexitCalled;
|
|
}
|
|
|
|
string getDbContext() {
|
|
return "?";
|
|
}
|
|
|
|
bool haveLocalShardingInfo( const string& ns ){
|
|
assert( 0 );
|
|
return false;
|
|
}
|
|
|
|
void usage( char * argv[] ){
|
|
out() << argv[0] << " usage:\n\n";
|
|
out() << " -v+ verbose\n";
|
|
out() << " --port <portno>\n";
|
|
out() << " --configdb <configdbname> [<configdbname>...]\n";
|
|
out() << endl;
|
|
}
|
|
|
|
class ShardingConnectionHook : public DBConnectionHook {
|
|
public:
|
|
virtual void onCreate( DBClientBase * conn ){
|
|
conn->simpleCommand( "admin" , 0 , "switchtoclienterrors" );
|
|
}
|
|
virtual void onHandedOut( DBClientBase * conn ){
|
|
ClientInfo::get()->addShard( conn->getServerAddress() );
|
|
}
|
|
} shardingConnectionHook;
|
|
|
|
class ShardedMessageHandler : public MessageHandler {
|
|
public:
|
|
virtual ~ShardedMessageHandler(){}
|
|
virtual void process( Message& m , AbstractMessagingPort* p ){
|
|
Request r( m , p );
|
|
if ( logLevel > 5 ){
|
|
log(5) << "client id: " << hex << r.getClientId() << "\t" << r.getns() << "\t" << dec << r.op() << endl;
|
|
}
|
|
try {
|
|
setClientId( r.getClientId() );
|
|
r.process();
|
|
}
|
|
catch ( DBException& e ){
|
|
m.data->id = r.id();
|
|
log() << "UserException: " << e.what() << endl;
|
|
if ( r.expectResponse() ){
|
|
BSONObj err = BSON( "$err" << e.what() );
|
|
replyToQuery( QueryResult::ResultFlag_ErrSet, p , m , err );
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
void sighandler(int sig){
|
|
dbexit(EXIT_CLEAN, (string("recieved signal ") + BSONObjBuilder::numStr(sig)).c_str());
|
|
}
|
|
|
|
void init(){
|
|
serverID.init();
|
|
setupSIGTRAPforGDB();
|
|
signal(SIGTERM, sighandler);
|
|
signal(SIGINT, sighandler);
|
|
}
|
|
|
|
void start() {
|
|
log() << "waiting for connections on port " << cmdLine.port << endl;
|
|
//DbGridListener l(port);
|
|
//l.listen();
|
|
ShardedMessageHandler handler;
|
|
MessageServer * server = createServer( cmdLine.port , &handler );
|
|
server->run();
|
|
}
|
|
|
|
DBClientBase *createDirectClient(){
|
|
uassert( 10197 , "createDirectClient not implemented for sharding yet" , 0 );
|
|
return 0;
|
|
}
|
|
|
|
} // namespace mongo
|
|
|
|
using namespace mongo;
|
|
|
|
int main(int argc, char* argv[], char *envp[] ) {
|
|
|
|
bool justTests = false;
|
|
vector<string> configdbs;
|
|
|
|
for (int i = 1; i < argc; i++) {
|
|
if ( argv[i] == 0 ) continue;
|
|
string s = argv[i];
|
|
if ( s == "--port" ) {
|
|
cmdLine.port = atoi(argv[++i]);
|
|
}
|
|
else if ( s == "--configdb" ) {
|
|
|
|
while ( ++i < argc )
|
|
configdbs.push_back(argv[i]);
|
|
|
|
if ( configdbs.size() == 0 ) {
|
|
out() << "error: no args for --configdb\n";
|
|
return 4;
|
|
}
|
|
|
|
if ( configdbs.size() > 2 ) {
|
|
out() << "error: --configdb does not support more than 2 parameters yet\n";
|
|
return 5;
|
|
}
|
|
}
|
|
else if ( s.find( "-v" ) == 0 ){
|
|
logLevel = s.size() - 1;
|
|
}
|
|
else if ( s == "--test" ) {
|
|
justTests = true;
|
|
logLevel = 5;
|
|
}
|
|
else {
|
|
usage( argv );
|
|
return 3;
|
|
}
|
|
}
|
|
|
|
if ( justTests ){
|
|
UnitTest::runTests();
|
|
cout << "tests passed" << endl;
|
|
return 0;
|
|
}
|
|
|
|
pool.addHook( &shardingConnectionHook );
|
|
|
|
if ( argc <= 1 ) {
|
|
usage( argv );
|
|
return 3;
|
|
}
|
|
|
|
bool ok = cmdLine.port != 0 && configdbs.size();
|
|
|
|
if ( !ok ) {
|
|
usage( argv );
|
|
return 1;
|
|
}
|
|
|
|
log() << argv[0] << " v0.3- (alpha 3t) starting (--help for usage)" << endl;
|
|
printGitVersion();
|
|
printSysInfo();
|
|
|
|
if ( ! configServer.init( configdbs ) ){
|
|
cout << "couldn't connectd to config db" << endl;
|
|
return 7;
|
|
}
|
|
|
|
assert( configServer.ok() );
|
|
|
|
int configError = configServer.checkConfigVersion();
|
|
if ( configError ){
|
|
cout << "config server error: " << configError << endl;
|
|
return configError;
|
|
}
|
|
|
|
init();
|
|
start();
|
|
dbexit( EXIT_CLEAN );
|
|
return 0;
|
|
}
|
|
|
|
#undef exit
|
|
void mongo::dbexit( ExitCode rc, const char *why) {
|
|
dbexitCalled = true;
|
|
log() << "dbexit: " << why << " rc:" << rc << endl;
|
|
::exit(rc);
|
|
}
|