mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-30 00:56:44 +01:00
292 lines
10 KiB
C++
292 lines
10 KiB
C++
// dbgrid/request.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/>.
|
|
*/
|
|
|
|
/* TODO
|
|
_ concurrency control.
|
|
_ limit() works right?
|
|
_ KillCursors
|
|
|
|
later
|
|
_ secondary indexes
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "../util/message.h"
|
|
#include "../db/dbmessage.h"
|
|
#include "../client/connpool.h"
|
|
#include "../db/commands.h"
|
|
#include "config.h"
|
|
|
|
namespace mongo {
|
|
|
|
extern string ourHostname;
|
|
|
|
namespace dbgrid_cmds {
|
|
|
|
set<string> dbgridCommands;
|
|
|
|
class GridAdminCmd : public Command {
|
|
public:
|
|
GridAdminCmd( const char * n ) : Command( n ){
|
|
dbgridCommands.insert( n );
|
|
}
|
|
virtual bool slaveOk(){
|
|
return true;
|
|
}
|
|
virtual bool adminOnly() {
|
|
return true;
|
|
}
|
|
};
|
|
|
|
// --------------- misc commands ----------------------
|
|
|
|
class NetStatCmd : public GridAdminCmd {
|
|
public:
|
|
NetStatCmd() : GridAdminCmd("netstat") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
result.append("configserver", configServer.getPrimary() );
|
|
result.append("isdbgrid", 1);
|
|
return true;
|
|
}
|
|
} netstat;
|
|
|
|
class ListGridCommands : public GridAdminCmd {
|
|
public:
|
|
ListGridCommands() : GridAdminCmd("gridcommands") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
|
|
BSONObjBuilder arr;
|
|
int num=0;
|
|
for ( set<string>::iterator i = dbgridCommands.begin(); i != dbgridCommands.end(); i++ ){
|
|
string s = BSONObjBuilder::numStr( num++ );
|
|
arr.append( s.c_str() , *i );
|
|
}
|
|
|
|
result.appendArray( "commands" , arr.done() );
|
|
result.append("ok" , 1 );
|
|
return true;
|
|
}
|
|
} listGridCommands;
|
|
|
|
|
|
// ------------ database level commands -------------
|
|
|
|
class ListDatabaseCommand : public GridAdminCmd {
|
|
public:
|
|
ListDatabaseCommand() : GridAdminCmd("listdatabases") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
ScopedDbConnection conn( configServer.getPrimary() );
|
|
|
|
|
|
auto_ptr<DBClientCursor> cursor = conn->query( "config.databases" , emptyObj );
|
|
|
|
BSONObjBuilder list;
|
|
int num = 0;
|
|
while ( cursor->more() ){
|
|
string s = BSONObjBuilder::numStr( num++ );
|
|
|
|
BSONObj o = cursor->next();
|
|
list.append( s.c_str() , o["name"].valuestrsafe() );
|
|
}
|
|
|
|
result.appendArray("databases" , list.obj() );
|
|
conn.done();
|
|
|
|
return true;
|
|
}
|
|
} gridListDatabase;
|
|
|
|
class MoveDatabasePrimaryCommand : public GridAdminCmd {
|
|
public:
|
|
MoveDatabasePrimaryCommand() : GridAdminCmd("moveprimary") { }
|
|
virtual void help( stringstream& help ) const {
|
|
help << " example: { moveprimary : 'foo' , to : 'localhost:9999' } TODO: locking? ";
|
|
}
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
string dbname = cmdObj["moveprimary"].valuestrsafe();
|
|
|
|
if ( dbname.size() == 0 ){
|
|
errmsg = "no db";
|
|
return false;
|
|
}
|
|
|
|
if ( dbname == "config" ){
|
|
errmsg = "can't move config db";
|
|
return false;
|
|
}
|
|
|
|
DBConfig * config = grid.getDBConfig( dbname , false );
|
|
if ( ! config ){
|
|
errmsg = "can't find db!";
|
|
return false;
|
|
}
|
|
|
|
string to = cmdObj["to"].valuestrsafe();
|
|
if ( ! to.size() ){
|
|
errmsg = "you have to specify where you want to move it";
|
|
return false;
|
|
}
|
|
|
|
if ( to == config->getPrimary() ){
|
|
errmsg = "thats already the primary";
|
|
return false;
|
|
}
|
|
|
|
ScopedDbConnection conn( configServer.getPrimary() );
|
|
|
|
BSONObj server = conn->findOne( "config.servers" , BSON( "host" << to ) );
|
|
if ( server.isEmpty() ){
|
|
errmsg = "that server isn't known to me";
|
|
conn.done();
|
|
return false;
|
|
}
|
|
|
|
log() << "moving " << dbname << " primary from: " << config->getPrimary() << " to: " << to << endl;
|
|
|
|
// TODO LOCKING: this is not safe with multiple mongos
|
|
|
|
|
|
ScopedDbConnection toconn( to );
|
|
|
|
// TODO AARON - we need a clone command which replays operations from clone start to now
|
|
// using a seperate smaller oplog
|
|
BSONObj cloneRes;
|
|
bool worked = toconn->runCommand( "admin" , BSON( "clone" << config->getPrimary() ) , cloneRes );
|
|
toconn.done();
|
|
if ( ! worked ){
|
|
log() << "clone failed" << cloneRes << endl;
|
|
errmsg = "clone failed";
|
|
conn.done();
|
|
return false;
|
|
}
|
|
|
|
ScopedDbConnection fromconn( config->getPrimary() );
|
|
|
|
config->setPrimary( to );
|
|
config->save();
|
|
|
|
log() << " dropping " << dbname << " from old" << endl;
|
|
|
|
fromconn->dropDatabase( dbname.c_str() );
|
|
fromconn.done();
|
|
|
|
result << "ok" << 1;
|
|
result << "primary" << to;
|
|
|
|
conn.done();
|
|
return true;
|
|
}
|
|
} movePrimary;
|
|
|
|
|
|
// ------------ server level commands -------------
|
|
|
|
class ListServers : public GridAdminCmd {
|
|
public:
|
|
ListServers() : GridAdminCmd("listservers") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
ScopedDbConnection conn( configServer.getPrimary() );
|
|
|
|
vector<BSONObj> all;
|
|
auto_ptr<DBClientCursor> cursor = conn->query( "config.servers" , emptyObj );
|
|
while ( cursor->more() ){
|
|
BSONObj o = cursor->next();
|
|
all.push_back( o );
|
|
}
|
|
|
|
result.append("servers" , all );
|
|
result.append("ok" , 1 );
|
|
conn.done();
|
|
|
|
return true;
|
|
}
|
|
} listServers;
|
|
|
|
|
|
class AddServer : public GridAdminCmd {
|
|
public:
|
|
AddServer() : GridAdminCmd("addserver") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
ScopedDbConnection conn( configServer.getPrimary() );
|
|
|
|
BSONObj server = BSON( "host" << cmdObj["addserver"].valuestrsafe() );
|
|
|
|
BSONObj old = conn->findOne( "config.servers" , server );
|
|
if ( ! old.isEmpty() ){
|
|
result.append( "ok" , 0.0 );
|
|
result.append( "msg" , "already exists" );
|
|
conn.done();
|
|
return false;
|
|
}
|
|
|
|
conn->insert( "config.servers" , server );
|
|
result.append( "ok", 1 );
|
|
result.append( "added" , server["host"].valuestrsafe() );
|
|
conn.done();
|
|
return true;
|
|
}
|
|
} addServer;
|
|
|
|
class RemoveServer : public GridAdminCmd {
|
|
public:
|
|
RemoveServer() : GridAdminCmd("removeserver") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool){
|
|
ScopedDbConnection conn( configServer.getPrimary() );
|
|
|
|
BSONObj server = BSON( "host" << cmdObj["removeserver"].valuestrsafe() );
|
|
conn->remove( "config.servers" , server );
|
|
|
|
conn.done();
|
|
return true;
|
|
}
|
|
} removeServer;
|
|
|
|
|
|
// --------------- public commands ----------------
|
|
|
|
class IsDbGridCmd : public Command {
|
|
public:
|
|
virtual bool slaveOk() {
|
|
return true;
|
|
}
|
|
IsDbGridCmd() : Command("isdbgrid") { }
|
|
bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool) {
|
|
result.append("isdbgrid", 1);
|
|
result.append("hostname", ourHostname);
|
|
return true;
|
|
}
|
|
} isdbgrid;
|
|
|
|
class CmdIsMaster : public Command {
|
|
public:
|
|
virtual bool requiresAuth() { return false; }
|
|
virtual bool slaveOk() {
|
|
return true;
|
|
}
|
|
CmdIsMaster() : Command("ismaster") { }
|
|
virtual bool run(const char *ns, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool) {
|
|
result.append("ismaster", 0.0);
|
|
result.append("msg", "isdbgrid");
|
|
return true;
|
|
}
|
|
} ismaster;
|
|
|
|
}
|
|
|
|
} // namespace mongo
|