DalClientFactory.java

package com.ctrip.platform.dal.dao;

import java.util.concurrent.atomic.AtomicReference;

import com.ctrip.platform.dal.dao.configure.DalConfigLoader;
import com.ctrip.platform.dal.dao.helper.ServiceLoaderHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ctrip.platform.dal.dao.client.DalDirectClient;
import com.ctrip.platform.dal.dao.client.DalLogger;
import com.ctrip.platform.dal.dao.client.DalWatcher;
import com.ctrip.platform.dal.dao.configure.DalConfigure;
import com.ctrip.platform.dal.dao.configure.DalConfigureFactory;
import com.ctrip.platform.dal.dao.status.DalStatusManager;
import com.ctrip.platform.dal.dao.task.DalRequestExecutor;
import com.ctrip.platform.dal.dao.task.DalTaskFactory;

public class DalClientFactory {
    private static Logger logger = LoggerFactory.getLogger(Version.getLoggerName());

    private static AtomicReference<DalConfigure> configureRef = new AtomicReference<DalConfigure>();

    static {
        Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
            public void run() {
                shutdownFactory();
            }
        }));
    }

    /**
     * Initialize for DB All In One client. Load Dal.config from classpath
     * 
     * @throws Exception
     */
    public static void initClientFactory() throws Exception {
        internalInitClientFactory(null);
    }

    /**
     * Initialize for DB All In One client. Load Dal.config from give path
     * 
     * @param path Dal.Config file path
     * @throws Exception
     */
    public static void initClientFactory(String path) throws Exception {
        if (path == null)
            throw new NullPointerException("Path is empty");

        internalInitClientFactory(path);
    }

    private static void internalInitClientFactory(String path) throws Exception {
        if (configureRef.get() != null) {
            logger.warn("Dal Java Client Factory is already initialized.");
            return;
        }

        synchronized (DalClientFactory.class) {
            if (configureRef.get() != null) {
                return;
            }

            DalConfigure config = null;
            if (path == null) {
                DalConfigLoader loader = ServiceLoaderHelper.getInstance(DalConfigLoader.class);
                if (loader == null)
                    config = DalConfigureFactory.load();
                else
                    config = loader.load();
                logger.info("Successfully initialized Dal Java Client Factory");
            } else {
                config = DalConfigureFactory.load(path);
                logger.info("Successfully initialized Dal Java Client Factory with " + path);
            }

            DalWatcher.init();
            DalRequestExecutor.init(config.getFacory().getProperty(DalRequestExecutor.MAX_POOL_SIZE));
            DalStatusManager.initialize(config);

            configureRef.set(config);
        }
    }

    /**
     * Actively initialize connection pools for all the logic db in the Dal.config
     */
    public static void warmUpConnections() {
        getDalConfigure().warmUpConnections();
    }

    public static DalClient getClient(String logicDbName) {
        if (logicDbName == null)
            throw new NullPointerException("Database Set name can not be null");

        DalConfigure config = getDalConfigure();

        // Verify if it is existed
        config.getDatabaseSet(logicDbName);

        return new DalDirectClient(config, logicDbName);
    }

    public static DalConfigure getDalConfigure() {
        DalConfigure config = configureRef.get();
        if (config != null)
            return config;

        try {
            initClientFactory();
        } catch (Exception e) {
            throw new IllegalStateException("DalClientFactory initilization fail", e);
        }

        config = configureRef.get();
        if (config != null)
            return config;

        throw new IllegalStateException("DalClientFactory has not been not initialized or initilization fail");
    }

    public static DalLogger getDalLogger() {
        return getDalConfigure().getDalLogger();
    }

    public static DalTaskFactory getTaskFactory() {
        return getDalConfigure().getFacory();
    }

    /**
     * Release All resource the Dal client used.
     */
    public static void shutdownFactory() {
        if (configureRef.get() == null) {
            logger.warn("Dal Java Client Factory is already shutdown.");
            return;
        }

        synchronized (DalClientFactory.class) {
            if (configureRef.get() == null) {
                return;
            }

            try {
                logger.info("Start shutdown Dal Java Client Factory");
                getDalLogger().shutdown();
                logger.info("Dal Logger is shutdown");

                DalRequestExecutor.shutdown();
                logger.info("Dal Java Client Factory is shutdown");

                DalStatusManager.shutdown();

                DalWatcher.destroy();
                logger.info("DalWatcher has been destoryed");
            } catch (Throwable e) {
                logger.error("Error during shutdown", e);
            }

            configureRef.set(null);
        }
    }

}