Sharding-JDBC ShardingRule(三)

@Bean
@Conditional(ShardingRuleCondition.class)
public DataSource shardingDataSource() throws SQLException {
	// 4. 通过:ShardingDataSourceFactory的工厂方法创建:DataSource
	return ShardingDataSourceFactory
	       .createDataSource(
		        dataSourceMap, 
				// 3. 把SpringBootShardingRuleConfigurationProperties配置,转换成规则配置模型:ShardingRuleConfiguration.  
				new ShardingRuleConfigurationYamlSwapper().swap(shardingRule), 
				props.getProps()
			); // end createDataSource
}

public ShardingRuleConfiguration swap(final YamlShardingRuleConfiguration yamlConfiguration) {
	ShardingRuleConfiguration result = new ShardingRuleConfiguration();
	
	for (Entry<String, YamlTableRuleConfiguration> entry : yamlConfiguration.getTables().entrySet()) {
		YamlTableRuleConfiguration tableRuleConfig = entry.getValue();
		tableRuleConfig.setLogicTable(entry.getKey());
		result.getTableRuleConfigs().add(tableRuleConfigurationYamlSwapper.swap(tableRuleConfig));
	}
	
	result.setDefaultDataSourceName(yamlConfiguration.getDefaultDataSourceName());
	result.getBindingTableGroups().addAll(yamlConfiguration.getBindingTables());
	result.getBroadcastTables().addAll(yamlConfiguration.getBroadcastTables());
	if (null != yamlConfiguration.getDefaultDatabaseStrategy()) {
		result.setDefaultDatabaseShardingStrategyConfig(shardingStrategyConfigurationYamlSwapper.swap(yamlConfiguration.getDefaultDatabaseStrategy()));
	}
	if (null != yamlConfiguration.getDefaultTableStrategy()) {
		result.setDefaultTableShardingStrategyConfig(shardingStrategyConfigurationYamlSwapper.swap(yamlConfiguration.getDefaultTableStrategy()));
	}
	if (null != yamlConfiguration.getDefaultKeyGenerator()) {
		result.setDefaultKeyGeneratorConfig(keyGeneratorConfigurationYamlSwapper.swap(yamlConfiguration.getDefaultKeyGenerator()));
	}
	Collection<MasterSlaveRuleConfiguration> masterSlaveRuleConfigs = new LinkedList<>();
	for (Entry<String, YamlMasterSlaveRuleConfiguration> entry : yamlConfiguration.getMasterSlaveRules().entrySet()) {
		YamlMasterSlaveRuleConfiguration each = entry.getValue();
		each.setName(entry.getKey());
		masterSlaveRuleConfigs.add(masterSlaveRuleConfigurationYamlSwapper.swap(entry.getValue()));
	}
	result.setMasterSlaveRuleConfigs(masterSlaveRuleConfigs);
	if (null != yamlConfiguration.getEncryptRule()) {
		result.setEncryptRuleConfig(encryptRuleConfigurationYamlSwapper.swap(yamlConfiguration.getEncryptRule()));
	}
	return result;
}

public static DataSource createDataSource(
		// 所有的真实数据源
		final Map<String, DataSource> dataSourceMap, 
		// 在第3步中,把:SpringBootShardingRuleConfigurationProperties转换成:ShardingRuleConfiguration.   
		final ShardingRuleConfiguration shardingRuleConfig, 
		// 其它的属性配置信息.
		final Properties props) throws SQLException {
		
	return new ShardingDataSource(  
	     dataSourceMap, 
		 // 5. 将:ShardingRuleConfiguration再次进行解析,转换成:ShardingRule
		 new ShardingRule(shardingRuleConfig, dataSourceMap.keySet()), 
		 props);
}

public class ShardingRule implements BaseRule {
    // 规则配置
    private final ShardingRuleConfiguration ruleConfiguration;
    // 分片的数据源名称
    private final ShardingDataSourceNames shardingDataSourceNames;
    
	// 表规则
    private final Collection<TableRule> tableRules;
    
	// 绑定表
    private final Collection<BindingTableRule> bindingTableRules;
    
	// 公共表
    private final Collection<String> broadcastTables;
    
	// 默认的数据源分片策略
    private final ShardingStrategy defaultDatabaseShardingStrategy;
    
	// 默认的表分片策略
    private final ShardingStrategy defaultTableShardingStrategy;
    
	// 默认的主键生成
    private final ShardingKeyGenerator defaultShardingKeyGenerator;
    
	// 主从(读写分离)规则
    private final Collection<MasterSlaveRule> masterSlaveRules;
    
	// 加密解密规则
    private final EncryptRule encryptRule;
    
    public ShardingRule(final ShardingRuleConfiguration shardingRuleConfig, final Collection<String> dataSourceNames) {
        Preconditions.checkArgument(null != shardingRuleConfig, "ShardingRuleConfig cannot be null.");
        Preconditions.checkArgument(null != dataSourceNames && !dataSourceNames.isEmpty(), "Data sources cannot be empty.");
		
        this.ruleConfiguration = shardingRuleConfig;
        shardingDataSourceNames = new ShardingDataSourceNames(shardingRuleConfig, dataSourceNames);
		
		// 1. 把 Collection<TableRuleConfiguration> 转换成 Collection<TableRule> 
		//    TableRule是含有:分库/分表的规则的
        tableRules = createTableRules(shardingRuleConfig);
		// 2. 公共表
        broadcastTables = shardingRuleConfig.getBroadcastTables();
		// 绑定表
        bindingTableRules = createBindingTableRules(shardingRuleConfig.getBindingTableGroups());
		
		// 把 ShardingStrategyConfiguration(defaultDatabaseShardingStrategyConfig) 转换成  ShardingStrategy
        defaultDatabaseShardingStrategy = createDefaultShardingStrategy(shardingRuleConfig.getDefaultDatabaseShardingStrategyConfig());
		
		// 把 ShardingStrategyConfiguration(defaultTableShardingStrategyConfig) 转换成  ShardingStrategy
        defaultTableShardingStrategy = createDefaultShardingStrategy(shardingRuleConfig.getDefaultTableShardingStrategyConfig());
		
		// 把 KeyGeneratorConfiguration 转换成 ShardingKeyGenerator
        defaultShardingKeyGenerator = createDefaultKeyGenerator(shardingRuleConfig.getDefaultKeyGeneratorConfig());
		// 把 Collection<MasterSlaveRuleConfiguration>  转换成 Collection<MasterSlaveRule>
        masterSlaveRules = createMasterSlaveRules(shardingRuleConfig.getMasterSlaveRuleConfigs());
		
		把 EncryptRuleConfiguration 转换成  EncryptRule
        encryptRule = createEncryptRule(shardingRuleConfig.getEncryptRuleConfig());
    } // end 构造器
}