diff options
Diffstat (limited to 's/shard.h')
| -rw-r--r-- | s/shard.h | 251 |
1 files changed, 251 insertions, 0 deletions
diff --git a/s/shard.h b/s/shard.h new file mode 100644 index 0000000..ee807d8 --- /dev/null +++ b/s/shard.h @@ -0,0 +1,251 @@ +// shard.h + +/** +* 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/>. +*/ + +#pragma once + +#include "../pch.h" +#include "../client/connpool.h" + +namespace mongo { + + class ShardConnection; + class ShardStatus; + + class Shard { + public: + Shard() + : _name("") , _addr("") , _maxSize(0) , _isDraining( false ){ + } + + Shard( const string& name , const string& addr, long long maxSize = 0 , bool isDraining = false ) + : _name(name) , _addr( addr ) , _maxSize( maxSize ) , _isDraining( isDraining ){ + } + + Shard( const string& ident ){ + reset( ident ); + } + + Shard( const Shard& other ) + : _name( other._name ) , _addr( other._addr ) , _maxSize( other._maxSize ) , _isDraining( other._isDraining ){ + } + + Shard( const Shard* other ) + : _name( other->_name ) , _addr( other->_addr ), _maxSize( other->_maxSize ) , _isDraining( other->_isDraining ){ + } + + static Shard make( const string& ident ){ + Shard s; + s.reset( ident ); + return s; + } + + static bool isAShard( const string& ident ); + + /** + * @param ident either name or address + */ + void reset( const string& ident ); + + void setAddress( const string& addr , bool authoritative = false ); + + string getName() const { + assert( _name.size() ); + return _name; + } + + string getConnString() const { + assert( _addr.size() ); + return _addr; + } + + long long getMaxSize() const { + return _maxSize; + } + + bool isDraining() const { + return _isDraining; + } + + string toString() const { + return _name + ":" + _addr; + } + + friend ostream& operator << (ostream& out, const Shard& s) { + return (out << s.toString()); + } + + bool operator==( const Shard& s ) const { + bool n = _name == s._name; + bool a = _addr == s._addr; + + assert( n == a ); // names and address are 1 to 1 + return n; + } + + bool operator!=( const Shard& s ) const { + bool n = _name == s._name; + bool a = _addr == s._addr; + return ! ( n && a ); + } + + + bool operator==( const string& s ) const { + return _name == s || _addr == s; + } + + bool operator!=( const string& s ) const { + return _name != s && _addr != s; + } + + bool operator<(const Shard& o) const { + return _name < o._name; + } + + bool ok() const { + return _addr.size() > 0 && _addr.size() > 0; + } + + BSONObj runCommand( const string& db , const string& simple ) const { + return runCommand( db , BSON( simple << 1 ) ); + } + BSONObj runCommand( const string& db , const BSONObj& cmd ) const ; + + ShardStatus getStatus() const ; + + static void getAllShards( vector<Shard>& all ); + static void printShardInfo( ostream& out ); + + /** + * picks a Shard for more load + */ + static Shard pick(); + + static void reloadShardInfo(); + + static void removeShard( const string& name ); + + static bool isMember( const string& addr ); + + static Shard EMPTY; + + private: + string _name; + string _addr; + long long _maxSize; // in MBytes, 0 is unlimited + bool _isDraining; // shard is currently being removed + }; + + class ShardStatus { + public: + + ShardStatus( const Shard& shard , const BSONObj& obj ); + + friend ostream& operator << (ostream& out, const ShardStatus& s) { + out << s.toString(); + return out; + } + + string toString() const { + stringstream ss; + ss << "shard: " << _shard << " mapped: " << _mapped << " writeLock: " << _writeLock; + return ss.str(); + } + + bool operator<( const ShardStatus& other ) const{ + return _mapped < other._mapped; + } + + Shard shard() const { + return _shard; + } + + long long mapped() const { + return _mapped; + } + + private: + Shard _shard; + long long _mapped; + double _writeLock; + }; + + class ShardConnection : public AScopedConnection { + public: + ShardConnection( const Shard * s , const string& ns ); + ShardConnection( const Shard& s , const string& ns ); + ShardConnection( const string& addr , const string& ns ); + + ~ShardConnection(); + + void done(); + void kill(); + + DBClientBase& conn(){ + _finishInit(); + assert( _conn ); + return *_conn; + } + + DBClientBase* operator->(){ + _finishInit(); + assert( _conn ); + return _conn; + } + + DBClientBase* get(){ + _finishInit(); + assert( _conn ); + return _conn; + } + + string getHost() const { + return _addr; + } + + bool setVersion() { + _finishInit(); + return _setVersion; + } + + static void sync(); + + void donotCheckVersion(){ + _setVersion = false; + _finishedInit = true; + } + + /** + this just passes through excpet it checks for stale configs + */ + bool runCommand( const string& db , const BSONObj& cmd , BSONObj& res ); + + /** checks all of my thread local connections for the version of this ns */ + static void checkMyConnectionVersions( const string & ns ); + + private: + void _init(); + void _finishInit(); + + bool _finishedInit; + + string _addr; + string _ns; + DBClientBase* _conn; + bool _setVersion; + }; +} |