首页 前端知识 MyBatisPlus详解(四)通用枚举、JSON类型处理器、配置加密、自动分页插件与工具类

MyBatisPlus详解(四)通用枚举、JSON类型处理器、配置加密、自动分页插件与工具类

2024-05-10 08:05:51 前端知识 前端哥 304 894 我要收藏

文章目录

  • 前言
  • 3 扩展功能
    • 3.3 通用枚举
      • 3.3.1 使用枚举类
      • 3.3.2 功能测试
    • 3.4 JSON类型处理器
      • 3.4.1 使用JSON类型处理器
      • 3.4.2 功能测试
    • 3.5 配置加密
      • 3.5.1 生成密钥
      • 3.5.2 修改配置
      • 3.5.3 功能测试
  • 4 插件功能
    • 4.1 自动分页插件
      • 4.1.1 配置分页插件
      • 4.1.2 分页API
    • 4.2 通用分页实体
      • 4.2.1 创建实体
      • 4.2.2 开发接口
      • 4.2.3 封装PageQuery工具方法
      • 4.2.4 封装PageDTO工具方法

前言

MyBatisPlus详解系列文章:

MyBatisPlus详解(一)项目搭建、@TableName、@TableId、@TableField注解与常见配置
MyBatisPlus详解(二)条件构造器Wrapper、自定义SQL、Service接口
MyBatisPlus详解(三)lambdaQuery、lambdaUpdate、批量新增、代码生成、Db静态工具、逻辑删除

3 扩展功能

3.3 通用枚举

在User实体类中有一个状态字段status

对于这样的码表字段,一般会定义一个枚举类,做业务判断的时候直接基于枚举类进行比较。但是该字段在数据库采用的是int类型,对应的实体类中是Integer类型,因此业务操作时必须手动把枚举类与Integer进行转换,相对麻烦。

为此,MybatisPlus提供了一个处理枚举类的类型转换器,可以对枚举类型与数据库类型进行自动转换。

3.3.1 使用枚举类

  • 1)定义一个用户状态的枚举类UserStatus
// com.star.learning.enums.UserStatus

@Getter
public enum  UserStatus {

    NORMAL(1, "正常"),
    FREEZE(2, "冻结")
    ;

    private final int value;
    private final String desc;

    UserStatus(int value, String desc) {
        this.value = value;
        this.desc = desc;
    }
}
  • 2)将User类中的status字段改为UserStatus类型
// com.star.learning.pojo.User

/**
 * 使用状态(1正常 2冻结)
 */
//private Integer status;
private UserStatus status;
  • 3)使用@EnumValue注解标记枚举属性:

要让MybatisPlus处理枚举类与数据库类型自动转换,就必须告诉MybatisPlus,枚举中的哪个字段的值作为数据库值。MybatisPlus提供了@EnumValue注解来标记枚举属性:

// com.star.learning.enums.UserStatus

@EnumValue
private final int value;

// ...
  • 4)配置枚举处理器

在application.yaml文件中添加配置:

# src\main\resources\application.yaml

mybatis-plus:
  configuration:
    # 默认枚举类处理器(从3.5.2开始无需配置)
    default-enum-type-handler: com.baomidou.mybatisplus.core.handlers.MybatisEnumTypeHandler

3.3.2 功能测试

执行前面几节编写好的/user/{id}接口,根据id查询用户信息,可以发现查询出的User类的status字段是枚举类型:

此时返回前端的信息是:

可见,status字段JSON序列化后的值为NORMAL,这显然是不符合要求的。

为此,MybatisPlus支持通过@JsonValue注解,用于标记JSON序列化时展示的字段:

// com.star.learning.enums.UserStatus

@EnumValue
private final int value;
@JsonValue
private final String desc;

再次执行以上接口,返回前端的信息是:

3.4 JSON类型处理器

在数据库的t_user表中,有一个类型为JSON的字段info(从MySQL5.7开始支持JSON类型),保存的数据是JSON字符串:

在其对应的实体类User中,该字段是String类型的:

