diff options
Diffstat (limited to 'src/mongo/s/client.h')
-rw-r--r-- | src/mongo/s/client.h | 128 |
1 files changed, 128 insertions, 0 deletions
diff --git a/src/mongo/s/client.h b/src/mongo/s/client.h new file mode 100644 index 00000000000..1237f66b88a --- /dev/null +++ b/src/mongo/s/client.h @@ -0,0 +1,128 @@ +// @file s/client.h + +/* + * 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/>. + */ +#pragma once + +#include "../pch.h" +#include "writeback_listener.h" +#include "../db/security.h" +#include "../db/client_common.h" + +namespace mongo { + + /** + * holds information about a client connected to a mongos + * 1 per client socket + * currently implemented with a thread local + */ + class ClientInfo : public ClientBasic { + public: + ClientInfo(); + ~ClientInfo(); + + /** new request from client, adjusts internal state */ + void newRequest( AbstractMessagingPort* p = 0 ); + + /** client disconnected */ + void disconnect(); + + bool hasRemote() const { return true; } + + /** + * @return remote socket address of the client + */ + HostAndPort getRemote() const { return _remote; } + + /** + * notes that this client use this shard + * keeps track of all shards accessed this request + */ + void addShard( const string& shard ); + + /** + * gets shards used on the previous request + */ + set<string> * getPrev() const { return _prev; }; + + /** + * gets all shards we've accessed since the last time we called clearSinceLastGetError + */ + const set<string>& sinceLastGetError() const { return _sinceLastGetError; } + + /** + * clears list of shards we've talked to + */ + void clearSinceLastGetError() { _sinceLastGetError.clear(); } + + + /** + * resets the list of shards using to process the current request + */ + void clearCurrentShards(){ _cur->clear(); } + + /** + * calls getLastError + * resets shards since get last error + * @return if the command was ok or if there was an error + */ + bool getLastError( const BSONObj& options , BSONObjBuilder& result , bool fromWriteBackListener = false ); + + /** @return if its ok to auto split from this client */ + bool autoSplitOk() const { return _autoSplitOk; } + + void noAutoSplit() { _autoSplitOk = false; } + + static ClientInfo * get(); + const AuthenticationInfo* getAuthenticationInfo() const { return (AuthenticationInfo*)&_ai; } + AuthenticationInfo* getAuthenticationInfo() { return (AuthenticationInfo*)&_ai; } + bool isAdmin() { return _ai.isAuthorized( "admin" ); } + private: + AuthenticationInfo _ai; + struct WBInfo { + WBInfo( const WriteBackListener::ConnectionIdent& c , OID o ) : ident( c ) , id( o ) {} + WriteBackListener::ConnectionIdent ident; + OID id; + }; + + // for getLastError + void _addWriteBack( vector<WBInfo>& all , const BSONObj& o ); + vector<BSONObj> _handleWriteBacks( vector<WBInfo>& all , bool fromWriteBackListener ); + + + int _id; // unique client id + HostAndPort _remote; // server:port of remote socket end + + // we use _a and _b to store shards we've talked to on the current request and the previous + // we use 2 so we can flip for getLastError type operations + + set<string> _a; // actual set for _cur or _prev + set<string> _b; // " + + set<string> * _cur; // pointer to _a or _b depending on state + set<string> * _prev; // "" + + + set<string> _sinceLastGetError; // all shards accessed since last getLastError + + int _lastAccess; + bool _autoSplitOk; + + static boost::thread_specific_ptr<ClientInfo> _tlInfo; + }; + + +} |