com.ebay.jetstream.configurationmanagement.MongoConfigurationFactory Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 *  Copyright © 2012-2015 eBay Software Foundation
 *  This program is dual licensed under the MIT and Apache 2.0 licenses.
 *  Please see LICENSE for more information.
 *******************************************************************************/
package com.ebay.jetstream.configurationmanagement;

import java.util.ArrayList;
import java.util.List;

import com.ebay.jetstream.config.mongo.MongoConfiguration;

public class MongoConfigurationFactory {
    private MongoConfiguration mongoConfiguration;

    private MongoConfiguration getMongoConfiguration(String mongo_url) {
        String db = null;
        String port = null;
        List hosts = new ArrayList();

        String[] urls = mongo_url.split(";");
        for (String url : urls) {
            int index = url.indexOf("mongo://");
            url = url.substring(index + "mongo://".length());

            String[] hostAndDb = url.split("/");
            String hostAndPort = hostAndDb[0];
            db = hostAndDb[1];
            String[] hostPort = hostAndPort.split(":");
            String host = hostPort[0];
            port = hostPort[1];

            hosts.add(host);
        }

        if (db == null || port == null || hosts.isEmpty()) {
            throw new IllegalArgumentException();
        }

        mongoConfiguration = new MongoConfiguration();
        mongoConfiguration.setDb(db);
        mongoConfiguration.setHosts(hosts);
        mongoConfiguration.setPort(port);
        // mongoConfiguration.setUser(user);
        // mongoConfiguration.setPw(pw);

        return mongoConfiguration;
    }

    private String getPropOrEnv(String key) {
        String it = System.getProperty(key);
        if (it == null) {
            it = System.getenv(key);
        }
        return it;
    }

    public synchronized MongoConfiguration getMongoConfiguration() {
        if (mongoConfiguration == null) {
            try {
                getMongoConfiguration(getPropOrEnv("MONGO_HOME"));
            } catch (Exception ex) {
                throw new RuntimeException(ex);
            }
        }

        return mongoConfiguration;
    }
}