这样设计时,查询到的info字段就是一个JSON字符串,而要读取其中的属性,还需要将JSON字符串转换为对象。而写入数据库时,需要将对象转换为JSON字符串,较为繁琐。

为此,MybatisPlus提供了很多特殊类型字段的类型处理器,解决特殊字段类型与数据库类型转换的问题,例如处理JSON问题就可以使用JacksonTypeHandler处理器。

3.4.1 使用JSON类型处理器

  • 1)定义一个实体类UserInfo,对应info字段保存的信息:
// com.star.learning.pojo.UserInfo

@Data
public class UserInfo {
    private Integer age;
    private String intro;
    private String gender;
}
  • 2)将User类的info字段修改为UserInfo类型,并使用@TableField注解声明该字段的类型处理器为JacksonTypeHandler,同时在实体类的@TableName注解上添加autoResultMap = true属性:
// com.star.learning.pojo.User

@Data
@TableName(value = "t_user", autoResultMap = true)
public class User {
    
    // ...
    
    @TableField(typeHandler = JacksonTypeHandler.class)
    private UserInfo info;
}

3.4.2 功能测试

调用/user/{id}接口,根据id查询用户信息,可以发现查询出的User类的info字段是UserInfo类型,且正确封装:

3.5 配置加密

目前在application.yaml配置文件中,很多参数都是明文的,例如数据库的用户名和密码。如果开发人员发生流动,则很容易导致敏感信息的泄露。为此,MybatisPlus支持配置文件的加密和解密功能。

3.5.1 生成密钥

利用AES工具生成一个随机秘钥,然后使用该密钥对数据库用户名、密码加密:

@Test
public void testAES() {
    // 生成 16 位随机 AES 密钥
    String randomKey = AES.generateRandomKey();
    System.out.println("randomKey = " + randomKey);
    // 利用密钥对用户名加密
    String username = AES.encrypt("root", randomKey);
    System.out.println("username = " + username);
    // 利用密钥对用户名加密
    String password = AES.encrypt("123456", randomKey);
    System.out.println("password = " + password);
}

执行以上单元测试,控制台打印信息如下:

randomKey = cb3d37297a857ac9
username = nvC21NLjkZ8SToc+FxM7Jw==
password = p898OXS2GwKDdJI9P+GlGA==

3.5.2 修改配置

修改application.yaml文件,把jdbc的用户名、密码修改为刚刚加密生成的密文:

spring:
  datasource:
    url: jdbc:mysql://127.0.0.1:3306/mybatis_plus_db?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&serverTimezone=Asia/Shanghai&rewriteBatchedStatements=true
    driver-class-name: com.mysql.cj.jdbc.Driver
    # 密文要以 mpw:开头
    username: mpw:nvC21NLjkZ8SToc+FxM7Jw==
    password: mpw:p898OXS2GwKDdJI9P+GlGA==

3.5.3 功能测试

启动项目后,调用/user/{id}接口,发现报错:

java.sql.SQLNonTransientConnectionException: Could not create connection to database server. Attempted reconnect 3 times. Giving up.

报错是因为在启动项目的时,需要把刚才生成的秘钥添加到启动参数中(以IDEA配置为例):

重新启动项目,再次调用/user/{id}接口,发现可以正常访问数据库了。

4 插件功能

MybatisPlus提供了很多插件,进一步拓展其功能。目前已有的插件包括:

  • PaginationInnerInterceptor:自动分页
  • TenantLineInnerInterceptor:多租户
  • DynamicTableNameInnerInterceptor:动态表名
  • OptimisticLockerInnerInterceptor:乐观锁
  • IllegalSQLInnerInterceptor:SQL性能规范
  • BlockAttackInnerInterceptor:防止全表更新与删除

使用最多的是自动分页插件。

4.1 自动分页插件

4.1.1 配置分页插件

在未配置分页插件的情况下,MybatisPlus是不支持分页功能的,IService接口和BaseMapper类中的分页方法都无法正常使用。

要配置分页插件,需要新建一个配置类:

// com.star.learning.config.MyBatisConfig

