[MyBatisPlus]模拟多数据源环境及测试

Posted 2023-02-22 唐火

模拟多数据源环境

多数据源

适用于多种场景：纯粹多库、 读写分离、 一主多从、 混合模式等

目前我们就来模拟一个纯粹多库的一个场景，其他场景类似

场景说明：

我们创建两个库，分别为：mybatis_plus（以前的库不动）与mybatis_plus_1（新建），将mybatis_plus库的product表移动到mybatis_plus_1库，这样每个库一张表，通过一个测试用例分别获取用户数据与商品数据，如果获取到说明多库模拟成功

创建数据库及表

CREATE DATABASE `mybatis_plus_1` /*!40100 DEFAULT CHARACTER SET utf8mb4 */; use `mybatis_plus_1`; CREATE TABLE product ( id BIGINT(20) NOT NULL COMMENT '主键ID',
 name VARCHAR(30) NULL DEFAULT NULL COMMENT '商品名称', 
 price INT(11) DEFAULT 0 COMMENT '价格', 
 version INT(11) DEFAULT 0 COMMENT '乐观锁版本号', 
 PRIMARY KEY (id) );

添加测试数据

INSERT INTO product (id, NAME, price) VALUES (1, '外星人笔记本', 100);

删除mybatis_plus库product表

use mybatis_plus; 
DROP TABLE IF EXISTS product;

引入依赖

 <!--        mybatis plus 启动器-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.5.1</version>
        </dependency>

        <!--        lombok用于简化实体类开发-->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>


        <!--        mysql驱动-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
            <version>3.5.0</version>
        </dependency>

配置多数据源

spring:
  # 配置数据源信息
   datasource:
    dynamic:
      # 设置默认的数据源或者数据源组，默认值为master
      primary: master #设置默认的数据源或者数据源组,默认值即为master，如果读者只是单数据源只需要注释掉slave相关配置即可，这里为了方便演示master与slave保持相同
      # 严格匹配数据源，默认为false，true未匹配到指定数据源时抛出异常，false使用默认数据源
      strict: false
      datasource:
        master:
          url: jdbc:mysql://localhost:3306/mybatis_plus?serverTimezone=Hongkong&allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false  # serverTimezone=Hongkong 需要填上时区
          username: root
          password: password
          driverClassName:  com.mysql.cj.jdbc.Driver
        slave_1:
          url: jdbc:mysql://localhost:3306/mybatis_plus_1?serverTimezone=Hongkong&allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false  # serverTimezone=Hongkong 需要填上时区
          username: root
          password: password
          driverClassName: com.mysql.cj.jdbc.Driver

创建用户service

package com.xxxx.mybatisplus02.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.xxxx.mybatisplus02.pojo.User;

public interface UserService  extends IService<User> 

package com.xxxx.mybatisplus02.service.Impl;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xxxx.mybatisplus02.mapper.UserMapper;
import com.xxxx.mybatisplus02.pojo.User;
import com.xxxx.mybatisplus02.service.UserService;
import org.springframework.stereotype.Service;

@DS("master")
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService 

创建商品service

package com.xxxx.mybatisplus02.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.xxxx.mybatisplus02.pojo.Product;

public interface ProductService extends IService<Product> 

package com.xxxx.mybatisplus02.service.Impl;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xxxx.mybatisplus02.mapper.ProductMapper;
import com.xxxx.mybatisplus02.pojo.Product;
import com.xxxx.mybatisplus02.service.ProductService;
import org.springframework.stereotype.Service;

@DS("slave_1")
@Service
public class ProductServiceImpl extends ServiceImpl<ProductMapper, Product> implements ProductService 

测试

package com.xxxx.mybatisplus02;

import com.xxxx.mybatisplus02.service.ProductService;
import com.xxxx.mybatisplus02.service.UserService;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

@SpringBootTest
class Mybatisplus02ApplicationTests 


    @Autowired
    private UserService userService;

    @Autowired
    private ProductService productService;

    @Test
    public void test()
    
        System.out.println(userService.getById(1));
        System.out.println(productService.getById(1));
    

结果：

1. 都能顺利获取对象，则测试成功
2. 如果我们实现读写分离，将写操作方法加上主库数据源，读操作方法加上从库数据源，自动切换，是不是就能实现读写分离？

@DS 可以注解在方法上或类上，同时存在就近原则 方法上注解 优先于 类上注解。