详解基于spring多数据源动态调用及其事务处理

需求：

成都创新互联公司于2013年创立，先为贵池等服务建站，贵池等地企业，进行企业商务咨询服务。为贵池企业网站制作PC+手机+微官网三网同步一站式服务解决您的所有建站问题。

有些时候，我们需要连接多个数据库，但是，在方法调用前并不知道到底是调用哪个。即同时保持多个数据库的连接，在方法中根据传入的参数来确定。

下图的单数据源的调用和多数据源动态调用的流程，可以看出在Dao层中需要有一个DataSource选择器，来确定到底是调用哪个数据源。

实现方式

对Dao层提供一个公共父类，保持有多个数据源的连接（本人是基于iBatis，即保持多个SQLSessionTemplate)

/**

 * Created by hzlizhou on 2017/2/6.

 */

public abstract class MultiDatasourceDao implements IDaoSupport {

 

  private Map sqlSessionTemplateMap;

 

  private MultiDataSourceSelector multiDataSourceSelector;

 

  public MultiDatasourceDao(Map sqlSessionTemplateMap, MultiDataSourceSelector multiDataSourceSelector) {

    this.sqlSessionTemplateMap = sqlSessionTemplateMap;

    this.multiDataSourceSelector = multiDataSourceSelector;

  }

 

  public Map getSqlSessionTemplateMap() {

    return sqlSessionTemplateMap;

  }

 

  public void setSqlSessionTemplateMap(Map sqlSessionTemplateMap) {

    this.sqlSessionTemplateMap = sqlSessionTemplateMap;

  }

 

  //子类通过这个方法动态获取SqlSessionTemplate

  protected SqlSessionTemplate getSqlSessionTemplate() {

    String clusterName = multiDataSourceSelector.getName();

    SqlSessionTemplate result = sqlSessionTemplateMap.get(clusterName);

    Assert.notNull(result);

    return result;

  }

}

MultiDataSourceSelector是一个借口，根据当前的调用环境，返回不不同的参数，根据这个参数就可以确定使用哪一个SQLSessionTemplate，例如我是把当前环境放入到ThreadLocal中的

public interface MultiDataSourceSelector {

  String getName();

}

public class DubboContextDataSourceSelector implements MultiDataSourceSelector {

 

  private String defaultName;

 

  public DubboContextDataSourceSelector(String defaultName) {

    this.defaultName = defaultName;

  }

 

  @Override

  public String getName() {

    //DubboContextHolder 是一个保持一个ThreadLocal的Map

    String res = DubboContextHolder.getContext().get(DubboContextConstants.CLUSTER_NAME);

    if (res == null) {

      res = getDefaultName();

    }

    return res;

  }

 

  public String getDefaultName() {

    return defaultName;

  }

}

然后在Dao层的中获取SQLSessionTemplate的时候就是动态了。

动态事务

其实这个都好办，然后我们就面临一个稍微复杂一点的问题，那DataSource是动态的，事务也就必须是动态了的。而且还对原有的代码没有侵入（例如Spring中的@Transactional 注解），那实现一个类似@Transactional的方法吧。名字就叫做@DynamicTransactional

@Documented

@Target({METHOD, TYPE})

@Retention(RUNTIME)

public @interface DynamicTransactional {

 

  Propagation propagation() default Propagation.REQUIRED;

 

  Class<? extends Throwable>[] rollbackFor() default {};

}

基本思想是在通过AOP切面拦截@DynamicTransactional注解，调用，然后自己编程实现事务

详解基于spring多数据源动态调用及其事务处理

切面内的核心方法是

private Object invokeWithinTransaction(final ProceedingJoinPoint pjp, final DynamicTransactional dynamicTransaction) {

 

  //创建TransactionTemplate

  final PlatformTransactionManager tran = multiTransactionManagerHolder.getTransactionManager();

  TransactionTemplate transactionTemplate = new TransactionTemplate();

  transactionTemplate.setPropagationBehavior(dynamicTransaction.propagation().value());

  transactionTemplate.setTransactionManager(tran);

 

  //在事务中执行

  return transactionTemplate.execute(new TransactionCallback

详解基于spring多数据源动态调用及其事务处理

其他资讯