mybatis-plus enum实现枚举类型自动转换

 更新时间:2024年07月25日 09:59:41   作者:xy294636185  
本文主要介绍了mybatis-plus enum实现枚举类型自动转换,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

mybatis-plus实现了对“实体类指定了枚举类型,想查询时返回的是枚举值而非value值”,“插入数据时,实体赋值的是枚举类型,想入库时插入对应的value值”,“不想写其他的handler处理程序,希望能够自动处理”。

mybatis-plus对于上述的诉求都可以满足,简单的处理方案是:

* 1、实现 IEnum of T
* 2、注解 @EnumValue,不用实现 IEnum of T

具体的官方文档为 https://mp.baomidou.com/guide/enum.html

定义一个简单实体

先定义一个示例实体类,我们在实体Demo中用到了DemoStatusEnum

/**
 * 实体类
 */
@Data
@TableName("demo")
public class Demo extends Model<Demo> {
    @TableField("status")
    private DemoStatusEnum status;    
}

DemoStatusEnum枚举定义

我们采用了官方提到的两种方式的第一种:即实现IEnum<T>

/**
 * 支持枚举值的两种方式
 * 1、实现 IEnum of T
 * 2、注解 @EnumValue,不用实现 IEnum of T
 */
@Getter
public enum DemoStatusEnum implements IEnum<Integer> {
    DEFAULT(0, "default"),
    NORMAL(1, "normal"),
    LOCKED(2, "locked");

    DemoStatusEnum(Integer value, String desc) {
        this.value = value;
        this.desc = desc;
    }

    private Integer value;
    private String desc;

    @Override
    public Integer getValue() {
        return this.value;
    }
}

 如果不想继承IEnum<T>,可以通过第二种方式,就是增加一个注解的方式@EnumValue

    @EnumValue
    private Integer value;
    @JsonValue
    private String desc;

自动转换实现:

配置了Enums枚举,实体中设置了枚举类型,那么mybatis-plus如何转换的呢?重点是看这里

mybatis-plus:
  global-config:
    db-config:
      logic-not-delete-value: 0  #逻辑未删除值为数据库主键
      logic-delete-value: id #逻辑删除值是个d
      # logic-delete-value: "now()" #逻辑删除值是个db获取时间的函数
      # logic-not-delete-value: "null"  #逻辑未删除值为字符串 "null"
      # logic-not-delete-value: "1753-01-01 00:00:00"
  configuration:
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
    # 如果是放在src/main/java目录下 classpath:/com/*/*/mapper/*Mapper.xml
    # 如果是放在resource目录 classpath:/mapper/**.xml
    default-enum-type-handler: org.apache.ibatis.type.EnumOrdinalTypeHandler
    # 如果是放在src/main/java目录下 classpath:/com/*/*/mapper/*Mapper.xml
    # 如果是放在resource目录 classpath:/mapper/**.xml
    # 支持统配符 * 或者 ; 分割
    typeEnumsPackage: com.fii.gxback.eam.domain.enums

通过上述的配置后,就可以了。那么,接下来我们重点分析一下type-enums-package

按照官方文档说明:

枚举类 扫描路径,如果配置了该属性,会将路径下的枚举类进行注入,让实体类字段能够简单快捷的使用枚举属性

通过扒源码发现,其实是帮我们实现了对应的 typehandler(com.baomidou.mybatisplus.core.handlers.MybatisEnumTypeHandler)

其中,isMpEnums中,判定整合两种方式的判定,即 IEnum.classEnumValue.class

/**
 * 判断是否为MP枚举处理
 *
 * @param clazz class
 * @return 是否为MP枚举处理
 * @since 3.3.1
 */
public static boolean isMpEnums(Class<?> clazz) {
    return clazz != null && clazz.isEnum() && (IEnum.class.isAssignableFrom(clazz) || findEnumValueFieldName(clazz).isPresent());
}

/**
     * 查找标记标记EnumValue字段
     *
     * @param clazz class
     * @return EnumValue字段
     * @since 3.3.1
     */
public static Optional<String> findEnumValueFieldName(Class<?> clazz) {
    if (clazz != null && clazz.isEnum()) {
        String className = clazz.getName();
        return Optional.ofNullable(CollectionUtils.computeIfAbsent(TABLE_METHOD_OF_ENUM_TYPES, className, key -> {
            Optional<Field> optional = Arrays.stream(clazz.getDeclaredFields())
                // 过滤包含注解@EnumValue的字段
                .filter(field -> field.isAnnotationPresent(EnumValue.class))
                .findFirst();
            return optional.map(Field::getName).orElse(null);
        }));
    }
    return Optional.empty();
}

然后,在com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean#buildSqlSessionFactory中,

// 取得类型转换注册器
TypeHandlerRegistry typeHandlerRegistry = targetConfiguration.getTypeHandlerRegistry();
classes.stream()
    .filter(Class::isEnum)
    .filter(MybatisEnumTypeHandler::isMpEnums)
    .forEach(cls -> typeHandlerRegistry.register(cls, MybatisEnumTypeHandler.class));

验证逻辑

controller:

@RequestMapping("/{id}")
public Demo get(@PathVariable("id") Long id) {
    return demoService.getById(id);
}

输出:

"status": "DEFAULT"

可以看到,status最终输出的不是魔术数字0,而是枚举值default。

