summaryrefslogtreecommitdiff
path: root/tools/tool.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tools/tool.cpp')
-rw-r--r--tools/tool.cpp410
1 files changed, 230 insertions, 180 deletions
diff --git a/tools/tool.cpp b/tools/tool.cpp
index 8243a45..c92a0c4 100644
--- a/tools/tool.cpp
+++ b/tools/tool.cpp
@@ -1,3 +1,19 @@
+/*
+ * Copyright (C) 2010 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/>.
+ */
+
// Tool.cpp
#include "tool.h"
@@ -14,225 +30,259 @@ using namespace mongo;
namespace po = boost::program_options;
-mongo::Tool::Tool( string name , string defaultDB , string defaultCollection ) :
- _name( name ) , _db( defaultDB ) , _coll( defaultCollection ) , _conn(0), _paired(false) {
-
- _options = new po::options_description( "options" );
- _options->add_options()
- ("help","produce help message")
- ("host,h",po::value<string>(), "mongo host to connect to" )
- ("db,d",po::value<string>(), "database to use" )
- ("collection,c",po::value<string>(), "collection to use (some commands)" )
- ("username,u",po::value<string>(), "username" )
- ("password,p",po::value<string>(), "password" )
- ("dbpath",po::value<string>(), "directly access mongod data files in this path, instead of connecting to a mongod instance" )
- ("verbose,v", "be more verbose (include multiple times for more verbosity e.g. -vvvvv)")
- ;
-
- _hidden_options = new po::options_description( name + " hidden options" );
-
- /* support for -vv -vvvv etc. */
- for (string s = "vv"; s.length() <= 10; s.append("v")) {
- _hidden_options->add_options()(s.c_str(), "verbose");
- }
-}
-
-mongo::Tool::~Tool(){
- delete( _options );
- delete( _hidden_options );
- if ( _conn )
- delete _conn;
-}
-
-void mongo::Tool::printExtraHelp( ostream & out ){
-}
+namespace mongo {
-void mongo::Tool::printHelp(ostream &out) {
- printExtraHelp(out);
- _options->print(out);
-}
+ Tool::Tool( string name , bool localDBAllowed , string defaultDB , string defaultCollection ) :
+ _name( name ) , _db( defaultDB ) , _coll( defaultCollection ) , _conn(0), _paired(false) {
+
+ _options = new po::options_description( "options" );
+ _options->add_options()
+ ("help","produce help message")
+ ("verbose,v", "be more verbose (include multiple times for more verbosity e.g. -vvvvv)")
+ ("host,h",po::value<string>(), "mongo host to connect to (\"left,right\" for pairs)" )
+ ("db,d",po::value<string>(), "database to use" )
+ ("collection,c",po::value<string>(), "collection to use (some commands)" )
+ ("username,u",po::value<string>(), "username" )
+ ("password,p",po::value<string>(), "password" )
+ ;
+ if ( localDBAllowed )
+ _options->add_options()
+ ("dbpath",po::value<string>(), "directly access mongod data "
+ "files in the given path, instead of connecting to a mongod "
+ "instance - needs to lock the data directory, so cannot be "
+ "used if a mongod is currently accessing the same path" )
+ ("directoryperdb", "if dbpath specified, each db is in a separate directory" )
+ ;
+
+ _hidden_options = new po::options_description( name + " hidden options" );
+
+ /* support for -vv -vvvv etc. */
+ for (string s = "vv"; s.length() <= 10; s.append("v")) {
+ _hidden_options->add_options()(s.c_str(), "verbose");
+ }
+ }
-int mongo::Tool::main( int argc , char ** argv ){
- boost::filesystem::path::default_name_check( boost::filesystem::no_check );
-
- _name = argv[0];
-
- /* using the same style as db.cpp */
- int command_line_style = (((po::command_line_style::unix_style ^
- po::command_line_style::allow_guessing) |
- po::command_line_style::allow_long_disguise) ^
- po::command_line_style::allow_sticky);
- try {
- po::options_description all_options("all options");
- all_options.add(*_options).add(*_hidden_options);
-
- po::store( po::command_line_parser( argc , argv ).
- options(all_options).
- positional( _positonalOptions ).
- style(command_line_style).run() , _params );
-
- po::notify( _params );
- } catch (po::error &e) {
- cout << "ERROR: " << e.what() << endl << endl;
- printHelp(cout);
- return EXIT_BADOPTIONS;
+ Tool::~Tool(){
+ delete( _options );
+ delete( _hidden_options );
+ if ( _conn )
+ delete _conn;
}
- if ( _params.count( "help" ) ){
- printHelp(cerr);
- return 0;
+ void Tool::printExtraHelp( ostream & out ){
}
- if ( _params.count( "verbose" ) ) {
- logLevel = 1;
+ void Tool::printHelp(ostream &out) {
+ printExtraHelp(out);
+ _options->print(out);
}
- for (string s = "vv"; s.length() <= 10; s.append("v")) {
- if (_params.count(s)) {
- logLevel = s.length();
+ int Tool::main( int argc , char ** argv ){
+ static StaticObserver staticObserver;
+
+ cmdLine.prealloc = false;
+
+ boost::filesystem::path::default_name_check( boost::filesystem::no_check );
+
+ _name = argv[0];
+
+ /* using the same style as db.cpp */
+ int command_line_style = (((po::command_line_style::unix_style ^
+ po::command_line_style::allow_guessing) |
+ po::command_line_style::allow_long_disguise) ^
+ po::command_line_style::allow_sticky);
+ try {
+ po::options_description all_options("all options");
+ all_options.add(*_options).add(*_hidden_options);
+
+ po::store( po::command_line_parser( argc , argv ).
+ options(all_options).
+ positional( _positonalOptions ).
+ style(command_line_style).run() , _params );
+
+ po::notify( _params );
+ } catch (po::error &e) {
+ cerr << "ERROR: " << e.what() << endl << endl;
+ printHelp(cerr);
+ return EXIT_BADOPTIONS;
}
- }
- if ( ! hasParam( "dbpath" ) ) {
- _host = "127.0.0.1";
- if ( _params.count( "host" ) )
- _host = _params["host"].as<string>();
+ if ( _params.count( "help" ) ){
+ printHelp(cerr);
+ return 0;
+ }
- if ( _host.find( "," ) == string::npos ){
- DBClientConnection * c = new DBClientConnection();
- _conn = c;
+ if ( _params.count( "verbose" ) ) {
+ logLevel = 1;
+ }
- string errmsg;
- if ( ! c->connect( _host , errmsg ) ){
- cerr << "couldn't connect to [" << _host << "] " << errmsg << endl;
- return -1;
+ for (string s = "vv"; s.length() <= 10; s.append("v")) {
+ if (_params.count(s)) {
+ logLevel = s.length();
}
}
- else {
- log(1) << "using pairing" << endl;
- DBClientPaired * c = new DBClientPaired();
- _paired = true;
- _conn = c;
- if ( ! c->connect( _host ) ){
- cerr << "couldn't connect to paired server: " << _host << endl;
+ bool useDirectClient = hasParam( "dbpath" );
+
+ if ( ! useDirectClient ) {
+ _host = "127.0.0.1";
+ if ( _params.count( "host" ) )
+ _host = _params["host"].as<string>();
+
+ if ( _host.find( "," ) == string::npos ){
+ DBClientConnection * c = new DBClientConnection();
+ _conn = c;
+
+ string errmsg;
+ if ( ! c->connect( _host , errmsg ) ){
+ cerr << "couldn't connect to [" << _host << "] " << errmsg << endl;
+ return -1;
+ }
+ }
+ else {
+ log(1) << "using pairing" << endl;
+ DBClientPaired * c = new DBClientPaired();
+ _paired = true;
+ _conn = c;
+
+ if ( ! c->connect( _host ) ){
+ cerr << "couldn't connect to paired server: " << _host << endl;
+ return -1;
+ }
+ }
+
+ cerr << "connected to: " << _host << endl;
+ }
+ else {
+ if ( _params.count( "directoryperdb" ) ) {
+ directoryperdb = true;
+ }
+ Client::initThread("tools");
+ _conn = new DBDirectClient();
+ _host = "DIRECT";
+ static string myDbpath = getParam( "dbpath" );
+ dbpath = myDbpath.c_str();
+ try {
+ acquirePathLock();
+ }
+ catch ( DBException& e ){
+ cerr << endl << "If you are running a mongod on the same "
+ "path you should connect to that instead of direct data "
+ "file access" << endl << endl;
+ dbexit( EXIT_CLEAN );
return -1;
}
+
+ theFileAllocator().start();
}
- cerr << "connected to: " << _host << endl;
- }
- else {
- Client::initThread("tools");
- _conn = new DBDirectClient();
- _host = "DIRECT";
- static string myDbpath = getParam( "dbpath" );
- mongo::dbpath = myDbpath.c_str();
- mongo::acquirePathLock();
- theFileAllocator().start();
- }
+ if ( _params.count( "db" ) )
+ _db = _params["db"].as<string>();
- if ( _params.count( "db" ) )
- _db = _params["db"].as<string>();
+ if ( _params.count( "collection" ) )
+ _coll = _params["collection"].as<string>();
- if ( _params.count( "collection" ) )
- _coll = _params["collection"].as<string>();
+ if ( _params.count( "username" ) )
+ _username = _params["username"].as<string>();
- if ( _params.count( "username" ) )
- _username = _params["username"].as<string>();
+ if ( _params.count( "password" ) )
+ _password = _params["password"].as<string>();
- if ( _params.count( "password" ) )
- _password = _params["password"].as<string>();
+ int ret = -1;
+ try {
+ ret = run();
+ }
+ catch ( DBException& e ){
+ cerr << "assertion: " << e.toString() << endl;
+ ret = -1;
+ }
+
+ if ( currentClient.get() )
+ currentClient->shutdown();
- int ret = -1;
- try {
- ret = run();
- }
- catch ( DBException& e ){
- cerr << "assertion: " << e.toString() << endl;
- ret = -1;
+ if ( useDirectClient )
+ dbexit( EXIT_CLEAN );
+ return ret;
}
-
- if ( currentClient.get() )
- currentClient->shutdown();
-
- return ret;
-}
-mongo::DBClientBase& mongo::Tool::conn( bool slaveIfPaired ){
- if ( _paired && slaveIfPaired )
- return ((DBClientPaired*)_conn)->slaveConn();
- return *_conn;
-}
+ DBClientBase& Tool::conn( bool slaveIfPaired ){
+ if ( _paired && slaveIfPaired )
+ return ((DBClientPaired*)_conn)->slaveConn();
+ return *_conn;
+ }
-void mongo::Tool::addFieldOptions(){
- add_options()
- ("fields,f" , po::value<string>() , "comma seperated list of field names e.g. -f name,age" )
- ("fieldFile" , po::value<string>() , "file with fields names - 1 per line" )
- ;
-}
+ void Tool::addFieldOptions(){
+ add_options()
+ ("fields,f" , po::value<string>() , "comma seperated list of field names e.g. -f name,age" )
+ ("fieldFile" , po::value<string>() , "file with fields names - 1 per line" )
+ ;
+ }
-void mongo::Tool::needFields(){
+ void Tool::needFields(){
- if ( hasParam( "fields" ) ){
- BSONObjBuilder b;
+ if ( hasParam( "fields" ) ){
+ BSONObjBuilder b;
- string fields_arg = getParam("fields");
- pcrecpp::StringPiece input(fields_arg);
+ string fields_arg = getParam("fields");
+ pcrecpp::StringPiece input(fields_arg);
- string f;
- pcrecpp::RE re("([\\w\\.]+),?" );
- while ( re.Consume( &input, &f ) ){
- _fields.push_back( f );
- b.append( f.c_str() , 1 );
- }
+ string f;
+ pcrecpp::RE re("([\\w\\.\\s]+),?" );
+ while ( re.Consume( &input, &f ) ){
+ _fields.push_back( f );
+ b.append( f.c_str() , 1 );
+ }
- _fieldsObj = b.obj();
- return;
- }
+ _fieldsObj = b.obj();
+ return;
+ }
- if ( hasParam( "fieldFile" ) ){
- string fn = getParam( "fieldFile" );
- if ( ! exists( fn ) )
- throw UserException( 9999 , ((string)"file: " + fn ) + " doesn't exist" );
-
- const int BUF_SIZE = 1024;
- char line[ 1024 + 128];
- ifstream file( fn.c_str() );
-
- BSONObjBuilder b;
- while ( file.rdstate() == ios_base::goodbit ){
- file.getline( line , BUF_SIZE );
- const char * cur = line;
- while ( isspace( cur[0] ) ) cur++;
- if ( strlen( cur ) == 0 )
- continue;
-
- _fields.push_back( cur );
- b.append( cur , 1 );
+ if ( hasParam( "fieldFile" ) ){
+ string fn = getParam( "fieldFile" );
+ if ( ! exists( fn ) )
+ throw UserException( 9999 , ((string)"file: " + fn ) + " doesn't exist" );
+
+ const int BUF_SIZE = 1024;
+ char line[ 1024 + 128];
+ ifstream file( fn.c_str() );
+
+ BSONObjBuilder b;
+ while ( file.rdstate() == ios_base::goodbit ){
+ file.getline( line , BUF_SIZE );
+ const char * cur = line;
+ while ( isspace( cur[0] ) ) cur++;
+ if ( strlen( cur ) == 0 )
+ continue;
+
+ _fields.push_back( cur );
+ b.append( cur , 1 );
+ }
+ _fieldsObj = b.obj();
+ return;
}
- _fieldsObj = b.obj();
- return;
+
+ throw UserException( 9998 , "you need to specify fields" );
}
- throw UserException( 9998 , "you need to specify fields" );
-}
+ void Tool::auth( string dbname ){
+ if ( ! dbname.size() )
+ dbname = _db;
+
+ if ( ! ( _username.size() || _password.size() ) )
+ return;
-void mongo::Tool::auth( string dbname ){
- if ( ! dbname.size() )
- dbname = _db;
+ string errmsg;
+ if ( _conn->auth( dbname , _username , _password , errmsg ) )
+ return;
- if ( ! ( _username.size() || _password.size() ) )
- return;
+ // try against the admin db
+ string err2;
+ if ( _conn->auth( "admin" , _username , _password , errmsg ) )
+ return;
- string errmsg;
- if ( _conn->auth( dbname , _username , _password , errmsg ) )
- return;
+ throw UserException( 9997 , (string)"auth failed: " + errmsg );
+ }
- // try against the admin db
- string err2;
- if ( _conn->auth( "admin" , _username , _password , errmsg ) )
- return;
- throw mongo::UserException( 9997 , (string)"auth failed: " + errmsg );
+ void setupSignals(){}
}
generated by cgit v1.2.3 (git 2.39.1) at 2025-04-03 14:12:51 +0000