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());
}
}
数据库里面数据
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”
}
执行后数据库结果:
参考链接:
上一篇: 利用枚举,得到类型的字段