Spring Boot 实现读写分离，还有谁不会？

@SpringBootApplication

public class MySpringBootApplication {

/**

• Master data source.

*/

@Bean(“masterDataSource”)

@ConfigurationProperties(prefix = “spring.datasource”)

DataSource masterDataSource() {

logger.info(“create master datasource…”);

return DataSourceBuilder.create().build();

}

/**

• Slave (read only) data source.

*/

@Bean(“slaveDataSource”)

@ConfigurationProperties(prefix = “spring.ro-datasource”)

DataSource slaveDataSource() {

logger.info(“create slave datasource…”);

return DataSourceBuilder.create().build();

}

…

}

复制代码

第二步：编写 RoutingDataSource

然后，我们用 Spring 内置的 RoutingDataSource，把两个真实的数据源代理为一个动态数据源：

public class RoutingDataSource extends AbstractRoutingDataSource {

@Override

protected Object determineCurrentLookupKey() {

return “masterDataSource”;

}

}

复制代码

对这个RoutingDataSource，需要在 SpringBoot 中配置好并设置为主数据源：

@SpringBootApplication

public class MySpringBootApplication {

@Bean

@Primary

DataSource primaryDataSource(

@Autowired @Qualifier(“masterDataSource”) DataSource masterDataSource,

@Autowired @Qualifier(“slaveDataSource”) DataSource slaveDataSource

) {

logger.info(“create routing datasource…”);

Map<Object, Object> map = new HashMap<>();

map.put(“masterDataSource”, masterDataSource);

map.put(“slaveDataSource”, slaveDataSource);

RoutingDataSource routing = new RoutingDataSource();

routing.setTargetDataSources(map);

routing.setDefaultTargetDataSource(masterDataSource);

return routing;

}

…

}

复制代码

现在，RoutingDataSource 配置好了，但是，路由的选择是写死的，即永远返回"masterDataSource"，

现在问题来了：如何存储动态选择的 key 以及在哪设置 key？

在 Servlet 的线程模型中，使用 ThreadLocal 存储 key 最合适，因此，我们编写一个 RoutingDataSourceContext，来设置并动态存储 key：

public class RoutingDataSourceContext implements AutoCloseable {

// holds data source key in thread local:

static final ThreadLocal threadLocalDataSourceKey = new ThreadLocal<>();

public static String getDataSourceRoutingKey() {

String key = threadLocalDataSourceKey.get();

return key == null ? “masterDataSource” : key;

}

public RoutingDataSourceContext(String key) {

threadLocalDataSourceKey.set(key);

}

public void close() {

threadLocalDataSourceKey.remove();

}

}

复制代码

然后，修改 RoutingDataSource，获取 key 的代码如下：

public class RoutingDataSource extends AbstractRoutingDataSource {

protected Object determineCurrentLookupKey() {

return RoutingDataSourceContext.getDataSourceRoutingKey();

}

}

复制代码

这样，在某个地方，例如一个 Controller 的方法内部，就可以动态设置 DataSource 的 Key：

@Controller

public class MyController {

@Get(“/”)

public String index() {

String key = “slaveDataSource”;

try (RoutingDataSourceContext ctx = new RoutingDataSourceContext(key)) {

// TODO:

return “html… www.liaoxuefeng.com”;

}

}

}

复制代码

到此为止，我们已经成功实现了数据库的动态路由访问。

这个方法是可行的，但是，需要读从数据库的地方，就需要加上一大段try (RoutingDataSourceContext ctx = ...) {}代码，使用起来十分不便。有没有方法可以简化呢？

有！

我们仔细想想，Spring 提供的声明式事务管理，就只需要一个@Transactional()注解，放在某个 Java 方法上，这个方法就自动具有了事务。

我们也可以编写一个类似的@RoutingWith("slaveDataSource")注解，放到某个 Controller 的方法上，这个方法内部就自动选择了对应的数据源。代码看起来应该像这样：

@Controller

public class MyController {

@Get(“/”)

@RoutingWith(“slaveDataSource”)

public String index() {

return “html… www.liaoxuefeng.com”;

}

}

复制代码

这样，完全不修改应用程序的逻辑，只在必要的地方加上注解，自动实现动态数据源切换，这个方法是最简单的。

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

最后

笔者已经把面试题和答案整理成了面试专题文档

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
(img-NwpGaVIw-1713387250788)]

[外链图片转存中…(img-hLsqzOtt-1713387250788)]

[外链图片转存中…(img-krzZIH8N-1713387250789)]

[外链图片转存中…(img-ZsQ5dglX-1713387250789)]

[外链图片转存中…(img-H6vZUYNG-1713387250789)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！