com.liferay.mongodb.util.MongoDBUtil.java Source code

Introduction

Here is the source code for com.liferay.mongodb.util.MongoDBUtil.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library 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 library 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.
 */

package com.liferay.mongodb.util;

import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.util.GetterUtil;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.util.Validator;
import com.liferay.portal.util.PortalUtil;
import com.liferay.portlet.expando.model.ExpandoTable;

import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.Mongo;
import com.mongodb.MongoOptions;
import com.mongodb.ServerAddress;

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

/**
 * @author Raymond Aug
 * @author Brian Wing Shun Chan
 */
public class MongoDBUtil {

    public static boolean authenticate(long companyId) {
        return _instance._authenticate(companyId);
    }

    public static DBCollection getCollection(ExpandoTable expandoTable) {
        return _instance._getCollection(expandoTable);
    }

    public static DBCollection getCollection(long companyId, long classNameId, String tableName) {

        return _instance._getCollection(companyId, classNameId, tableName);
    }

    public static DBCollection getCollection(long companyId, String className, String tableName) {

        return _instance._getCollection(companyId, className, tableName);
    }

    public static String getCollectionName(String className, String tableName) {
        return _instance._getCollectionName(className, tableName);
    }

    public static DB getDB(long companyId) {
        return _instance._getDB(companyId);
    }

    public static Mongo getMongo() {
        return _instance._mongo;
    }

    private MongoDBUtil() {
        try {
            _mongo = new Mongo(_getServerAddresses(), _getMongoOptions());
        } catch (Exception e) {
            _log.error(e, e);
        }
    }

    private boolean _authenticate(long companyId) {
        if (Validator.isNull(PortletPropsValues.SERVER_USERNAME)
                || Validator.isNull(PortletPropsValues.SERVER_PASSWORD)) {

            return true;
        }

        DB db = _getDB(companyId);

        return db.authenticate(PortletPropsValues.SERVER_USERNAME,
                PortletPropsValues.SERVER_PASSWORD.toCharArray());
    }

    private DBCollection _getCollection(ExpandoTable expandoTable) {
        return _instance._getCollection(expandoTable.getCompanyId(), expandoTable.getClassName(),
                expandoTable.getName());
    }

    private DBCollection _getCollection(long companyId, long classNameId, String tableName) {

        String className = PortalUtil.getClassName(classNameId);

        return _getCollection(companyId, className, tableName);
    }

    private DBCollection _getCollection(long companyId, String className, String tableName) {

        DB db = _getDB(companyId);

        String collectionName = _getCollectionName(className, tableName);

        return db.getCollection(collectionName);
    }

    private String _getCollectionName(String className, String tableName) {
        return className.concat(StringPool.POUND).concat(tableName);
    }

    private DB _getDB(long companyId) {
        String dbName = PortletPropsValues.SERVER_DATABASE.concat(StringPool.UNDERLINE)
                .concat(String.valueOf(companyId));

        return _mongo.getDB(dbName);
    }

    private MongoOptions _getMongoOptions() {
        MongoOptions mongoOptions = new MongoOptions();

        mongoOptions.autoConnectRetry = GetterUtil.getBoolean(PortletPropsValues.DRIVER_AUTOCONNECT_RETRY,
                mongoOptions.autoConnectRetry);
        mongoOptions.connectTimeout = GetterUtil.getInteger(PortletPropsValues.DRIVER_CONNECT_TIMEOUT,
                mongoOptions.connectTimeout);
        mongoOptions.connectionsPerHost = GetterUtil.getInteger(PortletPropsValues.DRIVER_CONNECTIONS_PER_HOST,
                mongoOptions.connectionsPerHost);
        mongoOptions.maxWaitTime = GetterUtil.getInteger(PortletPropsValues.DRIVER_MAX_WAIT_TIME,
                mongoOptions.maxWaitTime);
        mongoOptions.socketTimeout = GetterUtil.getInteger(PortletPropsValues.DRIVER_SOCKET_TIMEOUT,
                mongoOptions.socketTimeout);
        mongoOptions.threadsAllowedToBlockForConnectionMultiplier = GetterUtil.getInteger(
                PortletPropsValues.DRIVER_THREADS_ALLOWED_TO_BLOCK,
                mongoOptions.threadsAllowedToBlockForConnectionMultiplier);

        return mongoOptions;
    }

    private List<ServerAddress> _getServerAddresses() throws Exception {
        List<ServerAddress> serverAddresses = new ArrayList<ServerAddress>();

        for (String hostname : PortletPropsValues.SERVER_HOSTNAMES) {
            ServerAddress serverAddress = new ServerAddress(hostname, PortletPropsValues.SERVER_PORT);

            serverAddresses.add(serverAddress);
        }

        return serverAddresses;
    }

    private static Log _log = LogFactoryUtil.getLog(MongoDBUtil.class);

    private static MongoDBUtil _instance = new MongoDBUtil();

    private Mongo _mongo;

}