• Home
• io.github.sichengtech
• rsf-core
• 1.3.7
• source code
• ConfigLoader.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.hc360.rsf.config.ConfigLoader Maven / Gradle / Ivy

/**
 * Copyright(c) 2000-2012 HC360.COM, All Rights Reserved.
 */
package com.hc360.rsf.config;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.DOMException;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import com.hc360.rsf.common.Constants;
import com.hc360.rsf.common.utils.PortUtils;
import com.hc360.rsf.common.utils.ReflectUtils;
import com.hc360.rsf.config.callback.CallBack;
import com.hc360.rsf.registry.RegistryFactory;

/**
 * 配置加载器，核心类，万分重要，是一切的开始，这里就是入口。
 * 
 * 用于读取RSF框架的配置文件,并解析。可以完成服务端在本地暴露服务,向注册中心注册服务
 * 客户端向注册中下载服务 等工作
 * 
 * @author zhaolei 2012-6-6
 */
public class ConfigLoader {
	private static Logger logger = LoggerFactory.getLogger(ConfigLoader.class);
	static final String SPRING_DEFAULT_NAMESPACE = "http://www.springframework.org/schema/beans";
	static final String RSF_DEFAULT_NAMESPACE = "http://code.hc360.com/schema/rsf";
	private final Map BEAN_MAP = new ConcurrentHashMap();
	/**
	 * 用于生成临时ID
	 */
	private static final AtomicLong TEMP_ID = new AtomicLong(0);
	
	static final String FILE_PREFIX="file:";
	static final String CLASS_PREFIX="classpath:";
	static final String WEB_PREFIX="/WEB-INF/";
	
	//启动标记
	private boolean isStart=false;
	
    /**
     * 释放RSF所有打开的资源
     * 
     */
	static public void destroy(){
		AbstractConfig.destroy();
	}
	static public void destroy(String msg){
		AbstractConfig.destroy(msg);
	}

	/**
	 * 构造方法,解析指定的配置文件
	 * 
	 * @param filePath 文件路径 
	 * file:D:\\config\\rsf.xml
	 * classpath:com/hc360/rsf/config/xml/rsf.xml
	 */
	public ConfigLoader(String filePath){
		logger.info("RSF加载配置文件："+filePath);
		load(filePath);
	}
	
	public ConfigLoader(String[] configFilePaths){
		if(configFilePaths!=null){
			for(String filePath:configFilePaths){
				logger.info("RSF加载配置文件："+filePath);
				load(filePath);
			}
		}else{
			throw new IllegalArgumentException("RSF配置文件路径为null");
		}
	}
	
	/**
	 * 把多种来源的配置文件转换为流 
	 * @param filePath
	 */
	private void load(String filePath){
		String xmlPath = null;
		InputStream input =null;
		if(filePath==null){
			throw new IllegalArgumentException("RSF配置文件路径为null");
		}
		xmlPath=filePath;
		if(filePath.toLowerCase().startsWith(FILE_PREFIX)){
			String path=filePath.substring(FILE_PREFIX.length(), filePath.length());
			File file =new File(path);
			try {
				input=new FileInputStream(file);
			} catch (FileNotFoundException e) {
				throw new IllegalArgumentException("RSF配置文件不存在,路径:"+filePath);
			}
			
		}else if(filePath.toLowerCase().startsWith(CLASS_PREFIX)){
			String path=filePath.substring(CLASS_PREFIX.length(), filePath.length());
			input = Thread.currentThread().getContextClassLoader().getResourceAsStream(path);
		}else{
			throw new IllegalArgumentException("RSF配置文件路径需要加前缀,"+FILE_PREFIX+"或"+CLASS_PREFIX+",路径："+filePath);
		}
		loadXML(input,xmlPath);
	}
	/**
	 * 构造方法,通过流取得配置文件并解析
	 * 
	 * @param in 输入流
	 */
	public ConfigLoader(InputStream input){
		logger.info("RSF加载配置文件："+"来自流");
		loadXML(input,"来自流");
	}
	/**
	 * 启动
	 * 
	 * 这是服务端、客户端启动的 重要方法 
	 * 
	 * 服务端启动，向注册中心注册服务
	 * 客户端启动，向注册中心下载服务
	 * 
	 */
	public synchronized void start() {
		if(isStart){
			logger.error("严重，不可以重复调用ConfigLoader的start()方法。");
			return;
		}
		
		//协议
		ProtocolConfig protocolConfig=null;
		for(String key:BEAN_MAP.keySet()){
			Object obj=BEAN_MAP.get(key);
			if(obj instanceof ProtocolConfig){
				protocolConfig=(ProtocolConfig)obj;
				break;//只支持配置一个协议
			}
		}
		
		for(String key:BEAN_MAP.keySet()){
			Object obj=BEAN_MAP.get(key);
			//服务提供者(服务端)
			if(obj instanceof ServiceConfig){
				ServiceConfig serviceConfig=(ServiceConfig)obj;
				//装配协议
				if(protocolConfig!=null){
					serviceConfig.setProtocol(protocolConfig);
				}
				//暴露服务
				serviceConfig.export();
				//装配注册中心
				String registriesStr=serviceConfig.getRegistriesStr();
				if(registriesStr!=null){
					String[] regs=registriesStr.split(",");
					List list=new ArrayList();
					for(String rs:regs){
						Object obj2=getBean(rs);
						if(obj2 instanceof RegistryConfig){
							RegistryConfig rc=(RegistryConfig)obj2;
							list.add(rc);
						}else{
							throw new RsfConfigFileException("id="+rs+"的RegistryConfig类型错误");
						}
					}
					//关联指定的注册中心
					serviceConfig.setRegistries(list);
				}else{
					//缺省关联所有注册中
					serviceConfig.setRegistries(getRegistryConfigList());
				}
			}
			//服务消费者（客户端）
			if(obj instanceof ClientConfig){
				ClientConfig clientConfig=(ClientConfig)obj;
				//装配注册中心
				String registriesStr=clientConfig.getRegistriesStr();
				if(registriesStr!=null){
					String[] regs=registriesStr.split(",");
					List list=new ArrayList();
					for(String rs:regs){
						Object obj2=getBean(rs);
						if(obj2 instanceof RegistryConfig){
							RegistryConfig rc=(RegistryConfig)obj2;
							list.add(rc);
						}else{
							throw new RsfConfigFileException("id="+rs+"的RegistryConfig类型错误");
						}
					}
					//关联指定的注册中心
					clientConfig.setRegistries(list);
				}else{
					//缺省关联所有注册中心
					clientConfig.setRegistries(getRegistryConfigList());
				}
			}
		}
		
		//---------------启动的工作 最后一步----------------------
		
		//服务端 要向注册中心 注册的服务,仅对XML配置有效
		List>  list_ServiceConfig=getServiceConfigList();
		GlobalManager.TIMER.appendServiceConfigList(list_ServiceConfig);//准备定时注册
		
		//客户端要定时下载的服务,仅对XML配置有效
		List>  list_ClientConfig=getClientConfigList();
		for(ClientConfig bean:list_ClientConfig){
			GlobalManager.TIMER.appendClientConfig(bean);//准备定时下载
		}
		
		//批量注册工具， 向多个注册中心，注册多个服务
		RegistryFactory.regBatch(list_ServiceConfig);
		
		//持有注册中心的引用
		GlobalManager.registryConfigList=getRegistryConfigList();
		
		//启动标记
		isStart=true;
	}
	
	/**
	 * 取得ProtocolConfig对象
	 * 目前只在服务端使用，全局只有一个对象
	 * 
	 * @return
	 */
	public ProtocolConfig getProtocolConfig(){
		for(String key:BEAN_MAP.keySet()){
			Object bean=BEAN_MAP.get(key);
			if(bean instanceof ProtocolConfig){
				ProtocolConfig protocolConfig=(ProtocolConfig)bean;
				return protocolConfig;
			}
		}
  		return null;
	}
	
	/**
	 * 取得全部ServiceConfig对象
	 * 一个ServiceConfig代表一个服务接口
	 * @return
	 */
	public List>  getServiceConfigList(){
		List> list_ServiceConfig=new ArrayList>();
  		for(String key:BEAN_MAP.keySet()){
			Object bean=BEAN_MAP.get(key);
			if(bean instanceof ServiceConfig){
				ServiceConfig serviceConfig=(ServiceConfig)bean;
				list_ServiceConfig.add(serviceConfig);
			}
		}
  		return list_ServiceConfig;
	}
	
	/**
	 * 取得全部ClientConfig对象
	 * 一个ClientConfig代表一个客户端接口
	 * @return
	 */
	public List>  getClientConfigList(){
		List> list_ClientConfig=new ArrayList>();
  		for(String key:BEAN_MAP.keySet()){
			Object bean=BEAN_MAP.get(key);
			if(bean instanceof ClientConfig){
				ClientConfig clientConfig=(ClientConfig)bean;
				list_ClientConfig.add(clientConfig);
			}
		}
  		return list_ClientConfig;
	}

	/**
	 * 取得bean,包括ClientConfig,ServiceCinfig,RegistryConfig等类型的bean.
	 * 本方法并不常用，业务人员常用的方法是getServiceProxyBean()
	 * 
	 * @param BeanName
	 * @return
	 */
	public Object getBean(String BeanName){
		return BEAN_MAP.get(BeanName);
	}
	/**
	 * 返回bean的总数量 ,包括ClientConfig,ServiceCinfig,RegistryConfig等类型的bean.
	 * @return
	 */
	public int getBeanCount(){
		return BEAN_MAP.size();
	}
	/**
	 * 按ID取得"远程服务接口"的"本地代理"
	 * 
	 * @param beanId
	 * @return
	 */
	public Object getServiceProxyBean(String beanId){
		Object obj=BEAN_MAP.get(beanId);
		if(obj!=null && obj instanceof ClientConfig){
			ClientConfig clientConfig=(ClientConfig)obj;
			return clientConfig.getInstance();
		}else{
			return null;
		}
	}
	
	/**
	 * 按类型取得"远程服务接口"的"本地代理"
	 * 
	 * @param clazz
	 * @return Object[]
	 */
	public Object[] getServiceProxyBean(Class clazz){
		List