springboot實戰代碼系列之【集成mybatis-plus和druid實現多數據源切換】

爲什麼要寫本文？

官方文檔給的不夠詳細，下面是一個能即讀即用的版本
主要也方便自己以後參考代碼，不再重複查找

本文使用的springboot版本：2.2.6.RELEASE

添加依賴

查看一下最新版本的插件：https://mvnrepository.com/artifact/com.baomidou/mybatis-plus-boot-starter

druid插件的最新版也要看看：https://mvnrepository.com/artifact/com.alibaba/druid-spring-boot-starter
不過這裏，還是建議看mybatis plus所支持的druid版本。

	<dependency>
		<groupId>org.projectlombok</groupId>
		<artifactId>lombok</artifactId>
		<optional>true</optional>
	</dependency>

	<!--@DS註解需要的依賴-->
	<dependency>
		<groupId>com.baomidou</groupId>
		<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
		<version>3.0.0</version>
	</dependency>
	<dependency>
		<groupId>com.baomidou</groupId>
		<artifactId>mybatis-plus-boot-starter</artifactId>
		<version>3.3.1</version>
	</dependency>
	<dependency>
		<groupId>com.alibaba</groupId>
		<artifactId>druid-spring-boot-starter</artifactId>
		<version>1.1.21</version>
	</dependency>
	<dependency>
		<groupId>mysql</groupId>
		<artifactId>mysql-connector-java</artifactId>
		<scope>runtime</scope>
	</dependency>

配置文件

首先要移除Druid自動配置類：

@SpringBootApplication(exclude = DruidDataSourceAutoConfigure.class)

某些springBoot的版本上面可能無法排除可用以下方式排除。

spring:
  autoconfigure:
    exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure

原因：DruidDataSourceAutoConfigure會注入一個DataSourceWrapper，其會在原生的spring.datasource下找url,username,password等。而我們動態數據源的配置路徑是變化的。

然後下面是非常簡單的一個druid配置（附錄有完整配置）：

spring:
  datasource:
    dynamic:
	  # 默認源
      primary: druid-admin
      druid:
        #初始化大小,最小,最大
        initial-size: 5
        min-idle: 5
        max-active: 20
      datasource:
        # 數據源1
        druid-admin:
          url: jdbc:mysql://IP:3306/db_admin?allowMultiQueries=true&useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
          username: xxx
          password: xxx
          driver-class-name: com.mysql.cj.jdbc.Driver
        # 數據源2
        druid-web:
          url: jdbc:mysql://IP:3306/db_web?allowMultiQueries=true&useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
          username: xxx
          password: xxx
          driver-class-name: com.mysql.cj.jdbc.Driver

代碼

因爲要用分頁查詢測試，所以加上分頁插件

    /**
     * 分頁插件,解決分頁時total=0問題
     */
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        return new PaginationInterceptor();
    }

準備實體類

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;

@Data
@TableName("t_user")
public class User {
    @TableId(type = IdType.AUTO)
    private Integer id;
    @TableField(value = "user_name")
    private String username;
    private String password;
    private String email;
}

@Data
@TableName("role")
public class Role {
    @TableId(type = IdType.AUTO)
    private Integer id;
    private String name;
}

準備mapper

只需要繼承BaseMapper得到常用的CRUD接口方法即可，再加上@Mapper註解標識

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.jimo.mybatisplusdruiddemo.model.User;
import org.apache.ibatis.annotations.Mapper;

@Mapper
public interface UserMapper extends BaseMapper<User> {
}

@Mapper
public interface RoleMapper extends BaseMapper<Role> {
}

寫service層

只需繼承ServiceImpl就好。

這裏的@DS註解就是選擇數據源的，名字與配置裏的對應

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jimo.mybatisplusdruiddemo.mapper.UserMapper;
import com.jimo.mybatisplusdruiddemo.model.User;
import org.springframework.stereotype.Service;

@Service
@DS("druid-web")
public class UserService extends ServiceImpl<UserMapper, User> {
}

/**
* 因爲默認就是druid-admin庫，所以，可以省略，加上也沒關係
*/
@Service
//@DS("druid-admin")
public class RoleService extends ServiceImpl<RoleMapper, Role> {
}

測試

測試類非常簡單，執行一個分頁查詢：

@SpringBootTest
class MybatisPlusDruidDemoApplicationTests {

    @Autowired
    private UserService userService;
    @Autowired
    private RoleService roleService;

    @Test
    void testUserService() {
        final int size = 10;
        final Page<User> page = userService.page(new Page<>(1, size));
        assertEquals(size, page.getRecords().size());
        assertTrue(page.getTotal() >= size);
    }

    @Test
    void testRoleService() {
        final int size = 1;
        final Page<Role> page = roleService.page(new Page<>(1, size));
        assertEquals(size, page.getRecords().size());
        assertTrue(page.getTotal() >= size);
    }
}

可以看到日誌裏數據庫的初始化和銷燬：

