TypeHandler转换指定数据库中数据为Enum枚举
在一些时候,我们的数据库需要存放一些例如状态信息的数据,通常的我们使用int整型来保存,例如(0:失败,1:成功)等,用这样的数据是可以的,但对于前端的编程是不友好的,所以我们可以使用TypeHandler来进行一个转换,将其转化成一个枚举型。
一、数据库中的数据定义
这里的user_state和user_priority表示了一些状态信息(范围是整数0~5),接下来我们要将这个转化为枚举型
二、创建枚举Enum类
以User_state为例:
package com.snapshot2.demo.enumeration;public enum UserStateEnum { NORMAL(0,"正常状态"), FROZEN(1,"冻结"), BAN(2,"禁用"), MUTE(3,"禁言"), SENSITIVE(4,"敏感用户"), DYF(5,"这是创作者董一帆的"); private int stateCode; private String stateString; public static UserStateEnum getUserStateByCode(int stateCode){ for (UserStateEnum stateEnum: values()){ if(stateEnum.getStateCode() == stateCode){ return stateEnum; } } return null; } /* Constructor */ UserStateEnum(int stateCode, String stateString) { this.stateCode = stateCode; this.stateString = stateString; } /* Getter and Setter */ public int getStateCode() { return stateCode; } public void setStateCode(int stateCode) { this.stateCode = stateCode; } public String getStateString() { return stateString; } public void setStateString(String stateString) { this.stateString = stateString; }}
三、创建相应得TypeHandler类,继承于BaseTypeHandler<T>类,并重写相应得方法
1.这里的<T>是我们指定转化的类型
2.@MappedJdbcTypes注解声明数据库中对应数据类型
3.@MappedTypes(value = UserStateEnum.class)声明转化后的类型
package com.snapshot2.demo.typehandler;import com.snapshot2.demo.enumeration.UserStateEnum;import org.apache.ibatis.type.BaseTypeHandler;import org.apache.ibatis.type.JdbcType;import org.apache.ibatis.type.MappedJdbcTypes;import org.apache.ibatis.type.MappedTypes;import java.sql.CallableStatement;import java.sql.PreparedStatement;import java.sql.ResultSet;import java.sql.SQLException;/* 数据库中的数据类型 */@MappedJdbcTypes(JdbcType.INTEGER)/* 转化后的数据类型 */@MappedTypes(value = UserStateEnum.class)public class UserStateTypeHandler extends BaseTypeHandler{ @Override public void setNonNullParameter(PreparedStatement preparedStatement, int i, UserStateEnum userStateEnum, JdbcType jdbcType) throws SQLException { preparedStatement.setInt(i,userStateEnum.getStateCode()); } @Override public UserStateEnum getNullableResult(ResultSet resultSet, String s) throws SQLException { int code =resultSet.getInt(s); if(code>=0&&code<=5){ return UserStateEnum.getUserStateByCode(code); } return null; } @Override public UserStateEnum getNullableResult(ResultSet resultSet, int i) throws SQLException { int code = resultSet.getInt(i); if(code>=0&&code<=5){ return UserStateEnum.getUserStateByCode(code); } return null; } @Override public UserStateEnum getNullableResult(CallableStatement callableStatement, int i) throws SQLException { int code = callableStatement.getInt(i); if(code>=0&&code<=5){ return UserStateEnum.getUserStateByCode(code); } return null; }}
四、最后在application.properties配置文件中springboot指定TypeHandler所在的包
mybatis.type-handlers-package=com.snapshot2.demo.typehandler