数据库-MySQL-MybatisPlus整合多数据源

真上帝的左手2024-11-27 12:12

文章目录

发现问题

在平时开发中，经常会遇到在一个项目里需要使用多个数据源的情况，比如有一部分数据在数据源A，另一部分数据在数据源B，业务需要把这两部分的数据做合并然后从接口返回。

又或者操作完数据源A后，需要切换数据源，操作数据源B。这样的需求，怎么实现？

解决问题

其实在mybatis-plus就有相关的实现，是一个基于SpringBoot快速集成多数据源的启动器。

首先要搭建一个springBoot+Mybatis+Mybatis-Plus的项目，搭建项目就不演示了，比较简单。这里讲怎么使用多数据源，首先引入dynamic-datasource-spring-boot-starter。

xml 复制代码 
<dependency>
  <groupId>com.baomidou</groupId>
  <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
  <version>${version}</version><!--版本号-->
</dependency>

第二步，修改application配置文件，配置数据源。

properties 复制代码 
# 默认数据源
spring.datasource.dynamic.primary=master
# 数据源A
spring.datasource.dynamic.datasource.master.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.dynamic.datasource.master.url=jdbc:mysql://192.168.0.101:3306/user?createDatabaseIfNotExist=true
spring.datasource.dynamic.datasource.master.username=root
spring.datasource.dynamic.datasource.master.password=
# 数据源B
spring.datasource.dynamic.datasource.slave.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.dynamic.datasource.slave.url=jdbc:mysql://192.168.0.102:3306/user?createDatabaseIfNotExist=true
spring.datasource.dynamic.datasource.slave.username=root
spring.datasource.dynamic.datasource.slave.password=

第三步，使用@DS注解切换数据源。一般我喜欢放在Mapper接口上面。注解的值则是数据源的名称。

现在有两个数据源：

slave数据源的user表对应的Mapper接口，如下：

java 复制代码 
@Mapper
@DS("slave")
public interface UserMapper extends BaseMapper<User> {
}

master数据源的commodity表对应的Mapper接口，如下：

java 复制代码 
@Mapper
@DS("master")
public interface CommodityMapper extends BaseMapper<Commodity> {
}

接着我们写一个Service进行测试：

java 复制代码 
@Service
public class DynamicServiceImpl implements DynamicService {

    @Resource
    private UserMapper userMapper;

    @Resource
    private CommodityMapper commodityMapper;

    @Override
    public ResultObject getUserListAndCommodityList() {
        //查询slave数据源
        List<User> userList = userMapper.selectList(null);
        //查询master数据源
        List<Commodity> commodityList = commodityMapper.selectList(null);
        //把两个数据源查询的整合到一起返回
        ResultObject resultObject = new ResultObject();
        resultObject.setUserList(userList);
        resultObject.setCommodityList(commodityList);
        return resultObject;
    }
}

Controller接口，如下：

java 复制代码 
@RestController
@RequestMapping("/dynamic")
public class DynamicController {

    @Resource
    private DynamicService dynamicService;

    @RequestMapping("/getUserListAndCommodityList")
    public ResultObject getUserListAndCommodityList() {
        return dynamicService.getUserListAndCommodityList();
    }
}

启动项目，然后请求接口地址，我们可以看到结果里包含了两个数据源查询到的数据：

一些疑问

问题一：@DS注解可以在什么地方使用？

看源码，可以知道是在类和方法上使用。

java 复制代码 
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DS {
    /**
     * groupName or specific database name or spring SPEL name.
     *
     * @return the database you want to switch
     */
    String value();
}

一般在Service或者Mapper上面进行标记。在方法上面标记优先级高于在类上标记。

问题二：支持多层嵌套切换数据源吗？

什么是嵌套切换数据源，先看个例子：

java 复制代码 
@Service
@DS("slave")
public class UserServiceImpl implements UserService {

    @Resource
    private UserMapper userMapper;

    @Override
    public List<User> getList() {
        return userMapper.selectList(null);
    }
}
java 复制代码 
@Service
@DS("master")
public class CommodityServiceImpl implements CommodityService {
    
    @Resource
    private UserService userService;
    
    @Override
    public ResultObject getUserAndCommodity() {
        //切换到slave数据源，查询user表
        List<User> userList = userService.getList();
        //切换回master数据源，查询commodity表
        List<Commodity> commodityList = commodityMapper.selectList(null);
        ResultObject resultObject = new ResultObject();
        resultObject.setUserList(userList);
        resultObject.setCommodityList(commodityList);
        return resultObject;
    }
}
java 复制代码 
@RestController
@RequestMapping("/dynamic")
public class DynamicController {

    @Resource
    private CommodityService commodityService;

    @RequestMapping("/getUserListAndCommodityList")
    public ResultObject getUserListAndCommodityList() {
        return commodityService.getUserAndCommodity();
    }
}

启动项目，然后请求地址，发现依然能查出结果，证明是支持嵌套切换数据源的。

问题三：如何使用@Transactional注解提供事务支持？

我们看个例子：

首先定义一个插入对象InsertRequest：

java 复制代码 
public class InsertRequest {

    private User user;

    private Commodity commodity;
}

然后在Service类加上@Transactional注解，并把数据分别插入两个数据源的表中：

java 复制代码 
@Service
@Transactional
public class DynamicServiceImpl implements DynamicService {

    @Resource
    private UserMapper userMapper;

