java处理枚举类型 查询和插入

SpringBoot整合Mybatis注解式开发*

在配置文件application.yml中添加 处理枚举的所在包：
type-handlers-package: cn.rcm.enums.handle

1、枚举类

public enum SexEnum implements BaseEnum {

    MAN(1, "男"),
    WOMAN(2, "女");

    private int key;
    private String value;

    private SexEnum(int key, String value) {
        this.key = key;
        this.value = value;
    }

    public int getKey() {
        return key;
    }

    public void setKey(int key) {
        this.key = key;
    }

    public String getValue() {
        return value;
    }

    public void setValue(String value) {
        this.value = value;
    }
}

2、实体类

@Data
public class User {

    private SexEnum sexEnum;
}

3、定义一个实现enum 接口


public interface BaseEnum {

    int getKey();

    void setKey(int key);

    String getValue();

    void setValue(String value);
}

4、

public class EnumKeyTypeHandler extends BaseTypeHandler<**BaseEnum** > {

    private Class type;
    private final BaseEnum[] enums;

    /**
     * 设置配置文件设置的转换类以及枚举类内容，供其他方法更便捷高效的实现
     *
     * @param type 配置文件中设置的转换类
     */
    public EnumKeyTypeHandler(Class type) {
        if (type == null)
            throw new IllegalArgumentException("Type argument cannot be null");
        this.type = type;
        this.enums = type.getEnumConstants();
        if (this.enums == null)
            throw new IllegalArgumentException(type.getSimpleName()
                    + " does not represent an enum type.");
    }

    @Override
    public BaseEnum getNullableResult(ResultSet rs, String columnName) throws SQLException {
        // 根据数据库存储类型决定获取类型，本例子中数据库中存放INT类型
        int i = rs.getInt(columnName);

        if (rs.wasNull()) {
            return null;
        } else {
            // 根据数据库中的code值，定位BaseEnum子类
            return locateBaseEnum(i);
        }
    }

    @Override
    public BaseEnum getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        // 根据数据库存储类型决定获取类型，本例子中数据库中存放INT类型
        int i = rs.getInt(columnIndex);
        if (rs.wasNull()) {
            return null;
        } else {
            // 根据数据库中的code值，定位BaseEnum子类
            return locateBaseEnum(i);
        }
    }

    public BaseEnum getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        // 根据数据库存储类型决定获取类型，本例子中数据库中存放INT类型
        int i = cs.getInt(columnIndex);
        if (cs.wasNull()) {
            return null;
        } else {
            // 根据数据库中的code值，定位BaseEnum子类
            return locateBaseEnum(i);
        }
    }

    public void setNonNullParameter(PreparedStatement ps, int i, BaseEnum parameter, JdbcType jdbcType)
            throws SQLException {
        // baseTypeHandler已经帮我们做了parameter的null判断
        ps.setInt(i, parameter.getKey());

    }

    /**
     * 枚举类型转换，由于构造函数获取了枚举的子类enums，让遍历更加高效快捷
     *
     * @param key 数据库中存储的自定义code属性
     * @return code对应的枚举类
     */
    private BaseEnum locateBaseEnum(int key) {
        for (BaseEnum status : enums) {
            if (status.getKey() == key) {
                return status;
            }
        }
        throw new IllegalArgumentException("未知的枚举类型：" + key + ",请核对" + type.getSimpleName());
    }

}

数据库里面数据

5、Dao 层实现数据的数字类型转换成我们想要的类型给前台

 //根据UserID查询User信息
    @Select("SELECT * FROM testuser WHERE userid=#{id}")
    @Results({
            @Result(property = "id", column = "userid"),
            @Result(property = "name", column = "username"),
            @Result(property = "birthday", column = "userbirth"),
            @Result(property = "description", column = "descript"),
            **@Result(property = "sexEnum",column =  "sex_enum",   typeHandler=cn.rcm.enums.handle.EnumKeyTypeHandler.class)}**)
     List selectByID(String id);

结果返回：
   [
      {
        "sexEnum": "MAN"
      }
    ]

6、POST请求插入一条记录

body：
{
“sexEnum”: “MAN”
}

执行后数据库结果：

参考链接：