ScanMgrImpl.java

package com.baidu.disconf.client.scan.impl;

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

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.baidu.disconf.client.config.DisClientConfig;
import com.baidu.disconf.client.scan.ScanMgr;
import com.baidu.disconf.client.scan.inner.dynamic.ScanDynamicStoreAdapter;
import com.baidu.disconf.client.scan.inner.statically.StaticScannerMgr;
import com.baidu.disconf.client.scan.inner.statically.StaticScannerMgrFactory;
import com.baidu.disconf.client.scan.inner.statically.impl.StaticScannerNonAnnotationFileMgrImpl;
import com.baidu.disconf.client.scan.inner.statically.model.ScanStaticModel;
import com.baidu.disconf.client.scan.inner.statically.strategy.ScanStaticStrategy;
import com.baidu.disconf.client.scan.inner.statically.strategy.impl.ReflectionScanStatic;
import com.baidu.disconf.client.store.inner.DisconfCenterHostFilesStore;
import com.baidu.disconf.client.support.registry.Registry;

/**
 * 扫描模块
 *
 * @author liaoqiqi
 * @version 2014-6-6
 */
public class ScanMgrImpl implements ScanMgr {

    protected static final Logger LOGGER = LoggerFactory.getLogger(ScanMgrImpl.class);

    // 扫描对象
    private volatile ScanStaticModel scanModel = null;

    //
    private Registry registry = null;

    private List<StaticScannerMgr> staticScannerMgrList = new ArrayList<StaticScannerMgr>();

    private ScanStaticStrategy scanStaticStrategy = new ReflectionScanStatic();

    /**
     *
     */
    public ScanMgrImpl(Registry registry) {

        this.registry = registry;

        // 配置文件
        staticScannerMgrList.add(StaticScannerMgrFactory.getDisconfFileStaticScanner());

        // 配置项
        staticScannerMgrList.add(StaticScannerMgrFactory.getDisconfItemStaticScanner());

        // 非注解 托管的配置文件
        staticScannerMgrList.add(StaticScannerMgrFactory.getDisconfNonAnnotationFileStaticScanner());
    }

    /**
     * 扫描并存储(静态)
     *
     * @throws Exception
     */
    public void firstScan(List<String> packageNameList) throws Exception {

        LOGGER.debug("start to scan package: " + packageNameList.toString());

        // 获取扫描对象并分析整合
        scanModel = scanStaticStrategy.scan(packageNameList);

        // 增加非注解的配置
        scanModel.setJustHostFiles(DisconfCenterHostFilesStore.getInstance().getJustHostFiles());

        // 放进仓库
        for (StaticScannerMgr scannerMgr : staticScannerMgrList) {

            // 扫描进入仓库
            scannerMgr.scanData2Store(scanModel);

            // 忽略哪些KEY
            scannerMgr.exclude(DisClientConfig.getInstance().getIgnoreDisconfKeySet());
        }
    }

    /**
     * 第二次扫描(动态)
     */
    public void secondScan() throws Exception {

        // 开启disconf才需要处理回调
        if (DisClientConfig.getInstance().ENABLE_DISCONF) {

            if (scanModel == null) {
                synchronized(scanModel) {
                    // 下载模块必须先初始化
                    if (scanModel == null) {
                        throw new Exception("You should run first scan before second Scan");
                    }
                }
            }

            // 将回调函数实例化并写入仓库
            ScanDynamicStoreAdapter.scanUpdateCallbacks(scanModel, registry);
        }
    }

    /**
     * reloadable file scan
     *
     * @throws Exception
     */
    @Override
    public void reloadableScan(String fileName) throws Exception {

        StaticScannerNonAnnotationFileMgrImpl.scanData2Store(fileName);
    }

}