mybatis-plus多数据源实现和原理
想了解mybatis-plus如何实现多数据源的小伙伴,可以参考多数据源解析 | 米二
该文章详细说明了如何通过实现多源数据(使用了@DS注解),并且解析了原理。简单来说,就是如果有@DS注解,则会新建ThreadLocal,并绑定对应的数据库key(DS中的数值),原来的线程入栈,切换到新的ThreadLocal,并重新建立数据库连接,从而实现数据源的切换,如果没有@DS注解,则会走主库。
mybatis-plus多数据源+事务
但是如果要在上面的基础上引入@Transactional,则会发现数据库切换失败,这是因为:
-
开启事务的同时,会从数据库连接池获取数据库连接;TransactionInterceptor 会使用 Spring DataSourceTransactionManager 创建事务,并将事务信息(获取数据源connection连接,此时获取到的数据源是默认配置的base数据源信息)连接信息,通过 ThreadLocal 绑定在当前线程。
- 此时当前线程事务绑定的连接信息是base数据源,当我们在内层service或者mapper使用@DS切换数据源,并没有重新开启新事务,没有改变当前线程事务的连接信息,仅仅是做了一次拦截,改变了DataSourceHolder的栈顶dataSource,对于整个事务的连接是没有影响的,所以会产生数据源没有切换的问题。
所以针对这个问题,通常有两种解决办法:
一.@DSTransactional
最简单的办法是使用@DSTransactional代替@Transactional。
@DSTransactional不像@Transactional那样可以定义回滚异常与非回滚异常,也没有事务的传播机制,但是它提供了对多数据源事务的支持。更复杂的请用seata
@DSTransactional的原理见@DSTransactional注解原理-优快云博客
二.@Transactional + Propagation.REQUIRES_NEW
@Transactional(rollbackFor = Exception.class),默认事务的传播机制是,PROPAGATION_REQUIRED ,
如果是在同一个事务里面,sqlsession是沿用同一个,导致connection不会切换,因此无法切换数据源,如果在内层使用事务已经将传播方式设置为@Transactional(propagation = Propagation.REQUIRES_NEW)
,那么就会当做另起一个新的事务,也就会重新获取连接,而且在内层方法里面如果抛出异常的话,也会导致外层事务回滚的(相当于外部事务套了一个内部事务)。但是如果内层事务执行完成,外层再抛出异常,则内层事务无法回滚,因为内层事务已经提交了。因此,只能有一个内层事务,否则外部事务仍然会失效。
代码实现如下:
yaml数据
spring:
datasource:
dynamic:
primary: master
datasource:
master:
driverClassName: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://user
username: root
password: root
common:
driverClassName: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://book
username: root
password: root
service如下,默认是master数据源,其中user为master数据库表,book为common数据库表
@Service
@Slf4j
public class MasterService {
@Autowired
UserService userService;
@Autowired
BookService bookService;
/**必须master库方法先执行,才能回滚,达到事务效果*/
@Transactional(rollbackFor = Exception.class)
public void upload(ReqDto reqDto){
userService.save(reqDto);
bookService.save(reqDto);
}
}
userService
@Service
@Log4j2
public class UserService extends ServiceImpl<UserMapper, User> {
@Resource
private UserMapper userMapper;
public void save(ReqDto reqDto) {
userMapper.save(reqDto);
}
}
bookService
@Service
@Log4j2
@DS("common")
public class BookService extends ServiceImpl<BookMapper, Book> {
@Resource
private BookMapper bookMapper;
@Transactional(propagation = Propagation.REQUIRES_NEW, rollbackFor = Exception.class)
public void save(ReqDto reqDto) {
bookMapper.save(reqDto);
}
}
common数据库的操作,需要在master之后,这样当bookService.save失败,会使得userService回滚;
如果先common的操作,那当userService失败,无法使bookService回滚