spring动态切换数据库支持事务

在项目中有mysql的多个库，在代码中同一个方法可能会操作不同的表。在网上学习了各种方法。大概总结了一下。

1.mycat、cobar等分布式数据库中间件。

可以很好的支持，但是太重量级了，对我们项目有点大材小用。

2.spring的AbstractRoutingDataSource实现数据库连接切换。

可以动态的切换数据源，但是对事务有影响，可以用JTA实现事务一致，但是效率较低。而且我们项目事务可以单库一致就满足需求。所以采用了这种方式。

下面是具体的实现过程：

1）spring的配置文件中配置多个数据源。

2）定义动态的数据源

3）定义动态数据源和辅助类

public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
String type = DataSourceContextHolder.getDataSourceType();
return type;
}
}

public class DataSourceContextHolder {
private static final ThreadLocal contextHolder = new ThreadLocal();

/**
* @Description: 设置数据源类型
* @param dataSourceType
* 数据库类型
* @return void
* @throws
*/
public static void setDataSourceType(String dataSourceType) {
contextHolder.set(dataSourceType);
}

/**
* @Description: 获取数据源类型
* @param
* @return String
* @throws
*/
public static String getDataSourceType() {
return contextHolder.get();
}

/**
* @Description: 清除数据源类型
* @param
* @return void
* @throws
*/
public static void clearDataSourceType() {
contextHolder.remove();
}
}

4）修改事务管理器的数据源为动态数据源，指定事务注解的排序为2，我们会指定切换数据源的注解为1，这样在事务之前切换数据源，否则在事务之后切换的的话，无效。

5）定义切换数据库的注解和aop切面，指定排序为1，这里有个疑问，通过切点获取代理方法的注解数据，我用的是反射，但是网上有说可以直接作为参数传入的，我一直没有试验成功，不知道哪里有错，后续哪位大神指导的，可以分享一下。

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Documented
public @interface DataSource {
String name();

}

@Component
@Aspect
@Order(1)
public class DataSourceProxy {

@Before(value="@annotation(com.futuren.wzk.common.datasource.DataSource)")
public void before(JoinPoint jp) {
String methodName = jp.getSignature().getName();
Method[] methods = jp.getTarget().getClass().getMethods();
for(Method method : methods) {
if(method.getName().equals(methodName)) {
DataSource ds = method.getAnnotation(DataSource.class);
DataSourceContextHolder.setDataSourceType(ds.name());
}
}
}
}

6）在项目中使用

@Override
@Transactional
@DataSource(name="ucenter")
public int addUser(User user) {
userMapper.insert(user);
return user.getUid();
}

这种方法，只支持单库事务，如果要多库事务，可能要引入JTA，或者是其他自定义实现。或者其他我不知道的技术。欢迎讨论！