5分钟带你手撸一个数据库读写分离、多数据源切换组件

最后，附一张自己面试前准备的脑图：

面试前一定少不了刷题，为了方便大家复习，我分享一波个人整理的面试大全宝典

• Java核心知识整理

• Spring全家桶（实战系列）

Step3：刷题

既然是要面试，那么就少不了刷题，实际上春节回家后，哪儿也去不了，我自己是刷了不少面试题的，所以在面试过程中才能够做到心中有数，基本上会清楚面试过程中会问到哪些知识点，高频题又有哪些，所以刷题是面试前期准备过程中非常重要的一点。

以下是我私藏的面试题库：

很多人感叹“学习无用”，实际上之所以产生无用论，是因为自己想要的与自己所学的匹配不上，这也就意味着自己学得远远不够。无论是学习还是工作，都应该有主动性，所以如果拥有大厂梦，那么就要自己努力去实现它。

最后祝愿各位身体健康，顺利拿到心仪的offer！

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取

这种方式当然可以实现，但是呢不够优雅，下面我们讲下如何使用更加优雅的实现方式。重点：AbstractRoutingDataSource

2. 准备动态数据源

AbstractRoutingDataSource是什么？

这个可以直接查看源码，里面注释如下：

Abstract {@link javax.sql.DataSource} implementation that routes {@link #getConnection()}

calls to one of various target DataSources based on a lookup key. The latter is usually

(but not necessarily) determined through some thread-bound transaction context.

白话翻译一下：通过线程绑定的事务上下文动态确定数据源

核心属性、方法如下：

// 保存了key和数据源的映射

private Map<Object, Object> targetDataSources;

// 默认数据源

private Object defaultTargetDataSource;

// 抽像方法，需要重写然后在protected DataSource determineTargetDataSource() 中调用

// 用来确定使用哪个数据源，核心

protected abstract Object determineCurrentLookupKey();

那么我们创建建一个作为默认的datasource继承自AbstractRoutingDataSource，并且覆盖determineCurrentLookupKey()方法，来以实现数据库的动态切换

@Slf4j

public class ThreadLocalRountingDataSource extends AbstractRoutingDataSource {

private static final ThreadLocal ThreadLocalDataSourceKey = new ThreadLocal();

public static void clear() {

log.info(“清除当前线程数据源”);

ThreadLocalDataSourceKey.remove();

}

@Override

protected Object determineCurrentLookupKey() {

String dataSourceKey = ThreadLocalDataSourceKey.get();

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)

从日志中可以看到，已经实现了动态数据源的切换功能。

相关错误解决方案：

1. 数据源循环引用问题，错误如下:

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

└─────┘

解决方案：

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})

public class LakerApplication {

总结

---

总结下三板斧

• 核心 ：AbstractRoutingDataSource，spring爸爸给咱留的后门

• 自定义注解DataSource

• 自定义AOP拦截器

这些套路，会是我们增强spring功能的常用武器。

---

QQ群【837324215】

关注我的公众号【Java大厂面试官】，回复：架构、资源等关键词（更多关键词，关注后注意提示信息）获取更多免费资料。
公众号也会持续输出高质量文章，和大家共同进步。

读者福利

由于篇幅过长，就不展示所有面试题了，感兴趣的小伙伴

更多笔记分享

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取

是我们增强spring功能的常用武器。

---

QQ群【837324215】

关注我的公众号【Java大厂面试官】，回复：架构、资源等关键词（更多关键词，关注后注意提示信息）获取更多免费资料。
公众号也会持续输出高质量文章，和大家共同进步。

读者福利

由于篇幅过长，就不展示所有面试题了，感兴趣的小伙伴

[外链图片转存中…(img-krXoyUmJ-1715796090068)]

[外链图片转存中…(img-LmI2awy0-1715796090068)]

[外链图片转存中…(img-KHDPF5WF-1715796090069)]

更多笔记分享

[外链图片转存中…(img-7yodvrlA-1715796090069)]

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取