com.taobao.tddl.jdbc.atom.config.DiamondDbConfManager.java Source code

Introduction

Here is the source code for com.taobao.tddl.jdbc.atom.config.DiamondDbConfManager.java

Source

/*(C) 2007-2012 Alibaba Group Holding Limited.   
 *This program is free software; you can redistribute it and/or modify   
*it under the terms of the GNU General Public License version 2 as   
* published by the Free Software Foundation.   
* Authors:   
*   junyu <junyu@taobao.com> , shenxun <shenxun@taobao.com>,   
*   linxuan <linxuan@taobao.com> ,qihao <qihao@taobao.com>    
*/
package com.taobao.tddl.jdbc.atom.config;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executors;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.taobao.tddl.common.TDDLConstant;
import com.taobao.tddl.common.config.ConfigDataHandler;
import com.taobao.tddl.common.config.ConfigDataHandlerFactory;
import com.taobao.tddl.common.config.ConfigDataListener;
import com.taobao.tddl.common.config.impl.DefaultConfigDataHandlerFactory;
import com.taobao.tddl.jdbc.atom.common.TAtomConstants;

/**
 * Diamond
 * 
 * @author qihao
 * 
 */
public class DiamondDbConfManager implements DbConfManager {
    private static Log logger = LogFactory.getLog(DiamondDbConfManager.class);
    private String globalConfigDataId;
    private String appConfigDataId;
    private ConfigDataHandlerFactory configFactory;
    private ConfigDataHandler globalHandler;
    private ConfigDataHandler appDBHandler;
    private volatile List<ConfigDataListener> globalDbConfListener = new ArrayList<ConfigDataListener>();
    private volatile List<ConfigDataListener> appDbConfListener = new ArrayList<ConfigDataListener>();

    public void init() {
        configFactory = new DefaultConfigDataHandlerFactory();
        Map<String, String> config = new HashMap<String, String>();
        config.put("group", TAtomConstants.DEFAULT_DIAMOND_GROUP);
        globalHandler = configFactory.getConfigDataHandlerWithListenerListCE(globalConfigDataId,
                globalDbConfListener, Executors.newSingleThreadScheduledExecutor(), config);
        appDBHandler = configFactory.getConfigDataHandlerWithListenerListCE(appConfigDataId, appDbConfListener,
                Executors.newSingleThreadScheduledExecutor(), config);
    }

    public String getAppDbConfDataId() {
        return appConfigDataId;
    }

    public String getAppDbDbConf() {
        if (null != appDBHandler) {
            return appDBHandler.getData(TDDLConstant.DIAMOND_GET_DATA_TIMEOUT,
                    ConfigDataHandler.FIRST_CACHE_THEN_SERVER_STRATEGY);
        }
        logger.error("[getDataError] appDBConfig not init !");
        return null;
    }

    public String getGlobalDbConf() {
        if (null != globalHandler) {
            return globalHandler.getData(TDDLConstant.DIAMOND_GET_DATA_TIMEOUT,
                    ConfigDataHandler.FIRST_CACHE_THEN_SERVER_STRATEGY);
        }
        logger.error("[getDataError] globalConfig not init !");
        return null;
    }

    public void setGlobalConfigDataId(String globalConfigDataId) {
        this.globalConfigDataId = globalConfigDataId;
    }

    public String getAppConfigDataId() {
        return appConfigDataId;
    }

    public void setAppConfigDataId(String appConfigDataId) {
        this.appConfigDataId = appConfigDataId;
    }

    /**
     * @param Listener
     */
    public void registerGlobaDbConfListener(ConfigDataListener listener) {
        globalDbConfListener.add(listener);
    }

    /**
     * @param Listener
     */
    public void registerAppDbConfListener(ConfigDataListener listener) {
        appDbConfListener.add(listener);
    }

    public void stopDbConfManager() {
        if (null != this.globalHandler) {
            this.globalHandler.closeUnderManager();
        }
        if (null != this.appDBHandler) {
            this.appDBHandler.closeUnderManager();
        }
    }
}