springboot+mybatis+枚举处理器的实现

springboot+mybatis+枚举处理器的实现

目录背景现状期望实现通用的枚举接口mybaits枚举处理器配置枚举处理器包含枚举类型字段的实体类查询mapper

背景

在Spring boot项目开发中经常遇到需要使用枚举的场景，比如描述状态、性别、类型等相关字段。通常这些字段在数据库会以tinyint类型存在，比如：0：女性，1:男性；或者，0:设备掉线，1:设备在线。如果在系统中我们也以int的方式到处使用，后期在维护的时候对数字的理解会非常困难，所以通常这种字段我们一般采用枚举的方式在系统的流转，而在存储的时候我们使用数字的方式存储。

以上，因为数据库存储字段的类型和我们在系统中流转的类型不同，我们需要实现一个两者自动转换的功能，也就是标题中提到的枚举处理器。

现状

如果我们采用全数字的方式在系统中流转，最终系统代码就会变成这样：

if(status == 1){ xAbPlxWB// 时间一长，谁来告诉我1代表什么状态

...

}else if (status == 0){ // 0又代表什么状态

...

}

...

以上这种操作会造成代码的可读性非常低，非常需要依靠对其中数字的注释进行编码。

期望

希望类型性别、状态这类字段可以全部使用枚举类型，提高代码的可维护性。考虑到大部分我们使用枚举的时候都可以使用ONLINE(1, "在线"), OFFLINE(0, "掉线")这样的方式，可以总结出一个通用的枚举接口，后续类似的枚举类都实现这个接口。

实现

为了满足期望的功能，通过一个通用的枚举接口和mybatis的枚举处理器来实现业务系统中使用枚举类型和mysql中使用数值类型的功能。

通用的枚举接口

简单粗暴的命名这个通用枚举接口为KeyValueEnum:

public interface KeyValueEnum {

int getKey();

String getValue();

}

基于该接口实现一个状态枚举类：

public enum StatusEnum implements KeyValueEnum {

ONLINE(1, "在线")，

OFFLINE(0, "掉线");

private int key;

private String value;

StatusEnum(int key, String value){

this.key = key;

this.value = value;

}

@Override

public int getKey() {

return key;

}

@Override

public String getValue() {

return value;

}

}

由此，所有类似的接口都可以用过getKey()和getValue()方法来获取相关数值。

mybaits枚举处理器

@MappedTypes(value = {Status.class, Sex.class}) // 每增加一种枚举类型，就在此添加。

public class EnumTypeHandler extends BaseTypeHandler {

private Class type;

private E[] enums;

public EnumTypeHandler(Class type){

this.type = type;

this.enums = type.getEnumConstants();

}

@Override

public void setNonNullParameter(PreparedStatement preparedStatement, int i, KeyValueEnum keyValueEnum, JdbcType jdbcType) throws SQLException {

preparedStatement.setInt(i, keyValueEnum.getKey());

}

@Override

public E getNullableResult(ResultSet resultSet, int i) throws SQLException {

int key = resultSet.getInt(i);

return getEnum(key);

}

@Override

public E getNullableResult(ResultSet resultSet, String s) throws SQLException {

int key = resultSet.getInt(s);

return getEnum(key);

}

@Override

public E getNullableResult(CallableStatement callableStatement, int i) throws SQLException {

int key = callableStatement.getInt(i);

return getEnum(key);

}

private E getEnum(int key){

for (E keyValue:

enums) {

if (keyValue.getKey() == key){

return keyValue;

}

}

return null;

}

}

枚举处理器是通过继承mybatis自带的BaseTypeHandler抽象类实现的，其中包含的具体方法参考mybatis的BaseTypeHandler文档即可。

需要注意的是，如果每增加一种需要存储到数据库的枚举类型就需要在@MappedTypes(value = {Status.class, Sex.class})注解中添加一个对应的枚举类。

配置枚举处理器

# application.yaml

mybatis:

configuration:

map-underscore-to-camel-case: true # 下划线转驼峰

type-handlers-package: xxx.xxx.xxx.handler.mybatisTypeHandler # 配置刚才编辑的枚举处理器

包含枚举类型字段的实体类

@Data

public class device {

private long id;

private Status status;

}

期望实现的目标是直接通过数据库查询出对应的枚举类型。

查询mapper

public interface deviceMapper {

@Results(id="aaa", value = {

@Result(id = true, column = "id", property = "id"),

@Result(column = "status", property = "status")

})

@Select("select * from device where id = #{id}")

Device getDeviceById(long id);

}

通过mapper查询最终可以实现数据库tinyint类型查询后转换为实体类字段对应的枚举类型。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。