diff options
Diffstat (limited to 'client/dbclient.cpp')
| -rw-r--r-- | client/dbclient.cpp | 1053 |
1 files changed, 0 insertions, 1053 deletions
diff --git a/client/dbclient.cpp b/client/dbclient.cpp deleted file mode 100644 index 6b9631b09ee..00000000000 --- a/client/dbclient.cpp +++ /dev/null @@ -1,1053 +0,0 @@ -// dbclient.cpp - connect to a Mongo database as a database, from C++ - -/* Copyright 2009 10gen Inc. - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include "pch.h" -#include "../db/pdfile.h" -#include "dbclient.h" -#include "../bson/util/builder.h" -#include "../db/jsobj.h" -#include "../db/json.h" -#include "../db/instance.h" -#include "../util/md5.hpp" -#include "../db/dbmessage.h" -#include "../db/cmdline.h" -#include "connpool.h" -#include "../s/util.h" -#include "syncclusterconnection.h" - -namespace mongo { - - void ConnectionString::_fillServers( string s ) { - - { - string::size_type idx = s.find( '/' ); - if ( idx != string::npos ) { - _setName = s.substr( 0 , idx ); - s = s.substr( idx + 1 ); - _type = SET; - } - } - - string::size_type idx; - while ( ( idx = s.find( ',' ) ) != string::npos ) { - _servers.push_back( s.substr( 0 , idx ) ); - s = s.substr( idx + 1 ); - } - _servers.push_back( s ); - - } - - void ConnectionString::_finishInit() { - stringstream ss; - if ( _type == SET ) - ss << _setName << "/"; - for ( unsigned i=0; i<_servers.size(); i++ ) { - if ( i > 0 ) - ss << ","; - ss << _servers[i].toString(); - } - _string = ss.str(); - } - - - DBClientBase* ConnectionString::connect( string& errmsg, double socketTimeout ) const { - switch ( _type ) { - case MASTER: { - DBClientConnection * c = new DBClientConnection(true); - c->setSoTimeout( socketTimeout ); - log(1) << "creating new connection to:" << _servers[0] << endl; - if ( ! c->connect( _servers[0] , errmsg ) ) { - delete c; - return 0; - } - log(1) << "connected connection!" << endl; - return c; - } - - case PAIR: - case SET: { - DBClientReplicaSet * set = new DBClientReplicaSet( _setName , _servers , socketTimeout ); - if( ! set->connect() ) { - delete set; - errmsg = "connect failed to set "; - errmsg += toString(); - return 0; - } - return set; - } - - case SYNC: { - // TODO , don't copy - list<HostAndPort> l; - for ( unsigned i=0; i<_servers.size(); i++ ) - l.push_back( _servers[i] ); - SyncClusterConnection* c = new SyncClusterConnection( l, socketTimeout ); - return c; - } - - case INVALID: - throw UserException( 13421 , "trying to connect to invalid ConnectionString" ); - break; - } - - assert( 0 ); - return 0; - } - - ConnectionString ConnectionString::parse( const string& host , string& errmsg ) { - - string::size_type i = host.find( '/' ); - if ( i != string::npos && i != 0) { - // replica set - return ConnectionString( SET , host.substr( i + 1 ) , host.substr( 0 , i ) ); - } - - int numCommas = str::count( host , ',' ); - - if( numCommas == 0 ) - return ConnectionString( HostAndPort( host ) ); - - if ( numCommas == 1 ) - return ConnectionString( PAIR , host ); - - if ( numCommas == 2 ) - return ConnectionString( SYNC , host ); - - errmsg = (string)"invalid hostname [" + host + "]"; - return ConnectionString(); // INVALID - } - - string ConnectionString::typeToString( ConnectionType type ) { - switch ( type ) { - case INVALID: - return "invalid"; - case MASTER: - return "master"; - case PAIR: - return "pair"; - case SET: - return "set"; - case SYNC: - return "sync"; - } - assert(0); - return ""; - } - - - Query& Query::where(const string &jscode, BSONObj scope) { - /* use where() before sort() and hint() and explain(), else this will assert. */ - assert( ! isComplex() ); - BSONObjBuilder b; - b.appendElements(obj); - b.appendWhere(jscode, scope); - obj = b.obj(); - return *this; - } - - void Query::makeComplex() { - if ( isComplex() ) - return; - BSONObjBuilder b; - b.append( "query", obj ); - obj = b.obj(); - } - - Query& Query::sort(const BSONObj& s) { - appendComplex( "orderby", s ); - return *this; - } - - Query& Query::hint(BSONObj keyPattern) { - appendComplex( "$hint", keyPattern ); - return *this; - } - - Query& Query::explain() { - appendComplex( "$explain", true ); - return *this; - } - - Query& Query::snapshot() { - appendComplex( "$snapshot", true ); - return *this; - } - - Query& Query::minKey( const BSONObj &val ) { - appendComplex( "$min", val ); - return *this; - } - - Query& Query::maxKey( const BSONObj &val ) { - appendComplex( "$max", val ); - return *this; - } - - bool Query::isComplex( bool * hasDollar ) const { - if ( obj.hasElement( "query" ) ) { - if ( hasDollar ) - hasDollar[0] = false; - return true; - } - - if ( obj.hasElement( "$query" ) ) { - if ( hasDollar ) - hasDollar[0] = true; - return true; - } - - return false; - } - - BSONObj Query::getFilter() const { - bool hasDollar; - if ( ! isComplex( &hasDollar ) ) - return obj; - - return obj.getObjectField( hasDollar ? "$query" : "query" ); - } - BSONObj Query::getSort() const { - if ( ! isComplex() ) - return BSONObj(); - BSONObj ret = obj.getObjectField( "orderby" ); - if (ret.isEmpty()) - ret = obj.getObjectField( "$orderby" ); - return ret; - } - BSONObj Query::getHint() const { - if ( ! isComplex() ) - return BSONObj(); - return obj.getObjectField( "$hint" ); - } - bool Query::isExplain() const { - return isComplex() && obj.getBoolField( "$explain" ); - } - - string Query::toString() const { - return obj.toString(); - } - - /* --- dbclientcommands --- */ - - bool DBClientWithCommands::isOk(const BSONObj& o) { - return o["ok"].trueValue(); - } - - bool DBClientWithCommands::isNotMasterErrorString( const BSONElement& e ) { - return e.type() == String && str::contains( e.valuestr() , "not master" ); - } - - - enum QueryOptions DBClientWithCommands::availableOptions() { - if ( !_haveCachedAvailableOptions ) { - BSONObj ret; - if ( runCommand( "admin", BSON( "availablequeryoptions" << 1 ), ret ) ) { - _cachedAvailableOptions = ( enum QueryOptions )( ret.getIntField( "options" ) ); - } - _haveCachedAvailableOptions = true; - } - return _cachedAvailableOptions; - } - - inline bool DBClientWithCommands::runCommand(const string &dbname, const BSONObj& cmd, BSONObj &info, int options) { - string ns = dbname + ".$cmd"; - info = findOne(ns, cmd, 0 , options); - return isOk(info); - } - - /* note - we build a bson obj here -- for something that is super common like getlasterror you - should have that object prebuilt as that would be faster. - */ - bool DBClientWithCommands::simpleCommand(const string &dbname, BSONObj *info, const string &command) { - BSONObj o; - if ( info == 0 ) - info = &o; - BSONObjBuilder b; - b.append(command, 1); - return runCommand(dbname, b.done(), *info); - } - - unsigned long long DBClientWithCommands::count(const string &myns, const BSONObj& query, int options, int limit, int skip ) { - NamespaceString ns(myns); - BSONObj cmd = _countCmd( myns , query , options , limit , skip ); - BSONObj res; - if( !runCommand(ns.db.c_str(), cmd, res, options) ) - uasserted(11010,string("count fails:") + res.toString()); - return res["n"].numberLong(); - } - - BSONObj DBClientWithCommands::_countCmd(const string &myns, const BSONObj& query, int options, int limit, int skip ) { - NamespaceString ns(myns); - BSONObjBuilder b; - b.append( "count" , ns.coll ); - b.append( "query" , query ); - if ( limit ) - b.append( "limit" , limit ); - if ( skip ) - b.append( "skip" , skip ); - return b.obj(); - } - - const BSONObj getlasterrorcmdobj = fromjson("{getlasterror:1}"); - - BSONObj DBClientWithCommands::getLastErrorDetailed() { - BSONObj info; - runCommand("admin", getlasterrorcmdobj, info); - return info; - } - - string DBClientWithCommands::getLastError() { - BSONObj info = getLastErrorDetailed(); - return getLastErrorString( info ); - } - - string DBClientWithCommands::getLastErrorString( const BSONObj& info ) { - BSONElement e = info["err"]; - if( e.eoo() ) return ""; - if( e.type() == Object ) return e.toString(); - return e.str(); - } - - const BSONObj getpreverrorcmdobj = fromjson("{getpreverror:1}"); - - BSONObj DBClientWithCommands::getPrevError() { - BSONObj info; - runCommand("admin", getpreverrorcmdobj, info); - return info; - } - - BSONObj getnoncecmdobj = fromjson("{getnonce:1}"); - - string DBClientWithCommands::createPasswordDigest( const string & username , const string & clearTextPassword ) { - md5digest d; - { - md5_state_t st; - md5_init(&st); - md5_append(&st, (const md5_byte_t *) username.data(), username.length()); - md5_append(&st, (const md5_byte_t *) ":mongo:", 7 ); - md5_append(&st, (const md5_byte_t *) clearTextPassword.data(), clearTextPassword.length()); - md5_finish(&st, d); - } - return digestToString( d ); - } - - bool DBClientWithCommands::auth(const string &dbname, const string &username, const string &password_text, string& errmsg, bool digestPassword) { - string password = password_text; - if( digestPassword ) - password = createPasswordDigest( username , password_text ); - - BSONObj info; - string nonce; - if( !runCommand(dbname, getnoncecmdobj, info) ) { - errmsg = "getnonce fails - connection problem?"; - return false; - } - { - BSONElement e = info.getField("nonce"); - assert( e.type() == String ); - nonce = e.valuestr(); - } - - BSONObj authCmd; - BSONObjBuilder b; - { - - b << "authenticate" << 1 << "nonce" << nonce << "user" << username; - md5digest d; - { - md5_state_t st; - md5_init(&st); - md5_append(&st, (const md5_byte_t *) nonce.c_str(), nonce.size() ); - md5_append(&st, (const md5_byte_t *) username.data(), username.length()); - md5_append(&st, (const md5_byte_t *) password.c_str(), password.size() ); - md5_finish(&st, d); - } - b << "key" << digestToString( d ); - authCmd = b.done(); - } - - if( runCommand(dbname, authCmd, info) ) - return true; - - errmsg = info.toString(); - return false; - } - - BSONObj ismastercmdobj = fromjson("{\"ismaster\":1}"); - - bool DBClientWithCommands::isMaster(bool& isMaster, BSONObj *info) { - BSONObj o; - if ( info == 0 ) - info = &o; - bool ok = runCommand("admin", ismastercmdobj, *info); - isMaster = info->getField("ismaster").trueValue(); - return ok; - } - - bool DBClientWithCommands::createCollection(const string &ns, long long size, bool capped, int max, BSONObj *info) { - assert(!capped||size); - BSONObj o; - if ( info == 0 ) info = &o; - BSONObjBuilder b; - string db = nsToDatabase(ns.c_str()); - b.append("create", ns.c_str() + db.length() + 1); - if ( size ) b.append("size", size); - if ( capped ) b.append("capped", true); - if ( max ) b.append("max", max); - return runCommand(db.c_str(), b.done(), *info); - } - - bool DBClientWithCommands::copyDatabase(const string &fromdb, const string &todb, const string &fromhost, BSONObj *info) { - BSONObj o; - if ( info == 0 ) info = &o; - BSONObjBuilder b; - b.append("copydb", 1); - b.append("fromhost", fromhost); - b.append("fromdb", fromdb); - b.append("todb", todb); - return runCommand("admin", b.done(), *info); - } - - bool DBClientWithCommands::setDbProfilingLevel(const string &dbname, ProfilingLevel level, BSONObj *info ) { - BSONObj o; - if ( info == 0 ) info = &o; - - if ( level ) { - // Create system.profile collection. If it already exists this does nothing. - // TODO: move this into the db instead of here so that all - // drivers don't have to do this. - string ns = dbname + ".system.profile"; - createCollection(ns.c_str(), 1024 * 1024, true, 0, info); - } - - BSONObjBuilder b; - b.append("profile", (int) level); - return runCommand(dbname, b.done(), *info); - } - - BSONObj getprofilingcmdobj = fromjson("{\"profile\":-1}"); - - bool DBClientWithCommands::getDbProfilingLevel(const string &dbname, ProfilingLevel& level, BSONObj *info) { - BSONObj o; - if ( info == 0 ) info = &o; - if ( runCommand(dbname, getprofilingcmdobj, *info) ) { - level = (ProfilingLevel) info->getIntField("was"); - return true; - } - return false; - } - - DBClientWithCommands::MROutput DBClientWithCommands::MRInline (BSON("inline" << 1)); - - BSONObj DBClientWithCommands::mapreduce(const string &ns, const string &jsmapf, const string &jsreducef, BSONObj query, MROutput output) { - BSONObjBuilder b; - b.append("mapreduce", nsGetCollection(ns)); - b.appendCode("map", jsmapf); - b.appendCode("reduce", jsreducef); - if( !query.isEmpty() ) - b.append("query", query); - b.append("out", output.out); - BSONObj info; - runCommand(nsGetDB(ns), b.done(), info); - return info; - } - - bool DBClientWithCommands::eval(const string &dbname, const string &jscode, BSONObj& info, BSONElement& retValue, BSONObj *args) { - BSONObjBuilder b; - b.appendCode("$eval", jscode); - if ( args ) - b.appendArray("args", *args); - bool ok = runCommand(dbname, b.done(), info); - if ( ok ) - retValue = info.getField("retval"); - return ok; - } - - bool DBClientWithCommands::eval(const string &dbname, const string &jscode) { - BSONObj info; - BSONElement retValue; - return eval(dbname, jscode, info, retValue); - } - - list<string> DBClientWithCommands::getDatabaseNames() { - BSONObj info; - uassert( 10005 , "listdatabases failed" , runCommand( "admin" , BSON( "listDatabases" << 1 ) , info ) ); - uassert( 10006 , "listDatabases.databases not array" , info["databases"].type() == Array ); - - list<string> names; - - BSONObjIterator i( info["databases"].embeddedObjectUserCheck() ); - while ( i.more() ) { - names.push_back( i.next().embeddedObjectUserCheck()["name"].valuestr() ); - } - - return names; - } - - list<string> DBClientWithCommands::getCollectionNames( const string& db ) { - list<string> names; - - string ns = db + ".system.namespaces"; - auto_ptr<DBClientCursor> c = query( ns.c_str() , BSONObj() ); - while ( c->more() ) { - string name = c->next()["name"].valuestr(); - if ( name.find( "$" ) != string::npos ) - continue; - names.push_back( name ); - } - return names; - } - - bool DBClientWithCommands::exists( const string& ns ) { - list<string> names; - - string db = nsGetDB( ns ) + ".system.namespaces"; - BSONObj q = BSON( "name" << ns ); - return count( db.c_str() , q, QueryOption_SlaveOk ) != 0; - } - - /* --- dbclientconnection --- */ - - bool DBClientConnection::auth(const string &dbname, const string &username, const string &password_text, string& errmsg, bool digestPassword) { - string password = password_text; - if( digestPassword ) - password = createPasswordDigest( username , password_text ); - - if( autoReconnect ) { - /* note we remember the auth info before we attempt to auth -- if the connection is broken, we will - then have it for the next autoreconnect attempt. - */ - pair<string,string> p = pair<string,string>(username, password); - authCache[dbname] = p; - } - - return DBClientBase::auth(dbname, username, password.c_str(), errmsg, false); - } - - /** query N objects from the database into an array. makes sense mostly when you want a small number of results. if a huge number, use - query() and iterate the cursor. - */ - void DBClientInterface::findN(vector<BSONObj>& out, const string& ns, Query query, int nToReturn, int nToSkip, const BSONObj *fieldsToReturn, int queryOptions) { - out.reserve(nToReturn); - - auto_ptr<DBClientCursor> c = - this->query(ns, query, nToReturn, nToSkip, fieldsToReturn, queryOptions); - - uassert( 10276 , str::stream() << "DBClientBase::findN: transport error: " << getServerAddress() << " query: " << query.toString(), c.get() ); - - if ( c->hasResultFlag( ResultFlag_ShardConfigStale ) ) - throw StaleConfigException( ns , "findN stale config" ); - - for( int i = 0; i < nToReturn; i++ ) { - if ( !c->more() ) - break; - out.push_back( c->nextSafe().copy() ); - } - } - - BSONObj DBClientInterface::findOne(const string &ns, const Query& query, const BSONObj *fieldsToReturn, int queryOptions) { - vector<BSONObj> v; - findN(v, ns, query, 1, 0, fieldsToReturn, queryOptions); - return v.empty() ? BSONObj() : v[0]; - } - - bool DBClientConnection::connect(const HostAndPort& server, string& errmsg) { - _server = server; - _serverString = _server.toString(); - return _connect( errmsg ); - } - - bool DBClientConnection::_connect( string& errmsg ) { - _serverString = _server.toString(); - // we keep around SockAddr for connection life -- maybe MessagingPort - // requires that? - server.reset(new SockAddr(_server.host().c_str(), _server.port())); - p.reset(new MessagingPort( _so_timeout, _logLevel )); - - if (server->getAddr() == "0.0.0.0") { - _failed = true; - return false; - } - - // if( _so_timeout == 0 ){ - // printStackTrace(); - // log() << "Connecting to server " << _serverString << " timeout " << _so_timeout << endl; - // } - if ( !p->connect(*server) ) { - stringstream ss; - ss << "couldn't connect to server " << _serverString; - errmsg = ss.str(); - _failed = true; - return false; - } - -#ifdef MONGO_SSL - if ( cmdLine.sslOnNormalPorts ) { - p->secure( sslManager() ); - } -#endif - - return true; - } - - - inline bool DBClientConnection::runCommand(const string &dbname, const BSONObj& cmd, BSONObj &info, int options) { - if ( DBClientWithCommands::runCommand( dbname , cmd , info , options ) ) - return true; - - if ( clientSet && isNotMasterErrorString( info["errmsg"] ) ) { - clientSet->isntMaster(); - // At this point, we've probably deleted *this* object, do *not* use afterward - } - - return false; - } - - - void DBClientConnection::_checkConnection() { - if ( !_failed ) - return; - if ( lastReconnectTry && time(0)-lastReconnectTry < 2 ) { - // we wait a little before reconnect attempt to avoid constant hammering. - // but we throw we don't want to try to use a connection in a bad state - throw SocketException( SocketException::FAILED_STATE , toString() ); - } - if ( !autoReconnect ) - throw SocketException( SocketException::FAILED_STATE , toString() ); - - lastReconnectTry = time(0); - log(_logLevel) << "trying reconnect to " << _serverString << endl; - string errmsg; - _failed = false; - if ( ! _connect(errmsg) ) { - _failed = true; - log(_logLevel) << "reconnect " << _serverString << " failed " << errmsg << endl; - throw SocketException( SocketException::CONNECT_ERROR , toString() ); - } - - log(_logLevel) << "reconnect " << _serverString << " ok" << endl; - for( map< string, pair<string,string> >::iterator i = authCache.begin(); i != authCache.end(); i++ ) { - const char *dbname = i->first.c_str(); - const char *username = i->second.first.c_str(); - const char *password = i->second.second.c_str(); - if( !DBClientBase::auth(dbname, username, password, errmsg, false) ) - log(_logLevel) << "reconnect: auth failed db:" << dbname << " user:" << username << ' ' << errmsg << '\n'; - } - } - - auto_ptr<DBClientCursor> DBClientBase::query(const string &ns, Query query, int nToReturn, - int nToSkip, const BSONObj *fieldsToReturn, int queryOptions , int batchSize ) { - auto_ptr<DBClientCursor> c( new DBClientCursor( this, - ns, query.obj, nToReturn, nToSkip, - fieldsToReturn, queryOptions , batchSize ) ); - if ( c->init() ) - return c; - return auto_ptr< DBClientCursor >( 0 ); - } - - auto_ptr<DBClientCursor> DBClientBase::getMore( const string &ns, long long cursorId, int nToReturn, int options ) { - auto_ptr<DBClientCursor> c( new DBClientCursor( this, ns, cursorId, nToReturn, options ) ); - if ( c->init() ) - return c; - return auto_ptr< DBClientCursor >( 0 ); - } - - struct DBClientFunConvertor { - void operator()( DBClientCursorBatchIterator &i ) { - while( i.moreInCurrentBatch() ) { - _f( i.nextSafe() ); - } - } - boost::function<void(const BSONObj &)> _f; - }; - - unsigned long long DBClientConnection::query( boost::function<void(const BSONObj&)> f, const string& ns, Query query, const BSONObj *fieldsToReturn, int queryOptions ) { - DBClientFunConvertor fun; - fun._f = f; - boost::function<void(DBClientCursorBatchIterator &)> ptr( fun ); - return DBClientConnection::query( ptr, ns, query, fieldsToReturn, queryOptions ); - } - - unsigned long long DBClientConnection::query( boost::function<void(DBClientCursorBatchIterator &)> f, const string& ns, Query query, const BSONObj *fieldsToReturn, int queryOptions ) { - // mask options - queryOptions &= (int)( QueryOption_NoCursorTimeout | QueryOption_SlaveOk ); - unsigned long long n = 0; - - bool doExhaust = ( availableOptions() & QueryOption_Exhaust ); - if ( doExhaust ) { - queryOptions |= (int)QueryOption_Exhaust; - } - auto_ptr<DBClientCursor> c( this->query(ns, query, 0, 0, fieldsToReturn, queryOptions) ); - uassert( 13386, "socket error for mapping query", c.get() ); - - if ( !doExhaust ) { - while( c->more() ) { - DBClientCursorBatchIterator i( *c ); - f( i ); - n += i.n(); - } - return n; - } - - try { - while( 1 ) { - while( c->moreInCurrentBatch() ) { - DBClientCursorBatchIterator i( *c ); - f( i ); - n += i.n(); - } - - if( c->getCursorId() == 0 ) - break; - - c->exhaustReceiveMore(); - } - } - catch(std::exception&) { - /* connection CANNOT be used anymore as more data may be on the way from the server. - we have to reconnect. - */ - _failed = true; - p->shutdown(); - throw; - } - - return n; - } - - void DBClientBase::insert( const string & ns , BSONObj obj , int flags) { - Message toSend; - - BufBuilder b; - b.appendNum( flags ); - b.appendStr( ns ); - obj.appendSelfToBufBuilder( b ); - - toSend.setData( dbInsert , b.buf() , b.len() ); - - say( toSend ); - } - - void DBClientBase::insert( const string & ns , const vector< BSONObj > &v , int flags) { - Message toSend; - - BufBuilder b; - b.appendNum( flags ); - b.appendStr( ns ); - for( vector< BSONObj >::const_iterator i = v.begin(); i != v.end(); ++i ) - i->appendSelfToBufBuilder( b ); - - toSend.setData( dbInsert, b.buf(), b.len() ); - - say( toSend ); - } - - void DBClientBase::remove( const string & ns , Query obj , bool justOne ) { - Message toSend; - - BufBuilder b; - int opts = 0; - b.appendNum( opts ); - b.appendStr( ns ); - - int flags = 0; - if ( justOne ) - flags |= RemoveOption_JustOne; - b.appendNum( flags ); - - obj.obj.appendSelfToBufBuilder( b ); - - toSend.setData( dbDelete , b.buf() , b.len() ); - - say( toSend ); - } - - void DBClientBase::update( const string & ns , Query query , BSONObj obj , bool upsert , bool multi ) { - - BufBuilder b; - b.appendNum( (int)0 ); // reserved - b.appendStr( ns ); - - int flags = 0; - if ( upsert ) flags |= UpdateOption_Upsert; - if ( multi ) flags |= UpdateOption_Multi; - b.appendNum( flags ); - - query.obj.appendSelfToBufBuilder( b ); - obj.appendSelfToBufBuilder( b ); - - Message toSend; - toSend.setData( dbUpdate , b.buf() , b.len() ); - - say( toSend ); - - - } - - - - auto_ptr<DBClientCursor> DBClientWithCommands::getIndexes( const string &ns ) { - return query( Namespace( ns.c_str() ).getSisterNS( "system.indexes" ).c_str() , BSON( "ns" << ns ) ); - } - - void DBClientWithCommands::dropIndex( const string& ns , BSONObj keys ) { - dropIndex( ns , genIndexName( keys ) ); - } - - - void DBClientWithCommands::dropIndex( const string& ns , const string& indexName ) { - BSONObj info; - if ( ! runCommand( nsToDatabase( ns.c_str() ) , - BSON( "deleteIndexes" << NamespaceString( ns ).coll << "index" << indexName ) , - info ) ) { - log(_logLevel) << "dropIndex failed: " << info << endl; - uassert( 10007 , "dropIndex failed" , 0 ); - } - resetIndexCache(); - } - - void DBClientWithCommands::dropIndexes( const string& ns ) { - BSONObj info; - uassert( 10008 , "dropIndexes failed" , runCommand( nsToDatabase( ns.c_str() ) , - BSON( "deleteIndexes" << NamespaceString( ns ).coll << "index" << "*") , - info ) ); - resetIndexCache(); - } - - void DBClientWithCommands::reIndex( const string& ns ) { - list<BSONObj> all; - auto_ptr<DBClientCursor> i = getIndexes( ns ); - while ( i->more() ) { - all.push_back( i->next().getOwned() ); - } - - dropIndexes( ns ); - - for ( list<BSONObj>::iterator i=all.begin(); i!=all.end(); i++ ) { - BSONObj o = *i; - insert( Namespace( ns.c_str() ).getSisterNS( "system.indexes" ).c_str() , o ); - } - - } - - - string DBClientWithCommands::genIndexName( const BSONObj& keys ) { - stringstream ss; - - bool first = 1; - for ( BSONObjIterator i(keys); i.more(); ) { - BSONElement f = i.next(); - - if ( first ) - first = 0; - else - ss << "_"; - - ss << f.fieldName() << "_"; - if( f.isNumber() ) - ss << f.numberInt(); - } - return ss.str(); - } - - bool DBClientWithCommands::ensureIndex( const string &ns , BSONObj keys , bool unique, const string & name , bool cache, bool background, int version ) { - BSONObjBuilder toSave; - toSave.append( "ns" , ns ); - toSave.append( "key" , keys ); - - string cacheKey(ns); - cacheKey += "--"; - - if ( name != "" ) { - toSave.append( "name" , name ); - cacheKey += name; - } - else { - string nn = genIndexName( keys ); - toSave.append( "name" , nn ); - cacheKey += nn; - } - - if( version >= 0 ) - toSave.append("v", version); - - if ( unique ) - toSave.appendBool( "unique", unique ); - - if( background ) - toSave.appendBool( "background", true ); - - if ( _seenIndexes.count( cacheKey ) ) - return 0; - - if ( cache ) - _seenIndexes.insert( cacheKey ); - - insert( Namespace( ns.c_str() ).getSisterNS( "system.indexes" ).c_str() , toSave.obj() ); - return 1; - } - - void DBClientWithCommands::resetIndexCache() { - _seenIndexes.clear(); - } - - /* -- DBClientCursor ---------------------------------------------- */ - -#ifdef _DEBUG -#define CHECK_OBJECT( o , msg ) massert( 10337 , (string)"object not valid" + (msg) , (o).isValid() ) -#else -#define CHECK_OBJECT( o , msg ) -#endif - - void assembleRequest( const string &ns, BSONObj query, int nToReturn, int nToSkip, const BSONObj *fieldsToReturn, int queryOptions, Message &toSend ) { - CHECK_OBJECT( query , "assembleRequest query" ); - // see query.h for the protocol we are using here. - BufBuilder b; - int opts = queryOptions; - b.appendNum(opts); - b.appendStr(ns); - b.appendNum(nToSkip); - b.appendNum(nToReturn); - query.appendSelfToBufBuilder(b); - if ( fieldsToReturn ) - fieldsToReturn->appendSelfToBufBuilder(b); - toSend.setData(dbQuery, b.buf(), b.len()); - } - - void DBClientConnection::say( Message &toSend, bool isRetry ) { - checkConnection(); - try { - port().say( toSend ); - } - catch( SocketException & ) { - _failed = true; - throw; - } - } - - void DBClientConnection::sayPiggyBack( Message &toSend ) { - port().piggyBack( toSend ); - } - - bool DBClientConnection::recv( Message &m ) { - return port().recv(m); - } - - bool DBClientConnection::call( Message &toSend, Message &response, bool assertOk , string * actualServer ) { - /* todo: this is very ugly messagingport::call returns an error code AND can throw - an exception. we should make it return void and just throw an exception anytime - it fails - */ - checkConnection(); - try { - if ( !port().call(toSend, response) ) { - _failed = true; - if ( assertOk ) - uasserted( 10278 , str::stream() << "dbclient error communicating with server: " << getServerAddress() ); - - return false; - } - } - catch( SocketException & ) { - _failed = true; - throw; - } - return true; - } - - BSONElement getErrField(const BSONObj& o) { - BSONElement first = o.firstElement(); - if( strcmp(first.fieldName(), "$err") == 0 ) - return first; - - // temp - will be DEV only later - /*DEV*/ - if( 1 ) { - BSONElement e = o["$err"]; - if( !e.eoo() ) { - wassert(false); - } - return e; - } - - return BSONElement(); - } - - bool hasErrField( const BSONObj& o ){ - return ! getErrField( o ).eoo(); - } - - void DBClientConnection::checkResponse( const char *data, int nReturned, bool* retry, string* host ) { - /* check for errors. the only one we really care about at - * this stage is "not master" - */ - - *retry = false; - *host = _serverString; - - if ( clientSet && nReturned ) { - assert(data); - BSONObj o(data); - if ( isNotMasterErrorString( getErrField(o) ) ) { - clientSet->isntMaster(); - } - } - } - - void DBClientConnection::killCursor( long long cursorId ) { - StackBufBuilder b; - b.appendNum( (int)0 ); // reserved - b.appendNum( (int)1 ); // number - b.appendNum( cursorId ); - - Message m; - m.setData( dbKillCursors , b.buf() , b.len() ); - - if ( _lazyKillCursor ) - sayPiggyBack( m ); - else - say(m); - } - -#ifdef MONGO_SSL - SSLManager* DBClientConnection::sslManager() { - if ( _sslManager ) - return _sslManager; - - SSLManager* s = new SSLManager(true); - _sslManager = s; - return s; - } - - SSLManager* DBClientConnection::_sslManager = 0; -#endif - - AtomicUInt DBClientConnection::_numConnections; - bool DBClientConnection::_lazyKillCursor = true; - - - bool serverAlive( const string &uri ) { - DBClientConnection c( false, 0, 20 ); // potentially the connection to server could fail while we're checking if it's alive - so use timeouts - string err; - if ( !c.connect( uri, err ) ) - return false; - if ( !c.simpleCommand( "admin", 0, "ping" ) ) - return false; - return true; - } - -} // namespace mongo |