@Configuration
public class MyBatisConfig {

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        // 初始化核心插件
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        // 添加分页插件
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        return interceptor;
    }
    
}

4.1.2 分页API

分页API的基本使用如下:

@Test
public void testPageQuery() {
    // 1.分页查询,new Page()的两个参数分别是:页码、每页大小
    Page<User> page = new Page<>(2, 2);
    page = userService.page(page);
    // 2.总条数
    System.out.println("total = " + page.getTotal());
    // 3.总页数
    System.out.println("pages = " + page.getPages());
    // 4.数据
    List<User> records = page.getRecords();
    records.forEach(System.out::println);
}

执行以上单元测试,控制台打印信息如下:

// 查询总条数
==>  Preparing: SELECT COUNT(*) AS total FROM t_user
==> Parameters: 
<==      Total: 1
// 使用LIMIT进行分页查询
==>  Preparing: SELECT id,username,password,phone,info,status,balance,create_time,update_time FROM t_user LIMIT ?,?
==> Parameters: 2(Long), 2(Long)
<==      Total: 2
// 查询结果
total = 4
pages = 2
User(id=3, username=Hope, password=123, phone=13900112222, info=UserInfo(age=25, intro=上进青年, gender=male), status=NORMAL, balance=100000, createTime=2024-04-23T16:08:49, updateTime=2024-04-23T16:08:49, addressList=null)
User(id=4, username=Thomas, password=123, phone=17701265258, info=UserInfo(age=29, intro=伏地魔, gender=male), status=NORMAL, balance=800, createTime=2024-04-23T16:08:49, updateTime=2024-04-23T16:08:49, addressList=null)

分页的核心是Page类,除了可以设置当前页和每页大小,还支持排序参数,例如将以上单元测试修改如下:

// 1.分页查询,new Page()的两个参数分别是:页码、每页大小
Page<User> page = new Page<>(2, 2);
// 支持排序参数,第二个参数的true表示升序,false表示降序
page.addOrder(new OrderItem("balance", true));
// 支持多个排序参数
page.addOrder(new OrderItem("id", true));

// ...

再次执行以上单元测试,控制台打印信息如下:

==>  Preparing: SELECT id, username, password, phone, info, status, balance, create_time, update_time FROM t_user ORDER BY balance ASC, id ASC LIMIT ?,?
==> Parameters: 2(Long), 2(Long)
<==      Total: 2

可见,ORDER BY balance ASC, id ASC已经被添加到了SQL语句中。

4.2 通用分页实体

下面要实现分页查询用户信息的接口:

参数说明
请求方式GET
请求路径/user/page
请求参数{"pageNo": 1, "pageSize": 5, "sortBy": "balance", "isAsc": false, "name": "o", "status": 1}
返回值{"total": 100, "pages": 10, "list": [{"id": 1, "username": "user_1", "info": ...}]}
特殊说明如果排序字段为空,则默认按照更新时间排序

以上接口文档主要涉及3个实体类:

  • UserQuery:分页查询条件的实体,包含分页、排序参数、查询条件
  • PageDTO:分页结果实体,包含总条数、总页数、当前页数据
  • User:用户信息实体

4.2.1 创建实体

  • 1)创建一个PageQuery实体类,定义分页、排序参数:
// com.star.learning.pojo.PageQuery

@Data
public class PageQuery {

    /** 页码 */
    private Long pageNo;
    /** 每页大小 */
    private Long pageSize;
    /** 排序字段 */
    private String sortBy;
    /** 是否升序 */
    private Boolean isAsc;

}
  • 2)修改UserQuery实体类,继承PageQuery类,这样UserQuery就即包含查询条件,又包含分页信息:
// com.star.learning.pojo.UserQuery

@Data
public class UserQuery extends PageQuery {...}
  • 3)创建分页结果实体PageDTO类:
// com.star.learning.pojo.PageDTO

@Data
@AllArgsConstructor
public class PageDTO<T> {
    /** 总条数 */
    private Long total;
    /** 总页数 */
    private Long pages;
    /** 数据集合 */
    private List<T> list;
}

