org.exoplatform.addons.persistence.services.mongodb.MongoBootstrap.java Source code

Introduction

Here is the source code for org.exoplatform.addons.persistence.services.mongodb.MongoBootstrap.java

Source

/*
 * Copyright (C) 2012 eXo Platform SAS.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.exoplatform.addons.persistence.services.mongodb;

import com.mongodb.*;
import de.flapdoodle.embed.mongo.MongodExecutable;
import de.flapdoodle.embed.mongo.MongodProcess;
import de.flapdoodle.embed.mongo.MongodStarter;
import de.flapdoodle.embed.mongo.config.MongodConfig;
import de.flapdoodle.embed.mongo.distribution.Version;
import de.flapdoodle.embed.process.runtime.Network;
import org.exoplatform.addons.utils.PropertyManager;

import java.io.IOException;
import java.net.UnknownHostException;
import java.util.logging.Logger;

public class MongoBootstrap {
    private static MongodExecutable mongodExe;
    private static MongodProcess mongod;
    private Mongo m;
    private DB db;
    private static Logger log = Logger.getLogger("MongoBootstrap");

    private Mongo mongo() {
        if (m == null) {
            try {
                if (PropertyManager.PROPERTY_DB_SERVER_TYPE_EMBED
                        .equals(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_TYPE))) {
                    log.warning("WE WILL NOW USE MONGODB IN EMBED MODE...");
                    log.warning("BE AWARE...");
                    log.warning("EMBED MODE SHOULD NEVER BE USED IN PRODUCTION!");
                    setupEmbedMongo();
                }

                MongoOptions options = new MongoOptions();
                options.connectionsPerHost = 200;
                options.connectTimeout = 60000;
                options.threadsAllowedToBlockForConnectionMultiplier = 10;
                options.autoConnectRetry = true;
                String host = PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_HOST);
                int port = Integer.parseInt(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_PORT));
                m = new Mongo(new ServerAddress(host, port), options);
                m.setWriteConcern(WriteConcern.SAFE);
            } catch (UnknownHostException e) {
            } catch (IOException e) {
            }
        }
        return m;
    }

    public void close() {
        try {
            if (mongod != null) {
                mongod.stop();
                mongodExe.stop();
            }
            if (m != null)
                m.close();
        } catch (NullPointerException e) {
        }
    }

    public void initialize() {
        this.close();
        this.m = null;
        this.mongo();
    }

    public void dropDB(String dbName) {
        log.info("---- Dropping DB " + dbName);
        mongo().dropDatabase(dbName);
        log.info("-------- DB " + dbName + " dropped!");

    }

    public DB getDB() {
        return getDB(null);
    }

    public DB getDB(String dbName) {
        if (db == null || dbName != null) {
            if (dbName != null)
                db = mongo().getDB(dbName);
            else
                db = mongo().getDB(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_NAME));
            boolean authenticate = "true"
                    .equals(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_AUTHENTICATION));
            if (authenticate) {
                db.authenticate(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_USER),
                        PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_PASSWORD).toCharArray());
            }
            initCollection("statistics");

        }
        return db;
    }

    private static Mongo setupEmbedMongo() throws IOException {
        MongodStarter runtime = MongodStarter.getDefaultInstance();
        int port = Integer.parseInt(PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_PORT));
        mongodExe = runtime.prepare(new MongodConfig(Version.V2_3_0, port, Network.localhostIsIPv6()));
        mongod = mongodExe.start();
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e) {
            log.info(e.getMessage());
        }
        String host = PropertyManager.getProperty(PropertyManager.PROPERTY_DB_SERVER_HOST);

        return new Mongo(new ServerAddress(host, port));
    }

    public void initCappedCollection(String name, int size) {
        initCollection(name, true, size);
    }

    private void initCollection(String name) {
        initCollection(name, false, 0);
    }

    private void initCollection(String name, boolean isCapped, int size) {
        if (getDB().collectionExists(name))
            return;

        BasicDBObject doc = new BasicDBObject();
        doc.put("capped", isCapped);
        if (isCapped)
            doc.put("size", size);
        getDB().createCollection(name, doc);

    }
}