2020-04-01 10:34:20.661  INFO 15600 --- [           main] j.m.MybatisPlusDruidDemoApplicationTests : No active profile set, falling back to default profiles: default
2020-04-01 10:34:22.805  INFO 15600 --- [           main] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1,druid-admin} inited
2020-04-01 10:34:23.118  INFO 15600 --- [           main] com.alibaba.druid.pool.DruidDataSource   : {dataSource-2,druid-web} inited
2020-04-01 10:34:23.118  INFO 15600 --- [           main] c.b.d.d.DynamicRoutingDataSource         : dynamic-datasource - load a datasource named [druid-admin] success
2020-04-01 10:34:23.119  INFO 15600 --- [           main] c.b.d.d.DynamicRoutingDataSource         : dynamic-datasource - load a datasource named [druid-web] success
2020-04-01 10:34:23.119  INFO 15600 --- [           main] c.b.d.d.DynamicRoutingDataSource         : dynamic-datasource initial loaded [2] datasource,primary datasource named [druid-admin]
 _ _   |_  _ _|_. ___ _ |    _ 
| | |\/|_)(_| | |_\  |_)||_|_\ 
     /               |         
                        3.3.1 
2020-04-01 10:34:24.379  INFO 15600 --- [           main] o.s.s.concurrent.ThreadPoolTaskExecutor  : Initializing ExecutorService 'applicationTaskExecutor'
2020-04-01 10:34:24.888  INFO 15600 --- [           main] j.m.MybatisPlusDruidDemoApplicationTests : Started MybatisPlusDruidDemoApplicationTests in 4.695 seconds (JVM running for 6.375)

...

2020-04-01 10:34:56.112  INFO 15600 --- [extShutdownHook] o.s.s.concurrent.ThreadPoolTaskExecutor  : Shutting down ExecutorService 'applicationTaskExecutor'
2020-04-01 10:34:56.113  INFO 15600 --- [extShutdownHook] c.b.d.d.DynamicRoutingDataSource         : dynamic-datasource start closing ....
2020-04-01 10:34:56.116  INFO 15600 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1} closing ...
2020-04-01 10:34:56.126  INFO 15600 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1} closed
2020-04-01 10:34:56.126  INFO 15600 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-2} closing ...
2020-04-01 10:34:56.130  INFO 15600 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-2} closed
2020-04-01 10:34:56.130  INFO 15600 --- [extShutdownHook] c.b.d.d.DynamicRoutingDataSource         : dynamic-datasource all closed success,bye

總結

重點強調

@DS("druid-web")註解用來選擇數據源，是可以用在mapper層的

@Mapper
@DS("druid-web")
public interface UserMapper extends BaseMapper<User> {
}

就像有時候偷懶，不想寫service層，也可以在mapper就實現多數據源切換：

    @Autowired
    private UserMapper userMapper;

    @Test
    void testUserMapper() {
        final int size = 10;
        final Page<User> page = userMapper.selectPage(new Page<>(1, size), null);
        assertEquals(size, page.getRecords().size());
        assertTrue(page.getTotal() >= size);
    }

附錄

druid的完整配置：

spring:
  datasource:
    dynamic:
      primary: druid-admin
      druid:
        #初始化大小,最小,最大
        initial-size: 5
        min-idle: 5
        max-active: 20
        #配置獲取連接等待超時的時間
        max-wait: 60000
        #配置間隔多久才進行一次檢測,檢測需要關閉的空閒連接,單位是毫秒
        time-between-eviction-runs-millis: 60000
        #配置一個連接在池中最小生存的時間,單位是毫秒
        min-evictable-idle-time-millis: 300000
        validation-query: SELECT 1 FROM DUAL
        test-while-idle: true
        test-on-borrow: false
        test-on-return: false
        #打開PSCache,並且指定每個連接上PSCache的大小
        pool-prepared-statements: true
        max-pool-prepared-statement-per-connection-size: 20
        #配置監控統計攔截的filters,去掉後監控界面sql無法統計,'wall'用於防火牆
        filters: stat,wall # 注意這個值和druid原生不一致，默認啓動了stat,wall
        filter:
          log4j:
            enabled: true
          wall:
            config:
              # 允許多個語句一起執行
              multi-statement-allow: true
            enabled: true
            db-type: mysql
          stat:
            enabled: true
            db-type: mysql
      datasource:
        # 數據源1
        druid-admin:
          url: jdbc:mysql://IP:3306/db_admin?allowMultiQueries=true&useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
          username: xxx
          password: xxx
          driver-class-name: com.mysql.cj.jdbc.Driver
        # 數據源2
        druid-web:
          url: jdbc:mysql://IP:3306/db_web?allowMultiQueries=true&useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
          username: xxx
          password: xxx
          driver-class-name: com.mysql.cj.jdbc.Driver

springboot實戰代碼系列之【集成mybatis-plus和druid實現多數據源切換】

添加依賴

配置文件

代碼

準備實體類

準備mapper

寫service層

測試

總結

重點強調

附錄

參考

論戰被黑客入侵的MQ

ActiveMQ問題與調優

spring應用集成網關代理

JanusGraph快速入門

作爲一名帝都的程序員，我爲什麼去撿垃圾？

https://yachay.unat.edu.pe/blog/index.php?comment_area=format_blog&comment_component=blog&comment_co

linux以太網驅動總結