    @Resource
    private CommodityMapper commodityMapper;
    
    
    @Override
    public void insertUserAndCommodity(InsertRequest insertRequest) {
        commodityMapper.insert(insertRequest.getCommodity());
        userMapper.insert(insertRequest.getUser());
    }
}

然后在Controller接口层增加一个入口方法：

java 复制代码 
@RequestMapping("/insertUserAndCommodity")
public String insertUserAndCommodity(@RequestParam(name = "name") String name,
                                     @RequestParam(name = "age") Integer age,
                                     @RequestParam("commodityName") String commodityName,
                                     @RequestParam("commodityPrice") String commodityPrice) {
    //构建user对象
    User user = new User();
    user.setName(name);
    user.setAge(age);
    //构建commodity对象
    Commodity commodity = new Commodity();
    commodity.setCommodityName(commodityName);
    commodity.setCommodityPrice(commodityPrice);
    //构建insertRequest对象
    InsertRequest insertRequest = new InsertRequest();
    insertRequest.setUser(user);
    insertRequest.setCommodity(commodity);
    //调用插入方法
    dynamicService.insertUserAndCommodity(insertRequest);
    return "success";
}

启动项目，测试，结果是500错误：

我们看控制台日志可以发现问题：

学过Spring都知道@Transactional是本地事务管理，不是分布式事务，所以我们不能在需要切换数据源的方法上加@Transactional注解。

不使用分布式事务的情况下，我们只能让其各自管理各自的事务，中间加多一层Service。

java 复制代码 
@Service
@DS("master")
public class CommodityServiceImpl implements CommodityService {
    
    @Resource
    private CommodityMapper commodityMapper;
    
    @Override
    @Transactional
    public int insertOne(Commodity commodity) {
        return commodityMapper.insert(commodity);
    }
}

@Service
@DS("slave")
public class UserServiceImpl implements UserService {

    @Resource
    private UserMapper userMapper;
    
    @Override
    @Transactional
    public int insertOne(User user) {
        return userMapper.insert(user);
    }
}
java 复制代码 
@Service
public class DynamicServiceImpl implements DynamicService {
    @Resource
    private UserService userService;

    @Resource
    private CommodityService commodityService;
    
    @Override
    public void insertUserAndCommodity(InsertRequest insertRequest) {
        commodityService.insertOne(insertRequest.getCommodity());
        userService.insertOne(insertRequest.getUser());
    }
}

这样改完之后，我们再测一下，可以看到数据源切换成功了。

如果需要使用分布式事务，官网也说明提供基于seata的分布式事务方案，这里就不演示了，读者有兴趣可以自己尝试一下。

问题四：同一个Service里面，定义了两个不同的方法对应不同的数据源，能互相调用吗？

我们还是以事实说话，先整个例子：

java 复制代码 
@Override
@DS("slave")
public ResultObject getResultObject() {
    List<User> userList = userService.getList();
    //调用同一个类的另一个方法切换数据源
    List<Commodity> commodityList = getCommodityList();
    ResultObject resultObject = new ResultObject();
    resultObject.setUserList(userList);
    resultObject.setCommodityList(commodityList);
    return resultObject;
}

@Override
@DS("master")
public List<Commodity> getCommodityList() {
    return commodityMapper.selectList(null);
}

启动项目，测试，发现报错了：

原因是数据源切换的底层原理是使用AOP实现的，如果是内部方法调用是不会使用AOP的，所以也就没法切换数据源。

解决的方法是把getCommodityList()方法提到另一个类里面去。

总结

dynamic-datasource其实还有很多功能没介绍，这里就不一一介绍了，有兴趣的可以到官网上去学习。多数据源在项目开发中是很常见的，所以学习这个插件还是很有用的。后面如果有时间的话，可以解读一下dynamic-datasource源码，探索一下底层的实现原理。

参考文章
mybatis-plus整合多数据源

相关推荐
炽天使3 分钟前
aws rds-mysql不支持性能详情监控
linux·数据库·mysql·云计算·aws·rds
QQ_11543203131 分钟前
基于Java+SpringBoot+Mysql在线简单拍卖竞价拍卖竞拍系统功能设计与实现九
java·spring boot·mysql·毕业设计·毕业源码·竞拍系统·竞拍项目
CQXXCL34 分钟前
MySQL-学习笔记
笔记·学习·mysql
Allen Bright40 分钟前
Redis主从架构
数据库·redis·架构
ZHOU西口43 分钟前
MySQL系列之远程管理（安全）
数据库·mysql·安全·tls·加密连接·require_secure
A_Tai23333331 小时前
MyBatis高级扩展
java·开发语言·mybatis
Karoku0661 小时前
【docker集群应用】Docker网络与资源控制
运维·数据库·docker·容器
扬子鳄0082 小时前
Spring Boot自动配置机制
java·数据库·spring boot
岁岁岁平安2 小时前
springboot实战(19)(条件分页查询、PageHelper、MYBATIS动态SQL、mapper映射配置文件、自定义类封装分页查询数据集)
java·spring boot·后端·mybatis·动态sql·pagehelper·条件分页查询
秋意钟2 小时前
sql漏洞
数据库·oracle
热门推荐
01(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明02玄机平台应急响应—webshell查杀03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04基于微信小程序的自习室预约系统的设计与实现05Docker 夺命连环 15 问06PyTorch机器学习实现液态神经网络0704 - matlab m_map地学绘图工具基础函数 - 设置网格08支付宝短视频平台创作分成激励项目09RAG 实践- Ollama+RagFlow 部署本地知识库10鸿蒙LiteOS+Hi3861智能家居网关设计：MQTT协议实现与代码示例