欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-04-23 11:49:54
...

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<BaseEnum> type;
    private final BaseEnum[] enums;

    /**
     * 设置配置文件设置的转换类以及枚举类内容,供其他方法更便捷高效的实现
     *
     * @param type 配置文件中设置的转换类
     */
    public EnumKeyTypeHandler(Class<BaseEnum> 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());
    }

}

数据库里面数据

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

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<User> selectByID(String id);

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

6、POST请求插入一条记录

body:
{
“sexEnum”: “MAN”
}

执行后数据库结果:
java处理枚举类型 查询和插入
参考链接: