Zebra源码分析-GroupDataSource

Zebra源码分析-GroupDataSource

1 简介

GroupDataSource是读写分离数据源，由多个SingleDataSource组成，其中有一个主库，用于写操作和一部分指定得读操作；多个从库之间使用负载均衡策略，根据不同的权重来路由请求，同时系统还是地理信息感知的，通过配置地理信息来更高效和灵活的路由请求

2. 配置文件

2.1 数据源配置

在使用读写分离的数据源的时候，需要在resource目录下创建{jdbcRef}.properties来对数据源进行配置，包括数据库基本信息和读写比例等

zebra.group.zebra=<groupConfig>\
  <singleConfig>\
  <name>zebra-n1</name>\
  <writeWeight>0</writeWeight>\
  <readWeight>-1</readWeight>\
  </singleConfig>\
  <singleConfig>\
  <name>zebra-n2</name>\
  <writeWeight>-1</writeWeight>\
  <readWeight>1</readWeight>\
  </singleConfig>\
  </groupConfig>

zebra.ds.zebra-n1=<dsConfig>\
  <url>jdbc:mysql://127.0.0.1:3306/zebra?characterEncoding=UTF8&amp;socketTimeout=60000</url>\
  <driverClass>com.mysql.jdbc.Driver</driverClass>\
  <active>true</active>\
  <username>root</username>\
 <properties>idleConnectionTestPeriod=80&amp;acquireRetryAttempts=50&amp;acquireRetryDelay=300&amp;maxStatements=1</properties>\
  <password>123456</password>\
  </dsConfig>

zebra.ds.zebra-n2=<dsConfig>\
  <url>jdbc:mysql://127.0.0.1:3306/zebra?characterEncoding=UTF8&amp;socketTimeout=60000</url>\
  <driverClass>com.mysql.jdbc.Driver</driverClass>\
  <active>true</active>\
  <username>root</username>\
 <properties>idleConnectionTestPeriod=80&amp;acquireRetryAttempts=50&amp;acquireRetryDelay=300&amp;maxStatements=1</properties>\
  <password>123456</password>\
  </dsConfig>

2.2 区域配置

3. 源码分析

3.1 配置处理

首先对配置服务和配置信息对GroupSource初始化，需要完成的步骤有：

• 创建配置服务，并初始化
• 创建配置服务的管理器，师姐的管理操作由配置服务完成
• 创建系统配置管理器
• 创建并汇总GroupSource需要的配置信息

protected void initConfig() {
   
		this.configService = ConfigServiceFactory.getConfigService(configManagerType, serviceConfigs);
		this.dataSourceConfigManager = DataSourceConfigManagerFactory.getConfigManager(jdbcRef, configService);
		this.dataSourceConfigManager.addListerner(new GroupDataSourceConfigChangedListener());
		this.systemConfigManager = SystemConfigManagerFactory.getConfigManger(configManagerType, configService);
		this.groupConfig = buildGroupConfig();

		if (this.groupConfig != null && this.useCustomRouterConfig) {
   
			this.groupConfig.setRouterStrategy(this.routerStrategy);
		} else if (!this.useCustomRouterConfig) {
   
			this.routerStrategy = this.groupConfig.getRouterStrategy();
		}
	}

3.1.1 创建配置服务

初始化配置服务-ConfigService

在初始化读写分离数据源时，需要对读写库的配置信息进行读取和构造，并使用配置对每一个数据源进行初始化

首先，需要对配置管理器进行加载，在Zebra中支持两种配置类型：

• 一种是local，根据本地的文件读取数据源的信息
• 另一种是zookeeper，zk的信息配置在reousrces/zookeeper.properties中

通过ConfigServiceFactory工厂和传入的配饰类型来构造ConfigService对象，并对配置服务初始化，代码如下

public class ConfigServiceFactory {
   
	public static ConfigService getConfigService(String name, Map<String, Object> configs) {
   
    //通过ExtensionLoader加载对应的ConfigService
		ConfigService configService = ExtensionLoader.getExtensionLoader(ConfigService.class).load(name);
    ...
		configService.init(configs);
    // 初始化
		return configService;
	}
}

在getExtensionLoader()中，两次对内部的loaderMap进行查询，如果没有查找到已经创建的ExtensionLoader则创建一个对应的对象，并放再map结构中，方便下一次获取，通过如下构造函数，完成对象的创建，参数包括ConfigService和类加载器

private ExtensionLoader(Class<T> service) {
   
		this(service, Thread.currentThread().getContextClassLoader());
	}

private ExtensionLoader(Class<T> service, ClassLoader loader) {
   
		this.service = service;
		this.loader = loader;
	}

在加载对应的参数服务的时候，需要先对刚创建的ExtensionLoader进行初始化，主要的工作如查下：

• 使用类加载器从给定的路径产生URL对象
• 通过读取URL指示的文件内容（配置服务的类名），保存到list中
• 通过类名，加载list中所有的类

private Map<String, Class<T>> loadExtensions() {
   
		String configFiles = PREFIX + this.service.getName();
		List<String> classNames = null;

		try {
   
      //URL代表路径"META-INF/services/com.dianping.zebra.config.ConfigService"
			Enumeration<URL> url = null;
			if (this.loader != null) {
   
				url = this.loader.getResources(configFiles);
			} else {
   
				url = ClassLoader.getSystemResources(configFiles);
			}
			while (url.hasMoreElements()) {
   
				URL u = url.nextElement();
        // 对读取的文件进行解析，将类放入List中保存
				classNames = parseClassNames(u);
			}
		} catch (Exception e) {
   
			throw new ZebraException("loadExtensions fail, configFileName:" + configFiles, e);
		}
    //对List中所有的类进行加载
		return this.loadAllClasses(classNames);
	}

在这一步存储在list中的类名使用类加载器进行加载，在加载之前判断类加载器是否已经初始化了，并对类进行检查：

• 是否是public类型
• 是否实现了service接口
• 有无默认的构造函数

private ConcurrentHashMap<String, Class<T>> loadAllClasses(List<String> classNames) {
   
		ConcurrentHashMap<String, Class<T>> classes = new ConcurrentHashMap<>();

		if (classNames != null && !classNames.isEmpty()) {
   
			for (String className : classN