log.info(“通过key确定当前应该使用哪个数据源, key:{}”, dataSourceKey);
return dataSourceKey;
}
public static void setDataSourceKey(String dataSourceKey) {
log.info(“设置需要使用的数据源key, key:{}”, dataSourceKey);
ThreadLocalDataSourceKey.set(dataSourceKey);
}
}
把动态路由数据源设置主数据源,并给配置不同的数据源
@Bean
@Primary
public ThreadLocalRountingDataSource threadLocalRountingDataSource() {
ThreadLocalRountingDataSource dataSource = new ThreadLocalRountingDataSource();
Map<Object, Object> targetDataSources = new HashMap();
DataSource master = dataSource1();//默认数据源
DataSource other = dataSource2();//其他数据源
targetDataSources.put(“master”, master);
targetDataSources.put(“other”, other);
// 设置数据源池的key和数据源的映射
dataSource.setTargetDataSources(targetDataSources);
// 把主数据源设置为默认数据源
dataSource.setDefaultTargetDataSource(master);
return dataSource;
}
3. 自定义注解+aop实现
自定义注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface LakerDataSource {
String value() default “master”;
}
定义拦截器
@Aspect
@Order(Ordered.LOWEST_PRECEDENCE - 1)
@Component
public class LakerDataSourceAspect {
protected Logger logger = LoggerFactory.getLogger(getClass());
@Pointcut(“@annotation(com.laker.notes.multidatasource.LakerDataSource)”)
public void lakerDataSource() {
}
@Around(“lakerDataSource()”)
public Object around(ProceedingJoinPoint point) throws Throwable {
LakerDataSource dataSource = getDataSource(point);
//设置数据源
if (dataSource != null) {
ThreadLocalRountingDataSource.setDataSourceKey(dataSource.value());
}
try {
return point.proceed();
} catch (Exception e) {
throw e;
} finally {
// 销毁数据源 在执行方法之后
ThreadLocalRountingDataSource.clear();
}
}
/**
- 获取需要切换的数据源
*/
public LakerDataSource getDataSource(ProceedingJoinPoint point) {
MethodSignature signature = (MethodSignature) point.getSignature();
LakerDataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(), LakerDataSource.class);
if (Objects.nonNull(dataSource)) {
return dataSource;
}
return AnnotationUtils.findAnnotation(signature.getDeclaringType(), LakerDataSource.class);
}
}
重点: 该切面必须要在事务注解@Transactional之前,由于在开始事务之前就需要确定数据源,所以设置DataSourceAsp的
@Order(Ordered.LOWEST_PRECEDENCE-1)
,@Transactional的order是最小值
4. 验证
sevice
@Service
public class UserService {
@Autowired
private UserDao userDao;
@LakerDataSource(“master”)
public LakerUser getDataSourceMaster(Long id) {
return userDao.findById(id).get();
}
@LakerDataSource(“other”)
public LakerUser getDataSourceOther(Long id) {
return userDao.findById(id).get();
}
}
controller
@RestController
@Slf4j
public class DataSourceController {
@Autowired
private UserService userService;
@GetMapping(“/master”)
public LakerUser master(Long id) throws Exception {
LakerUser lakerUser = userService.getDataSourceMaster(id);
log.info(lakerUser.toString());
return lakerUser;
}
@GetMapping(“/other”)
public LakerUser other(Long id) throws Exception {
LakerUser lakerUser = userService.getDataSourceOther(id);
log.info(lakerUser.toString());
return lakerUser;
}
}
浏览器输入:http://localhost:8080/master?id=1
[nio-8080-exec-1] c.l.n.m.ThreadLocalRountingDataSource : 设置需要使用的数据源key, key:master
[nio-8080-exec-1] c.l.n.m.ThreadLocalRountingDataSource : 通过key确定当前应该使用哪个数据源, key:master
[nio-8080-exec-1] c.l.n.m.ThreadLocalRountingDataSource : 清除当前线程数据源
[nio-8080-exec-1] c.l.n.m.DataSourceController : LakerUser(id=1, name=laker, age=30)
浏览器输入:http://localhost:8080/other?id=1
[nio-8080-exec-2] c.l.n.m.ThreadLocalRountingDataSource : 设置需要使用的数据源key, key:other
[nio-8080-exec-2] c.l.n.m.ThreadLocalRountingDataSource : 通过key确定当前应该使用哪个数据源, key:other
[nio-8080-exec-2] com.zaxxer.hikari.HikariDataSource : HikariPool-2 - Starting…
[nio-8080-exec-2] com.zaxxer.hikari.HikariDataSource : HikariPool-2 - Start completed.
[nio-8080-exec-2] c.l.n.m.ThreadLocalRountingDataSource : 清除当前线程数据源
[nio-8080-exec-2] c.l.n.m.DataSourceController : LakerUser(id=1, name=laker100, age=100)
从日志中可以看到,已经实现了动态数据源的切换功能。
相关错误解决方案:
- 数据源循环引用问题,错误如下:
The dependencies of some of the beans in the application context form a cycle:
org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaConfiguration
┌─────┐
| threadLocalRountingDataSource defined in class path resource [com/laker/notes/multidatasource/DataSourceConfig.class]
↑ ↓
| dataSource-1 defined in class path resource [com/laker/notes/multidatasource/DataSourceConfig.class]
↑ ↓
| org.springframework.boot.autoconfigure.jdbc.DataSourceInitializerInvoker
└─────┘
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。
深知大多数Java工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!
因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!
由于文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注Java)
总结
我们总是喜欢瞻仰大厂的大神们,但实际上大神也不过凡人,与菜鸟程序员相比,也就多花了几分心思,如果你再不努力,差距也只会越来越大。实际上,作为程序员,丰富自己的知识储备,提升自己的知识深度和广度是很有必要的。
Mybatis源码解析
知识点,真正体系化!**
由于文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注Java)
[外链图片转存中…(img-fZWal1Lr-1711655000600)]
总结
我们总是喜欢瞻仰大厂的大神们,但实际上大神也不过凡人,与菜鸟程序员相比,也就多花了几分心思,如果你再不努力,差距也只会越来越大。实际上,作为程序员,丰富自己的知识储备,提升自己的知识深度和广度是很有必要的。
Mybatis源码解析
[外链图片转存中…(img-0JImxkng-1711655000601)]
[外链图片转存中…(img-TiBg8Zx2-1711655000601)]