MyBatis-Plus
是一個(gè)非常不錯(cuò)的 MyBatis 的增強(qiáng)工具，為簡(jiǎn)化開(kāi)發(fā)、提高效率而生。
MyBatis-Plus支持多數(shù)據(jù)源和動(dòng)態(tài)數(shù)據(jù)源（dynamic-datasource），以適配不同需求和場(chǎng)景。
基本配置
引入Maven依賴
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.4.3.4</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
<version>3.5.2</version>
</dependency>
動(dòng)態(tài)數(shù)據(jù)源配置
spring:
datasource:
dynamic:
primary: master1
strict: true
datasource:
master1:
driverClassName: 驅(qū)動(dòng)
url: URL
username: 用戶名
password: 密碼
master2:
driverClassName: 驅(qū)動(dòng)
url: URL
username: 用戶名
password: 密碼
master3:
driverClassName: 驅(qū)動(dòng)
url: URL
username: 用戶名
password: 密碼
slave1:
driverClassName: 驅(qū)動(dòng)
url: URL
username: 用戶名
password: 密碼
slave2:
driverClassName: 驅(qū)動(dòng)
url: URL
username: 用戶名
password: 密碼
AOP切面類
@Component
@Order(value = 0)
@Slf4j
@Aspect
public class DataSourceAspectOnMapper {
@Value("${spring.datasource.dynamic.primary}")
private String defaultDatabase;
/**
* 匹配mapper所在的package所有類（及子類）中所有方法（所有返回類型，所有類型參數(shù)）的執(zhí)行
*/
@Pointcut("execution(* mapper所在的package..*.*(..))")
private void masterAspect() {
}
/**
* 執(zhí)行Mapper方法前設(shè)置數(shù)據(jù)源
*/
@Before("masterAspect()")
public void beforeMasterDb() throws Exception{
String dataSource = DataSourceUtil.chooseDatasource(defaultDatabase);
DynamicDataSourceContextHolder.push(dataSource);
}
@After("masterAspect()")
public void afterMasterDb() {
DynamicDataSourceContextHolder.poll();
}
}
因?yàn)樵谑聞?wù)中對(duì)Mapper切換數(shù)據(jù)源無(wú)效，需要把所有執(zhí)行數(shù)據(jù)庫(kù)事務(wù)的類也放在同一個(gè)代碼包內(nèi)，對(duì)事務(wù)也進(jìn)行AOP切面，動(dòng)態(tài)設(shè)置數(shù)據(jù)源。源碼略。
踩坑
針對(duì)數(shù)據(jù)庫(kù)表（以表名DEMO為例），自動(dòng)生成對(duì)應(yīng)的Entity、Mapper、Service等文件后，主要文件示例：
public interface DemoService extends IService<Demo> {
}
@Service
public class DemoServiceImpl extends ServiceImpl<DemoMapper, Demo> implements DemoService {
}
@SpringBootTest
class ApplicationTests {
@Autowired
private DemoService demoService;
@Test
public void batchInsert(){
demo.setGuid(java.util.UUID.randomUUID().toString());
//下面的方法正常
boolean ret = demoService.save(demo);
System.out.println(ret);
//下面的方法出錯(cuò)，提示“表或視圖不存在”
boolean ret1 = demoService.saveOrUpdate(demo);
System.out.println(ret1);
}
}
填坑
源碼分析
com.baomidou.mybatisplus.extension.service.IService.java 部分源碼：
public interface IService<T> {
/**
* 默認(rèn)批次提交數(shù)量
*/
int DEFAULT_BATCH_SIZE = 1000;
/**
* 插入一條記錄（選擇字段，策略插入）
*
* @param entity 實(shí)體對(duì)象
*/
default boolean save(T entity) {
return SqlHelper.retBool(getBaseMapper().insert(entity));
}
/**
* TableId 注解存在更新記錄，否插入一條記錄
*
* @param entity 實(shí)體對(duì)象
*/
boolean saveOrUpdate(T entity);
}
com.baomidou.mybatisplus.extension.service.impl.ServiceImpl.java 部分源碼：
/**
* IService 實(shí)現(xiàn)類（ 泛型：M 是 mapper 對(duì)象，T 是實(shí)體 ）
*/
@SuppressWarnings("unchecked")
public class ServiceImpl<M extends BaseMapper<T>, T> implements IService<T> {
protected Log log = LogFactory.getLog(getClass());
@Autowired
protected M baseMapper;
/**
* TableId 注解存在更新記錄，否插入一條記錄
*
* @param entity 實(shí)體對(duì)象
* @return boolean
*/
@Transactional(rollbackFor = Exception.class)
@Override
public boolean saveOrUpdate(T entity) {
if (null != entity) {
TableInfo tableInfo = TableInfoHelper.getTableInfo(this.entityClass);
Assert.notNull(tableInfo, "error: can not execute. because can not find cache of TableInfo for entity!");
String keyProperty = tableInfo.getKeyProperty();
Assert.notEmpty(keyProperty, "error: can not execute. because can not find column for id from entity!");
Object idVal = ReflectionKit.getFieldValue(entity, tableInfo.getKeyProperty());
return StringUtils.checkValNull(idVal) || Objects.isNull(getById((Serializable) idVal)) ? save(entity) : updateById(entity);
}
return false;
}
}
save方法和saveOrUpdate區(qū)別：
save()是在接口中定義的default方法，saveOrUpdate()是在實(shí)現(xiàn)類中實(shí)現(xiàn)的帶事務(wù)注解@Transactional的方法。
原因及解決方法
因?yàn)樵创a中saveOrUpdate使用了事務(wù)，故在Mapper切面設(shè)置數(shù)據(jù)源無(wú)效。
對(duì)MyBatis-plus的服務(wù)包進(jìn)行切面，代碼如下：
@Component
@Order(value = 0)
@Slf4j
@Aspect
public class DataSourceAspectOnService {
@Value("${spring.datasource.dynamic.primary}")
private String defaultDatabase;
/**
* 匹配service所在的package所有類（及子類）中所有方法（所有返回類型，所有類型參數(shù)）的執(zhí)行
*/
@Pointcut("execution(* service所在的package..*.*(..))")
private void masterAspect() {
}
/**
* 執(zhí)行Service方法前設(shè)置數(shù)據(jù)源
*/
@Before("masterAspect()")
public void beforeMasterDb() throws Exception{
String dataSource = DataSourceUtil.chooseDatasource(defaultDatabase);
DynamicDataSourceContextHolder.push(dataSource);
}
@After("masterAspect()")
public void afterMasterDb() {
DynamicDataSourceContextHolder.poll();
}
}
此時(shí)仍未解決問(wèn)題，經(jīng)查是因?yàn)镈emoServiceImpl未覆蓋父類方法saveOrUpdate，切面程序未執(zhí)行。
@Service
public class DemoServiceImpl extends ServiceImpl<DemoMapper, Demo> implements DemoService {
@Override
public boolean saveOrUpdate(Demo entity) {
return super.saveOrUpdate(entity);
}
}
問(wèn)題解決。這里不能使用@Pointcut("execution(* com.baomidou.mybatisplus.extension.service.impl.ServiceImpl+.*(..))")進(jìn)行切面，因?yàn)樵擃愅瑫r(shí)被master數(shù)據(jù)源和slave這兩類異構(gòu)的數(shù)據(jù)源的Service所繼承，AOP程序在此處無(wú)法確定使用哪個(gè)數(shù)據(jù)源。
本文到此結(jié)束，感謝您的觀看?。?！
