欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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
    # 支持統(tǒng)配符 * 或者 ; 分割
    typeEnumsPackage: com.fii.gxback.eam.domain.enums

通過上述的配置后，就可以了。那么，接下來我們重點分析一下type-enums-package

按照官方文檔說明：

枚舉類 掃描路徑，如果配置了該屬性，會將路徑下的枚舉類進行注入，讓實體類字段能夠簡單快捷的使用枚舉屬性

通過扒源碼發(fā)現，其實是幫我們實現了對應的 typehandler(com.baomidou.mybatisplus.core.handlers.MybatisEnumTypeHandler)

其中，isMpEnums中，判定整合兩種方式的判定，即 IEnum.class、EnumValue.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。

其實，跟一下代碼，可以發(fā)現，跟我們自己手寫一個typeHandler沒區(qū)別，這里最終獲取轉換值時，調用了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 枚舉自動轉換內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！ 

相關文章

最新評論