org.mongolink.DbFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongolink Show documentation
There is a newer version: 1.3.1
/*
 * MongoLink, Object Document Mapper for Java and MongoDB
 *
 * Copyright (c) 2012, Arpinum or third-party contributors as
 * indicated by the @author tags
 *
 * MongoLink is free software: you can redistribute it and/or modify
 * it under the terms of the Lesser GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MongoLink 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
 * Lesser GNU General Public License for more details.
 *
 * You should have received a copy of the Lesser GNU General Public License
 * along with MongoLink.  If not, see . 
 *
 */

package org.mongolink;

import com.google.common.base.Strings;
import com.mongodb.*;

import java.util.List;

public class DbFactory {

    public DB get(String dbName) {
        initializeMongo();
        DB db = mongoClient.getDB(dbName);
        if (isAuthenticationRequired()) {
            db.authenticate(user, password.toCharArray());
        }
        return db;
    }

    private boolean isAuthenticationRequired() {
        return !Strings.isNullOrEmpty(user);
    }

    private void initializeMongo() {
        if (mongoClient == null) {
            doInitializeMongo();
        }
    }

    private synchronized void doInitializeMongo() {
        if (mongoClient == null) {
            mongoClient = new MongoClient(addresses);
            mongoClient.setReadPreference(readPreference);
            mongoClient.setWriteConcern(writeConcern);
        }
    }

    public void close() {
        if (mongoClient != null) {
            mongoClient.close();
        }
    }

    public List getAddresses() {
        return addresses;
    }

    public void setAddresses(List adresses) {
        this.addresses = adresses;
    }

    public ReadPreference getReadPreference() {
        return readPreference;
    }

    public void setReadPreference(ReadPreference readPreference) {
        this.readPreference = readPreference;
    }

    public void setAuthInfos(String user, String password) {
        this.user = user;
        this.password = password;
    }

    public WriteConcern getWriteConcern() {
        return writeConcern;
    }

    public void setWriteConcern(WriteConcern writeConcern) {
        this.writeConcern = writeConcern;
    }

    private volatile MongoClient mongoClient;
    private volatile String user;
    private volatile String password;
    private List addresses;
    private ReadPreference readPreference;
    private WriteConcern writeConcern;
}