4.2.2 开发接口

  • 1)在UserController中定义分页查询用户的接口:
// com.star.learning.controller.UserController

@GetMapping("/page")
public PageDTO<User> queryUsersPage(UserQuery query){
    return userService.queryUsersPage(query);
}
  • 2)在IUserService接口中创建queryUsersPage方法,在UserServiceImpl中实现该方法:
// com.star.learning.service.IUserService

PageDTO<User> queryUsersPage(UserQuery userQuery);
// com.star.learning.service.impl.UserServiceImpl

@Override
public PageDTO<User> queryUsersPage(UserQuery userQuery) {
    // 1.构建条件
    // 1.1.分页条件
    Page<User> page = new Page<>(userQuery.getPageNo(), userQuery.getPageSize());
    // 1.2.排序条件
    if(userQuery.getSortBy() != null) {
        page.addOrder(new OrderItem(userQuery.getSortBy(), userQuery.getIsAsc()));
    } else {
        // 默认按照更新时间降序排列
        page.addOrder(new OrderItem("update_time", false));
    }
    // 2.查询数据
    lambdaQuery()
        .like(userQuery.getUsername() != null, User::getUsername, userQuery.getUsername())
        .eq(userQuery.getStatus() != null, User::getStatus, userQuery.getStatus())
        .ge(userQuery.getMinBalance() != null, User::getBalance, userQuery.getMinBalance())
        .le(userQuery.getMaxBalance() != null, User::getBalance, userQuery.getMaxBalance())
        .page(page);
    // 3.数据处理
    List<User> userList = page.getRecords();
    if (userList == null || userList.isEmpty()) {
        // 无数据,返回空结果
        return new PageDTO<>(page.getTotal(), page.getPages(), Collections.emptyList());
    }
    // 有数据,返回
    return new PageDTO<>(page.getTotal(), page.getPages(), userList);
}
  • 3)启动项目,调用接口进行测试

控制台打印信息如下:

==>  Preparing: SELECT COUNT(*) AS total FROM t_user WHERE (username LIKE ?)
==> Parameters: %o%(String)
<==      Total: 1
==>  Preparing: SELECT id, username, password, phone, info, status, balance, create_time, update_time FROM t_user WHERE (username LIKE ?) ORDER BY balance DESC LIMIT ?
==> Parameters: %o%(String), 2(Long)
<==      Total: 2

可见,SQL语句实现了分页、排序和条件查询。

4.2.3 封装PageQuery工具方法

在上述代码中,从PageQuery类到MyBatisPlus的Page类之间的转换逻辑是这样的:

// com.star.learning.service.impl.UserServiceImpl#queryUsersPage()

// 1.构建条件
// 1.1.分页条件
Page<User> page = new Page<>(userQuery.getPageNo(), userQuery.getPageSize());
// 1.2.排序条件
if(userQuery.getSortBy() != null) {
    page.addOrder(new OrderItem(userQuery.getSortBy(), userQuery.getIsAsc()));
} else {
    // 默认按照更新时间降序排列
    page.addOrder(new OrderItem("update_time", false));
}

这段代码还是比较麻烦的,并且每一个需要分页查询的Service方法都需要重写这一段转换逻辑。

为此,我们应该在PageQuery类中定义一个工具方法,完成PageQuery类到Page类之间的转换:

// com.star.learning.pojo.PageQuery

@Data
public class PageQuery {

    /** 页码 */
    private Long pageNo;
    /** 每页大小 */
    private Long pageSize;
    /** 排序字段 */
    private String sortBy;
    /** 是否升序 */
    private Boolean isAsc;

    public <T> Page<T> toMpPage(OrderItem... orders){
        // 1.分页条件
        Page<T> p = Page.of(pageNo, pageSize);
        // 2.排序条件
        // 2.1.先看前端有没有传排序字段
        if (sortBy != null) {
            p.addOrder(new OrderItem(sortBy, isAsc));
            return p;
        }
        // 2.2.再看有没有手动指定排序字段
        if(orders != null){
            p.addOrder(orders);
        }
        return p;
    }

