// 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 ShardInfo;
    
    class Shard : boost::noncopyable {
    public:
        
        BSONObj& getMin(){
            return _min;
        }
        BSONObj& getMax(){
            return _max;
        }

        string getServer(){
            return _server;
        }
        void setServer( string server );

        BSONObj getData() const;

        bool contains( const BSONObj& obj );

        void split();
        void split( const BSONObj& middle );
        
        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 );

    private:
        Shard( ShardInfo * info , BSONObj data );
        
        ShardInfo * _info;
        BSONObj _data;
        
        BSONObj _min;
        BSONObj _max;
        string _server;
        
        bool _modified;

        void _split( BSONObj& middle );

        friend class ShardInfo;
    };

    /* config.sharding
         { ns: 'alleyinsider.fs.chunks' , 
           key: { ts : 1 } ,
           shards: [ { min: 1, max: 100, server: a } , { min: 101, max: 200 , server : b } ]
         }
    */
    class ShardInfo : public Model {
    public:

        ShardInfo( DBConfig * config );
        virtual ~ShardInfo();

        string getns(){
            return _ns;
        }
        
        int numShards(){ return _shards.size(); }
        bool hasShardKey( const BSONObj& obj );

        Shard& findShard( const BSONObj& obj );
        ShardKey& getShardKey(){  return _key; }
        
        /**
         * @return number of shards added to the vector
         */
        int getShardsForQuery( vector<Shard*>& shards , const BSONObj& query );

        virtual const char * getNS(){ return "config.sharding"; }
        virtual void serialize(BSONObjBuilder& to);
        virtual void unserialize(BSONObj& from);
        virtual string modelServer();
        bool loadByName( const string& ns );

        string toString() const;
        operator string() const { return toString(); }
        
    private:
        DBConfig * _config;
        string _ns;
        ShardKey _key;
        vector<Shard*> _shards;
        
        friend class Shard;
    };

} // namespace mongo