mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-30 00:56:44 +01:00
140 lines
3.5 KiB
C++
140 lines
3.5 KiB
C++
// shard.h
|
|
|
|
/*
|
|
A "shard" is a database (replica pair typically) which represents
|
|
one partition of the overall database.
|
|
*/
|
|
|
|
/**
|
|
* 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 "../stdafx.h"
|
|
#include "../client/dbclient.h"
|
|
#include "../client/model.h"
|
|
#include "shardkey.h"
|
|
#include <boost/utility.hpp>
|
|
|
|
namespace mongo {
|
|
|
|
class DBConfig;
|
|
class ShardManager;
|
|
class ShardObjUnitTest;
|
|
|
|
/**
|
|
config.shard
|
|
{ ns : "alleyinsider.fs.chunks" , min : {} , max : {} , server : "localhost:30001" }
|
|
*/
|
|
class Shard : public Model , boost::noncopyable {
|
|
public:
|
|
|
|
BSONObj& getMin(){
|
|
return _min;
|
|
}
|
|
BSONObj& getMax(){
|
|
return _max;
|
|
}
|
|
|
|
string getServer(){
|
|
return _server;
|
|
}
|
|
void setServer( string server );
|
|
|
|
bool contains( const BSONObj& obj );
|
|
|
|
string toString() const;
|
|
operator string() const { return toString(); }
|
|
|
|
bool operator==(const Shard& s);
|
|
|
|
bool operator!=(const Shard& s){
|
|
return ! ( *this == s );
|
|
}
|
|
|
|
void getFilter( BSONObjBuilder& b );
|
|
|
|
Shard * split();
|
|
Shard * split( const BSONObj& middle );
|
|
|
|
virtual const char * getNS(){ return "config.shard"; }
|
|
virtual void serialize(BSONObjBuilder& to);
|
|
virtual void unserialize(BSONObj& from);
|
|
virtual string modelServer();
|
|
|
|
protected:
|
|
Shard( ShardManager * info );
|
|
|
|
private:
|
|
ShardManager * _manager;
|
|
|
|
string _ns;
|
|
BSONObj _min;
|
|
BSONObj _max;
|
|
string _server;
|
|
|
|
bool _modified;
|
|
|
|
void _split( BSONObj& middle );
|
|
|
|
friend class ShardManager;
|
|
friend class ShardObjUnitTest;
|
|
};
|
|
|
|
/* config.sharding
|
|
{ ns: 'alleyinsider.fs.chunks' ,
|
|
key: { ts : 1 } ,
|
|
shards: [ { min: 1, max: 100, server: a } , { min: 101, max: 200 , server : b } ]
|
|
}
|
|
*/
|
|
class ShardManager {
|
|
public:
|
|
|
|
ShardManager( DBConfig * config , string ns ,ShardKeyPattern pattern );
|
|
virtual ~ShardManager();
|
|
|
|
string getns(){
|
|
return _ns;
|
|
}
|
|
|
|
int numShards(){ return _shards.size(); }
|
|
bool hasShardKey( const BSONObj& obj );
|
|
|
|
Shard& findShard( const BSONObj& obj );
|
|
ShardKeyPattern& getShardKey(){ return _key; }
|
|
|
|
/**
|
|
* @return number of shards added to the vector
|
|
*/
|
|
int getShardsForQuery( vector<Shard*>& shards , const BSONObj& query );
|
|
|
|
void save();
|
|
|
|
string toString() const;
|
|
operator string() const { return toString(); }
|
|
|
|
private:
|
|
DBConfig * _config;
|
|
string _ns;
|
|
ShardKeyPattern _key;
|
|
|
|
vector<Shard*> _shards;
|
|
|
|
friend class Shard;
|
|
};
|
|
|
|
} // namespace mongo
|