    /**
     * 指定排序字段和升降序
     * @param defaultSortBy 排序字段
     * @param isAsc 是否升序
     */
    public <T> Page<T> toMpPage(String defaultSortBy, boolean isAsc){
        return this.toMpPage(new OrderItem(defaultSortBy, isAsc));
    }

    /**
     * 默认create_time降序
     */
    public <T> Page<T> toMpPageDefaultSortByCreateTimeDesc() {
        return toMpPage("create_time", false);
    }

    /**
     * 默认update_time降序
     */
    public <T> Page<T> toMpPageDefaultSortByUpdateTimeDesc() {
        return toMpPage("update_time", false);
    }

}

接下来改造UserServiceImpl类的queryUsersPage()方法:

// com.star.learning.service.impl.UserServiceImpl#queryUsersPage()

// 1.构建条件:使用工具方法获得Page对象
Page<User> page = userQuery.toMpPageDefaultSortByUpdateTimeDesc();

// 2.查询数据......

4.2.4 封装PageDTO工具方法

在上述代码中,查询出结果后,从Page类到PageDTO类之间的转换逻辑是这样的:

// com.star.learning.service.impl.UserServiceImpl#queryUsersPage()

// ...
// 3.数据处理
List<User> userList = page.getRecords();
if (userList == null || userList.isEmpty()) {
    // 无数据,返回空结果
    return new PageDTO<>(page.getTotal(), page.getPages(), Collections.emptyList());
}
// 有数据,返回
return new PageDTO<>(page.getTotal(), page.getPages(), userList);

对于不同的Service方法,这一段代码逻辑应该是差不多的,因此可以把它封装成工具方法,在Service方法中直接调用:

// com.star.learning.pojo.PageDTO

@Data
@AllArgsConstructor
public class PageDTO<T> {
    /** 总条数 */
    private Long total;
    /** 总页数 */
    private Long pages;
    /** 数据集合 */
    private List<T> list;

    /**
     * 返回空分页结果
     */
    public static <P> PageDTO<P> empty(Page<P> p){
        return new PageDTO<>(p.getTotal(), p.getPages(), Collections.emptyList());
    }

    /**
     * 将MybatisPlus分页结果转为PO分页结果
     * @return VO的分页对象
     */
    public static <P> PageDTO<P> of(Page<P> p) {
        // 1.非空校验
        List<P> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            // 无数据,返回空结果
            return empty(p);
        }
        // 2.数据返回
        return new PageDTO<>(p.getTotal(), p.getPages(), records);
    }
}

接下来改造UserServiceImpl类的queryUsersPage()方法,最终得到该方法的最简化版本:

// com.star.learning.service.impl.UserServiceImpl

@Override
    public PageDTO<User> queryUsersPage(UserQuery userQuery) {
    // 1.构建条件:使用工具方法获得Page对象
    Page<User> page = userQuery.toMpPageDefaultSortByUpdateTimeDesc();
    // 2.查询数据
    lambdaQuery()
            .like(userQuery.getUsername() != null, User::getUsername, userQuery.getUsername())
            .eq(userQuery.getStatus() != null, User::getStatus, userQuery.getStatus())
            .ge(userQuery.getMinBalance() != null, User::getBalance, userQuery.getMinBalance())
            .le(userQuery.getMaxBalance() != null, User::getBalance, userQuery.getMaxBalance())
            .page(page);
    // 3.数据处理,使用工具方法返回数据
    return PageDTO.of(page);
}

本节完,更多内容请查阅分类专栏:MyBatisPlus详解

本文涉及代码下载地址:https://gitee.com/weidag/mybatis_plus_learning.git

感兴趣的读者还可以查阅我的另外几个专栏:

  • SpringBoot源码解读与原理分析(已完结)
  • MyBatis3源码深度解析(已完结)
  • Redis从入门到精通(已完结)
  • 再探Java为面试赋能(持续更新中…)
转载请注明出处或者链接地址:https://www.qianduange.cn//article/7920.html
评论
会员中心 联系我 留言建议 回顶部
复制成功!