多数据源分布式事务

1 操作多数据源

创建多个数据源

1.先配置说有数据库

2.并将数据库添加DynamicRoutingDataSource这个动态数据库中

3.并设置了目标数据源和配置的数据源

4.配置的数据源以map形式存放，

5.然后将DynamicRoutingDataSource数据源传递给了mybatis，

mybatis操作数据库就是通过他来操作

@Configuration
@MapperScan("com.dw.seata.multiple.mapper")
public class DataSourceProxyConfig {
?
 ? ?@Bean("originOrder")
 ? ?@ConfigurationProperties(prefix = "spring.datasource.order")
 ? ?public DataSource dataSourceMaster() {
 ? ? ? ?return new DruidDataSource();
 ?  }
?
 ? ?@Bean("originStorage")
 ? ?@ConfigurationProperties(prefix = "spring.datasource.storage")
 ? ?public DataSource dataSourceStorage() {
 ? ? ? ?return new DruidDataSource();
 ?  }
?
 ? ?@Bean("originAccount")
 ? ?@ConfigurationProperties(prefix = "spring.datasource.account")
 ? ?public DataSource dataSourceAccount() {
 ? ? ? ?return new DruidDataSource();
 ?  }
?
?
 ? ?@Bean("dynamicDataSource")
 ? ?public DataSource dynamicDataSource(@Qualifier("originOrder") DataSource dataSourceOrder,
 ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?@Qualifier("originStorage") DataSource dataSourceStorage,
 ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?@Qualifier("originAccount") DataSource dataSourceAccount) {
?
 ? ? ? ?DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
?
 ? ? ? ?// 数据源的集合
 ? ? ? ?Map<Object, Object> dataSourceMap = new HashMap<>(3);
 ? ? ? ?dataSourceMap.put(DataSourceKey.ORDER.name(), dataSourceOrder);
 ? ? ? ?dataSourceMap.put(DataSourceKey.STORAGE.name(), dataSourceStorage);
 ? ? ? ?dataSourceMap.put(DataSourceKey.ACCOUNT.name(), dataSourceAccount);
?
 ? ? ? ?// 设置默认的数据源
 ? ? ? ?dynamicRoutingDataSource.setDefaultTargetDataSource(dataSourceOrder);
 ? ? ? ?// 设置目标数据源
 ? ? ? ?dynamicRoutingDataSource.setTargetDataSources(dataSourceMap);
?
 ? ? ? ?DynamicDataSourceContextHolder.getDataSourceKeys().addAll(dataSourceMap.keySet());
?
 ? ? ? ?return dynamicRoutingDataSource;
 ?  }
?
 ? ?@Bean
 ? ?@ConfigurationProperties(prefix = "mybatis")
 ? ?public SqlSessionFactoryBean sqlSessionFactoryBean(@Qualifier("dynamicDataSource") DataSource dataSource) {
 ? ? ? ?SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
 ? ? ? ?sqlSessionFactoryBean.setDataSource(dataSource);
 ?
 ? ? ? ?org.apache.ibatis.session.Configuration configuration=new org.apache.ibatis.session.Configuration();
 ? ? ? ?//使用jdbc的getGeneratedKeys获取数据库自增主键值
 ? ? ? ?configuration.setUseGeneratedKeys(true);
 ? ? ? ?//使用列别名替换列名
 ? ? ? ?configuration.setUseColumnLabel(true);
 ? ? ? ?//自动使用驼峰命名属性映射字段，如userId ---> user_id
 ? ? ? ?configuration.setMapUnderscoreToCamelCase(true);
 ? ? ? ?sqlSessionFactoryBean.setConfiguration(configuration);
 ?
 ? ? ? ?return sqlSessionFactoryBean;
 ?  }
?
}

2 创建DynamicRoutingDataSource

DynamicRoutingDataSource继承了AbstractRoutingDataSource，并且重写了determineCurrentLookupKey方法，

这个方法，是在getConnection时候我们会调用这个方法活去对应的key，然后再对应我们的map数据库集合中获取对应的数据源。后面我们在源码讲解

@Slf4j
public class DynamicRoutingDataSource extends AbstractRoutingDataSource {

    /**
     * 该方法的返回值就是项目中所要用的DataSource的key值，
     * 拿到该key后就可以在resolvedDataSource中取出对应的DataSource，
     * 如果key找不到对应的DataSource就使用默认的数据源。
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        log.info("当前数据源 [{}]", DynamicDataSourceContextHolder.getDataSourceKey());
        return DynamicDataSourceContextHolder.getDataSourceKey();

    }

    public static void main(String[] args) {
        System.out.println(DynamicDataSourceContextHolder.getDataSourceKey());
    }
}

3 创建切面

我们对应的service 上都是有@Datasource注解，这个注解DynamicDataSourceAspect会在方法调用的时候进行拦截，在方法之间会解析@Datasource中的值，会将其设置到ThreadLocal里面，这里面DynamicDataSourceContextHolder，然后调用mybatis时候获取链接的时候时候会带调用上面的类determineCurrentLookupKey获取对应的key，我们通过这个key可以获取对应的数据源

@Aspect
@Component
public class DynamicDataSourceAspect {
    private static final Logger logger = LoggerFactory.getLogger(DynamicDataSourceAspect.class);

    @Before("@annotation(ds)")
    public void changeDataSource(JoinPoint point, DataSource ds) throws Throwable {
        DataSourceKey dsId = ds.value();
        if (DynamicDataSourceContextHolder.isContainsDataSource(dsId.name())) {
            DynamicDataSourceContextHolder.setDataSourceKey(dsId);
            logger.debug("Use DataSource :{} >", dsId, point.getSignature());
        } else {
            logger.error("数据源[{}]不存在，使用默认数据源 >{}", dsId, point.getSignature());
        }
    }

    @After("@annotation(ds)")
    public void restoreDataSource(JoinPoint point, DataSource ds) {
        logger.debug("Revert DataSource : " + ds.value() + " > " + point.getSignature());
        DynamicDataSourceContextHolder.clearDataSourceKey();

    }
}

首先我们给DynamicRoutingDataSource设置数据源setTargetDataSources 是一个map数据源，设置完后afterPropertiesSet方法里面会将我们设置的map设置到我们的resovledDataSources中

当我调用getConnnection时候会调用DetermineTargetDataSources方法，这个方法会调用我们重写的determineCourrentLookupKey，从而获取数据源对应的ekey

最后 通过key从resolveDataSources中获取了对应的数据源。