MyBatis-Plus的通用枚举、代码生成器和多数据源

一、通用枚举

表中的有些字段值是固定的，例如性别（男或女），那么就可以使用MyBatis-Plus的通用枚举 来实现

1.数据库表中添加sex字段

2.创建通用枚举类型SexEnum

package com.qcby.mybatisPlus.enums;

import com.baomidou.mybatisplus.annotation.EnumValue;
import lombok.Getter;

@Getter
public enum SexEnum {
    MALE(1, "男"),
    FEMALE(0, "女");
    
    @EnumValue
    private Integer sex;
    private String sexName;
    
    SexEnum(Integer sex, String sexName) {
        this.sex = sex;
        this.sexName = sexName;
    }
}

3.User实体类

package com.qcby.mybatisPlus.model;

import com.baomidou.mybatisplus.annotation.*;
import com.qcby.mybatisPlus.enums.SexEnum;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

@Data
@ToString
@NoArgsConstructor
@AllArgsConstructor
//@TableName("t_user")
public class User {
    @TableId
    private Integer uid;
    private String name;
    private Integer age;
    private String email;
    @TableLogic
    private Integer isDeleted;
    private SexEnum sex;
}

4.配置扫描通用枚举

application.yml

# 配置MyBatis日志
mybatis-plus:
  configuration:
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
  global-config:
    db-config:
      # 配置mybatis_plus操作表的默认前缀
      table-prefix: t_
      #id-type: auto #配置MyBatis-Plus的主键策略
  mapper-locations: classpath:mapper/*.xml #对应mapper映射文件xml文件所在位置
  type-aliases-package: com.qcby.mybatisPlus.model #对应实体类所在位置
  # 配置通用扫描枚举
  type-enums-package: com.qcby.mybatisPlus.enums

5.测试类

/*
* 枚举
* */
@Test
public void testEnum(){
    User user=new User();
    user.setName("Enum");
    user.setAge(18);
    //设置性别信息为枚举项，会将@EnumValue注解所标识的属性值存储到数据库
    user.setSex(SexEnum.MALE);
    //INSERT INTO t_user ( uid, name, age, sex ) VALUES ( ?, ?, ?, ? )
    //Parameters: 432013313(Integer), Enum(String), 18(Integer), 1(Integer)
    userMapper.insert(user);
}

二、代码生成器

1.引入依赖

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.1</version>
</dependency>
<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.31</version>
</dependency>

2.快速生成

package com.qcby.mybatisPlus.util;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

import static com.baomidou.mybatisplus.generator.config.builder.GeneratorBuilder.packageConfig;

public class FastAutoGeneratorTest {
    public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql://127.0.0.1:3306/springboot_mybatisplus?characterEncoding=utf-8&useSSL=false", "root", "2020")
                .globalConfig(builder -> {
                    builder.author("qcby") // 设置作者，谁做的这个事
                            //.enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("D://mybatis_plus"); // 指定输出目录，这个目录要提前创建
                })
                .packageConfig(builder -> {
                    builder.parent("com.qcby") // 设置父包名
                            .moduleName("mybatisPlus") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D://mybatis_plus"));
                    // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("t_user") // 设置需要生成的表名
                            .addTablePrefix("t_", "c_"); // 设置过滤表前缀
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker 引擎模板，默认的是Velocity引擎模板
                .execute();
    }

}

3.运行

4.查看文件夹

可以看到mybatisplus逆向工程生成了所有的东西(实体类、持久层、表现层、业务层、mapper.xml文件)

三、多数据源

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

现在模拟的是一个纯粹多库的一个场景，其他场景类似

场景说明：

创建两个库，分别为： springboot_mybatisplus（以前的库不动）与springboot_mybatisplus1（新建）

在springboot_mybatisplus1库里创建product表，通过一个测试用例分别获取用户数据与商品数据，如果获取到说明多库模拟成功

1.创建数据库及表

创建数据库springboot_mybatisplus1和表product

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)
);

2.添加数据

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

3.引入依赖

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

4.配置多数据源

注释掉之前的数据库连接，添加新配置

spring:
  # 配置数据源信息
  datasource:
    dynamic:
      # 设置默认的数据源或者数据源组 ,默认值即为master
      primary: master
      # 严格匹配数据源 ,默认false.true未匹配到指定数据源时抛异常 ,false使用默认数据源
      strict: false
      datasource:
        master:
          url: jdbc:mysql://localhost:3306/springboot_mybatisplus?characterEncoding=utf-8&useSSL=false
          driver-class-name: com.mysql.jdbc.Driver
          username: root
          password: 2020
        slave_1:
          url: jdbc:mysql://localhost:3306/springboot_mybatisplus1?characterEncoding=utf-8&useSSL=false
          driver-class-name: com.mysql.jdbc.Driver
          username: root
          password: 2020
          
# 配置MyBatis日志
mybatis-plus:
  configuration:
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
#  global-config:  
#    db-config:
      # 配置mybatis_plus操作表的默认前缀
      # table-prefix: t_ 因为这里配置了前缀，但是新库里的表是product,没有前缀t_，所以这里要注释掉，User实体类里自己加注解@TableNmae
      #id-type: auto #配置MyBatis-Plus的主键策略
  mapper-locations: classpath:mapper/*.xml #对应mapper映射文件xml文件所在位置
  type-aliases-package: com.qcby.mybatisPlus.model #对应实体类所在位置
  # 配置通用扫描枚举
  type-enums-package: com.qcby.mybatisPlus.enums

5.创建用户接口及实现类

public interface UserService extends IService<User> {
}

@DS("master") //指定所操作的数据源
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
}

6.创建商品接口及实现类

public interface ProductService extends IService<Product> {
}

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

7.测试类

package com.qcby.mybatisPlus;

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

@SpringBootTest
public class DataTest {

    @Autowired
    private UserService userService;
    @Autowired
    private ProductService productService;

    @Test
    public void testDynamicDataSource(){
        System.out.println(userService.getById(1L));
        System.out.println(productService.getById(1L));
        /*
        * SELECT uid,name,age,email,is_deleted,sex FROM t_user WHERE uid=? AND is_deleted=0
        * Parameters: 1(Long)
        * User(uid=1, name=qcby, age=18, email=qcby@qq.com, isDeleted=0, sex=MALE)
        * 
        * SELECT id,name,price,version FROM product WHERE id=?
        * Parameters: 1(Long)
        * Product(id=1, name=外星人笔记本 , price=100, version=0)
        * */
    }
}

都能顺利获取对象，测试成功