其实,跟一下代码,可以发现,跟我们自己手写一个typeHandler没区别,这里最终获取转换值时,调用了com.baomidou.mybatisplus.extension.handlers.MybatisEnumTypeHandler#getNullableResult(java.sql.ResultSet, int),此时已经拿到了枚举值的具体类是什么了。接下来就是调用valueOf去获取对应的枚举值即可。

附源码:

package org.apache.ibatis.type;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class EnumOrdinalTypeHandler<E extends Enum<E>> extends BaseTypeHandler<E> {
    private final Class<E> type;
    private final E[] enums;

    public EnumOrdinalTypeHandler(Class<E> type) {
        if (type == null) {
            throw new IllegalArgumentException("Type argument cannot be null");
        } else {
            this.type = type;
            this.enums = (Enum[])type.getEnumConstants();
            if (this.enums == null) {
                throw new IllegalArgumentException(type.getSimpleName() + " does not represent an enum type.");
            }
        }
    }

    public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException {
        ps.setInt(i, parameter.ordinal());
    }

    public E getNullableResult(ResultSet rs, String columnName) throws SQLException {
        int ordinal = rs.getInt(columnName);
        return ordinal == 0 && rs.wasNull() ? null : this.toOrdinalEnum(ordinal);
    }

    public E getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        int ordinal = rs.getInt(columnIndex);
        return ordinal == 0 && rs.wasNull() ? null : this.toOrdinalEnum(ordinal);
    }

    public E getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        int ordinal = cs.getInt(columnIndex);
        return ordinal == 0 && cs.wasNull() ? null : this.toOrdinalEnum(ordinal);
    }

    private E toOrdinalEnum(int ordinal) {
        try {
            return this.enums[ordinal];
        } catch (Exception var3) {
            throw new IllegalArgumentException("Cannot convert " + ordinal + " to " + this.type.getSimpleName() + " by ordinal value.", var3);
        }
    }
}

到此这篇关于mybatis-plus enum实现枚举类型自动转换的文章就介绍到这了,更多相关mybatis-plus 枚举自动转换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家! 

相关文章

  • SpringBoot加载应用事件监听器代码实例

    SpringBoot加载应用事件监听器代码实例

    这篇文章主要介绍了SpringBoot加载应用事件监听器代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-06-06
  • JavaWeb中Servlet的深入理解

    JavaWeb中Servlet的深入理解

    Java Servlet 是运行在 Web 服务器或应用服务器上的程序,它是作为来自 Web 浏览器或其他 HTTP 客户端的请求和 HTTP 服务器上的数据库或应用程序之间的中间层
    2021-10-10
  • Springboot 注解EqualsAndHashCode详解

    Springboot 注解EqualsAndHashCode详解

    注解@EqualsAndHashCode主要用于自动生成equals方法和hashCode方法,callSuper属性为true时,生成的方法会包括父类字段,为false则只包含当前类字段,IDEA工具中有检查提示并可自动修复相关代码,确保注解正确使用,更多详解可查阅相关文档
    2024-10-10
  • Spring表达式语言SpEL用法详解

    Spring表达式语言SpEL用法详解

    这篇文章主要介绍了spring表达式语言SpEL用法详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-01-01
  • 一文精通Java中的volatile关键字

    一文精通Java中的volatile关键字

    volatile是java中的关键词之一,这篇文章主要给大家介绍了关于Java中volatile关键字的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用Java具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧
    2019-05-05
  • Java中处理金额计算之使用Long还是BigDecimal详解

    Java中处理金额计算之使用Long还是BigDecimal详解

    在Java后端开发中处理与钱有关的业务时,确保金额计算的准确性和避免错误非常重要,这篇文章主要给大家介绍了关于Java中处理金额计算之使用Long还是BigDecimal的相关资料,需要的朋友可以参考下
    2024-07-07
  • mybatis-plus数据权限实现代码

    mybatis-plus数据权限实现代码

    这篇文章主要介绍了mybatis-plus数据权限实现,结合了mybatis-plus的插件方式,做出了自己的注解方式的数据权限,虽然可能存在一部分的局限性,但很好的解决了我们自己去解析SQL的功能,需要的朋友可以参考下
    2023-06-06
  • RestTemplate文件上传下载与大文件流式下载

    RestTemplate文件上传下载与大文件流式下载

    这篇文章主要为大家介绍了RestTemplate文件上传下载与大文件流式下载的使用方法示例,有需要的朋友可以借鉴参考下,希望能够有所帮助祝大家多多进步
    2022-03-03
  • Springboot 通过FastJson实现bean对象和Json字符串互转问题

    Springboot 通过FastJson实现bean对象和Json字符串互转问题

    这篇文章主要介绍了Springboot 通过FastJson实现bean对象和Json字符串互转,本文尝试验证两种场景给大家详细介绍,对Springboot FastJson实现bean和Json互转问题,感兴趣的朋友一起看看吧
    2022-08-08
  • 使用java技术抓取网站上彩票双色球信息详解

    使用java技术抓取网站上彩票双色球信息详解

    这篇文章主要介绍了使用java技术抓取网站上彩票双色球信息详解,web结果由html+js+css组成,html结构都有一定的规范,数据动态交互可以通过js实现。,需要的朋友可以参考下
    2019-06-06

最新评论