mongo/dbgrid/gridconfig.cpp

// gridconfig.cpp

/**
*    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/>.
*/

#include "stdafx.h"
#include "../grid/message.h"
#include "../util/unittest.h"
#include "../client/connpool.h"
#include "../db/pdfile.h"
#include "gridconfig.h"
#include "../client/model.h"

namespace mongo {

    /* --- Machine --- */

    map<string, Machine*> Machine::machines;

    /* --- GridConfig --- */

//static boost::mutex loc_mutex;
    Grid grid;

    ClientConfig* GridConfig::getClientConfig(string database) {
        ClientConfig*& cc = databases[database];
        if ( cc == 0 ) {
            cc = new ClientConfig();
            if ( !cc->loadByName(database.c_str()) ) {
                log() << "couldn't find database " << database << " in grid db" << endl;
                // note here that cc->primary == 0.
            }
        }
        return cc;
    }

    /* --- Grid --- */

    Machine* Grid::owner(const char *ns, BSONObj& objOrKey) {
        ClientConfig *cc = gc.getClientConfig( nsToClient(ns) );
        if ( cc == 0 ) {
            throw UserAssertionException(
                string("dbgrid: no config for db for ") + ns);
        }

        if ( !cc->partitioned ) {
            if ( !cc->primary )
                throw UserAssertionException(string("dbgrid: no primary for ")+ns);
            return cc->primary;
        }

        uassert("dbgrid: not implemented 100", false);
        return 0;
    }

} // namespace mongo