diff options
Diffstat (limited to 'src/mongo/db/dbcommands_generic.cpp')
| -rw-r--r-- | src/mongo/db/dbcommands_generic.cpp | 432 |
1 files changed, 432 insertions, 0 deletions
diff --git a/src/mongo/db/dbcommands_generic.cpp b/src/mongo/db/dbcommands_generic.cpp new file mode 100644 index 00000000000..cfd833aa72d --- /dev/null +++ b/src/mongo/db/dbcommands_generic.cpp @@ -0,0 +1,432 @@ +/** @file dbcommands_generic.cpp commands suited for any mongo server (both mongod, mongos) */ + +/** +* +* 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 "pch.h" +#include "ops/query.h" +#include "pdfile.h" +#include "jsobj.h" +#include "../bson/util/builder.h" +#include <time.h> +#include "introspect.h" +#include "btree.h" +#include "../util/lruishmap.h" +#include "../util/md5.hpp" +#include "../util/processinfo.h" +#include "json.h" +#include "repl.h" +#include "repl_block.h" +#include "replutil.h" +#include "commands.h" +#include "db.h" +#include "instance.h" +#include "lasterror.h" +#include "security.h" +#include "../scripting/engine.h" +#include "stats/counters.h" +#include "background.h" +#include "../util/version.h" +#include "../util/ramlog.h" +#include "repl/multicmd.h" +#include "server.h" + +namespace mongo { + +#if 0 + namespace cloud { + SimpleMutex mtx("cloud"); + Guarded< vector<string>, mtx > ips; + bool startedThread = false; + + void thread() { + bson::bo cmd; + while( 1 ) { + list<Target> L; + { + SimpleMutex::scoped_lock lk(mtx); + if( ips.ref(lk).empty() ) + continue; + for( unsigned i = 0; i < ips.ref(lk).size(); i++ ) { + L.push_back( Target(ips.ref(lk)[i]) ); + } + } + + + /** repoll as machines might be down on the first lookup (only if not found previously) */ + sleepsecs(6); + } + } + } + + class CmdCloud : public Command { + public: + CmdCloud() : Command( "cloud" ) { } + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return true; } + virtual LockType locktype() const { return NONE; } + virtual void help( stringstream &help ) const { + help << "internal command facilitating running in certain cloud computing environments"; + } + bool run(const string& dbname, BSONObj& obj, int options, string& errmsg, BSONObjBuilder& result, bool fromRepl ) { + if( !obj.hasElement("servers") ) { + vector<string> ips; + obj["servers"].Obj().Vals(ips); + { + SimpleMutex::scoped_lock lk(cloud::mtx); + cloud::ips.ref(lk).swap(ips); + if( !cloud::startedThread ) { + cloud::startedThread = true; + boost::thread thr(cloud::thread); + } + } + } + return true; + } + } cmdCloud; +#endif + + class CmdBuildInfo : public Command { + public: + CmdBuildInfo() : Command( "buildInfo", true, "buildinfo" ) {} + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return false; } + virtual bool requiresAuth() { return false; } + virtual LockType locktype() const { return NONE; } + virtual void help( stringstream &help ) const { + help << "get version #, etc.\n"; + help << "{ buildinfo:1 }"; + } + bool run(const string& dbname, BSONObj& jsobj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl ) { + result << "version" << versionString << "gitVersion" << gitVersion() << "sysInfo" << sysInfo(); + result << "versionArray" << versionArray; + result << "bits" << ( sizeof( int* ) == 4 ? 32 : 64 ); + result.appendBool( "debug" , debug ); + result.appendNumber("maxBsonObjectSize", BSONObjMaxUserSize); + return true; + } + } cmdBuildInfo; + + /** experimental. either remove or add support in repl sets also. in a repl set, getting this setting from the + repl set config could make sense. + */ + unsigned replApplyBatchSize = 1; + + class CmdGet : public Command { + public: + CmdGet() : Command( "getParameter" ) { } + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return true; } + virtual LockType locktype() const { return NONE; } + virtual void help( stringstream &help ) const { + help << "get administrative option(s)\nexample:\n"; + help << "{ getParameter:1, notablescan:1 }\n"; + help << "supported so far:\n"; + help << " quiet\n"; + help << " notablescan\n"; + help << " logLevel\n"; + help << " syncdelay\n"; + help << "{ getParameter:'*' } to get everything\n"; + } + bool run(const string& dbname, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl ) { + bool all = *cmdObj.firstElement().valuestrsafe() == '*'; + + int before = result.len(); + + if( all || cmdObj.hasElement("quiet") ) { + result.append("quiet", cmdLine.quiet ); + } + if( all || cmdObj.hasElement("notablescan") ) { + result.append("notablescan", cmdLine.noTableScan); + } + if( all || cmdObj.hasElement("logLevel") ) { + result.append("logLevel", logLevel); + } + if( all || cmdObj.hasElement("syncdelay") ) { + result.append("syncdelay", cmdLine.syncdelay); + } + if( all || cmdObj.hasElement("replApplyBatchSize") ) { + result.append("replApplyBatchSize", replApplyBatchSize); + } + + if ( before == result.len() ) { + errmsg = "no option found to get"; + return false; + } + return true; + } + } cmdGet; + + // tempish + bool setParmsMongodSpecific(const string& dbname, BSONObj& cmdObj, string& errmsg, BSONObjBuilder& result, bool fromRepl ); + + class CmdSet : public Command { + public: + CmdSet() : Command( "setParameter" ) { } + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return true; } + virtual LockType locktype() const { return NONE; } + virtual void help( stringstream &help ) const { + help << "set administrative option(s)\n"; + help << "{ setParameter:1, <param>:<value> }\n"; + help << "supported so far:\n"; + help << " journalCommitInterval\n"; + help << " logLevel\n"; + help << " notablescan\n"; + help << " quiet\n"; + help << " syncdelay\n"; + } + bool run(const string& dbname, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl ) { + int s = 0; + bool found = setParmsMongodSpecific(dbname, cmdObj, errmsg, result, fromRepl); + if( cmdObj.hasElement("journalCommitInterval") ) { + if( !cmdLine.dur ) { + errmsg = "journaling is off"; + return false; + } + int x = (int) cmdObj["journalCommitInterval"].Number(); + assert( x > 1 && x < 500 ); + cmdLine.journalCommitInterval = x; + log() << "setParameter journalCommitInterval=" << x << endl; + s++; + } + if( cmdObj.hasElement("notablescan") ) { + assert( !cmdLine.isMongos() ); + if( s == 0 ) + result.append("was", cmdLine.noTableScan); + cmdLine.noTableScan = cmdObj["notablescan"].Bool(); + s++; + } + if( cmdObj.hasElement("quiet") ) { + if( s == 0 ) + result.append("was", cmdLine.quiet ); + cmdLine.quiet = cmdObj["quiet"].Bool(); + s++; + } + if( cmdObj.hasElement("syncdelay") ) { + assert( !cmdLine.isMongos() ); + if( s == 0 ) + result.append("was", cmdLine.syncdelay ); + cmdLine.syncdelay = cmdObj["syncdelay"].Number(); + s++; + } + if( cmdObj.hasElement( "logLevel" ) ) { + if( s == 0 ) + result.append("was", logLevel ); + logLevel = cmdObj["logLevel"].numberInt(); + s++; + } + if( cmdObj.hasElement( "replApplyBatchSize" ) ) { + if( s == 0 ) + result.append("was", replApplyBatchSize ); + BSONElement e = cmdObj["replApplyBatchSize"]; + ParameterValidator * v = ParameterValidator::get( e.fieldName() ); + assert( v ); + if ( ! v->isValid( e , errmsg ) ) + return false; + replApplyBatchSize = e.numberInt(); + s++; + } + if( cmdObj.hasElement( "traceExceptions" ) ) { + if( s == 0 ) result.append( "was", DBException::traceExceptions ); + DBException::traceExceptions = cmdObj["traceExceptions"].Bool(); + s++; + } + + if( s == 0 && !found ) { + errmsg = "no option found to set, use help:true to see options "; + return false; + } + + return true; + } + } cmdSet; + + class PingCommand : public Command { + public: + PingCommand() : Command( "ping" ) {} + virtual bool slaveOk() const { return true; } + virtual void help( stringstream &help ) const { help << "a way to check that the server is alive. responds immediately even if server is in a db lock."; } + virtual LockType locktype() const { return NONE; } + virtual bool requiresAuth() { return false; } + virtual bool run(const string& badns, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool) { + // IMPORTANT: Don't put anything in here that might lock db - including authentication + return true; + } + } pingCmd; + + class FeaturesCmd : public Command { + public: + FeaturesCmd() : Command( "features", true ) {} + void help(stringstream& h) const { h << "return build level feature settings"; } + virtual bool slaveOk() const { return true; } + virtual bool readOnly() { return true; } + virtual LockType locktype() const { return NONE; } + virtual bool run(const string& ns, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl) { + if ( globalScriptEngine ) { + BSONObjBuilder bb( result.subobjStart( "js" ) ); + result.append( "utf8" , globalScriptEngine->utf8Ok() ); + bb.done(); + } + if ( cmdObj["oidReset"].trueValue() ) { + result.append( "oidMachineOld" , OID::getMachineId() ); + OID::regenMachineId(); + } + result.append( "oidMachine" , OID::getMachineId() ); + return true; + } + + } featuresCmd; + + class LogRotateCmd : public Command { + public: + LogRotateCmd() : Command( "logRotate" ) {} + virtual LockType locktype() const { return NONE; } + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return true; } + virtual bool run(const string& ns, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl) { + rotateLogs(); + return 1; + } + + } logRotateCmd; + + class ListCommandsCmd : public Command { + public: + virtual void help( stringstream &help ) const { help << "get a list of all db commands"; } + ListCommandsCmd() : Command( "listCommands", false ) {} + virtual LockType locktype() const { return NONE; } + virtual bool slaveOk() const { return true; } + virtual bool adminOnly() const { return false; } + virtual bool run(const string& ns, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl) { + BSONObjBuilder b( result.subobjStart( "commands" ) ); + for ( map<string,Command*>::iterator i=_commands->begin(); i!=_commands->end(); ++i ) { + Command * c = i->second; + + // don't show oldnames + if (i->first != c->name) + continue; + + BSONObjBuilder temp( b.subobjStart( c->name ) ); + + { + stringstream help; + c->help( help ); + temp.append( "help" , help.str() ); + } + temp.append( "lockType" , c->locktype() ); + temp.append( "slaveOk" , c->slaveOk() ); + temp.append( "adminOnly" , c->adminOnly() ); + //optionally indicates that the command can be forced to run on a slave/secondary + if ( c->slaveOverrideOk() ) temp.append( "slaveOverrideOk" , c->slaveOverrideOk() ); + temp.done(); + } + b.done(); + + return 1; + } + + } listCommandsCmd; + + bool CmdShutdown::shutdownHelper() { + Client * c = currentClient.get(); + if ( c ) { + c->shutdown(); + } + + log() << "terminating, shutdown command received" << endl; + + dbexit( EXIT_CLEAN , "shutdown called" , true ); // this never returns + assert(0); + return true; + } + + /* for testing purposes only */ + class CmdForceError : public Command { + public: + virtual void help( stringstream& help ) const { + help << "for testing purposes only. forces a user assertion exception"; + } + virtual bool logTheOp() { + return false; + } + virtual bool slaveOk() const { + return true; + } + virtual LockType locktype() const { return NONE; } + CmdForceError() : Command("forceerror") {} + bool run(const string& dbnamne, BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool fromRepl) { + uassert( 10038 , "forced error", false); + return true; + } + } cmdForceError; + + class AvailableQueryOptions : public Command { + public: + AvailableQueryOptions() : Command( "availableQueryOptions" , false , "availablequeryoptions" ) {} + virtual bool slaveOk() const { return true; } + virtual LockType locktype() const { return NONE; } + virtual bool run(const string& dbname , BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool) { + result << "options" << QueryOption_AllSupported; + return true; + } + } availableQueryOptionsCmd; + + class GetLogCmd : public Command { + public: + GetLogCmd() : Command( "getLog" ){} + + virtual bool slaveOk() const { return true; } + virtual LockType locktype() const { return NONE; } + virtual bool adminOnly() const { return true; } + + virtual void help( stringstream& help ) const { + help << "{ getLog : '*' } OR { getLog : 'global' }"; + } + + virtual bool run(const string& dbname , BSONObj& cmdObj, int, string& errmsg, BSONObjBuilder& result, bool) { + string p = cmdObj.firstElement().String(); + if ( p == "*" ) { + vector<string> names; + RamLog::getNames( names ); + + BSONArrayBuilder arr; + for ( unsigned i=0; i<names.size(); i++ ) { + arr.append( names[i] ); + } + + result.appendArray( "names" , arr.arr() ); + } + else { + RamLog* rl = RamLog::get( p ); + if ( ! rl ) { + errmsg = str::stream() << "no RamLog named: " << p; + return false; + } + + vector<const char*> lines; + rl->get( lines ); + + BSONArrayBuilder arr( result.subarrayStart( "log" ) ); + for ( unsigned i=0; i<lines.size(); i++ ) + arr.append( lines[i] ); + arr.done(); + } + return true; + } + + } getLogCmd; + +} |