前言
笔者在最近的项目开发中，频繁地遇到了 Java 类型与 JDBC 类型之间的 2 个转换问题：
其实要处理也不麻烦，可以在每次入库地方的手动将 Java Bean 调用 JSON.toJSONString() 即可，取出数据库数据的时候再 JSON.parseObject()解析。再说处理枚举类型也并不难，无非就是手动将枚举的 int 型属性取出后 set 到数据库的 int 中去。
而本文要介绍的自定义 TypeHandler 处理器的作用，就是自动处理 Java Bean 与数据库类型的转换，提高编码效率，通过全局的统一处理省去繁琐的手动转换。
一、TypeHandler 简介
如果我们使用的是 Mybatis 或者是 Mybatis Plus 的话，在 SQL 语句执行过程中，无论是设置参数还是获取结果集，都需要通过 TypeHandler 进行类型转换。
MyBatis 提供了丰富的内置 TypeHandler 实现，以支持常见的数据类型转换，如以下几种：
表 1-1
1.1 转换步骤
当 MyBatis 执行一个预编译的 SQL 语句（如 INSERT、UPDATE 等）时，它需要将 Java 对象中的属性值设置到 SQL 语句中对应的占位符上。这个过程就是通过 TypeHandler 来实现的。
具体步骤如下：
MyBatis 会根据映射配置找到对应的 TypeHandle r 实例，这个映射配置可以在 MyBatis 的配置文件或者 Mapper 的 XML 文件中定义；
TypeHandler 实例会接收到 Java 对象中的属性值，并将其转换为 JDBC 能够识别的类型，这个转换过程是根据两者之间的映射关系来实现的；
转换后的值会被设置到 PreparedStatement 对象中对应的占位符上，以便数据库能够正确解析和执行 SQL 语句。
1.2 转换规则
再次强调，TypeHandler 的核心功能是实现 Java 类型和 JDBC 类型之间的映射和转换，这个映射和转换规则是根据 Java 类型和 JDBC 类型的特性和语义来定义的。
下面两章就举两个例子来加以说明。
二、JSON 转换
应用的 .yml 配置文件新增以下：
mybatis-plus: type-handlers-package: #自定义 handler 类所在的包路径
复制代码
/** * <p>作用：即 Java 实体属性可以直接使用 JSONObject 映射数据库的 varchar，方便入库、出库</p> * <p>注意：需要在 .yml 配置文件上加上 {@code mybatis:type-handlers-package: 本类所在包路径}</p> * * @param <T> 该泛型即为需要转换成 varchar 的 Java 对象 * @MappedTypes 注解很关键，指定了映射的类型 */@MappedTypes({JSONObject.class, JSONArray.class})public class JSONTypeHandler <T> extends BaseTypeHandler<T> {
@Override public void setNonNullParameter(PreparedStatement preparedStatement, int i, T param, JdbcType jdbcType) throws SQLException { //将指定的参数设置为给定的 Java String 值，数据库驱动程序及其转换成 varchar 类型 preparedStatement.setString(i, JSON.toJSONString(param));
}
@Override public T getNullableResult(ResultSet resultSet, String columnName) throws SQLException { //这里根据字段名去拿到之前放进来的 jsonStr 值 String jsonStr = resultSet.getString(columnName); return StringUtils.isNotBlank(jsonStr) ? JSON.parseObject(jsonStr, getRawType()) : null; }
@Override public T getNullableResult(ResultSet resultSet, int columnIndex) throws SQLException { //这里是根据位置来确定字段，进而拿到该字段的值（之前放进来的 jsonStr） String jsonStr = resultSet.getString(columnIndex); return StringUtils.isNotBlank(jsonStr) ? JSON.parseObject(jsonStr, getRawType()) : null; }
@Override public T getNullableResult(CallableStatement callableStatement, int columnIndex) throws SQLException { //这里是根据SQL存储过程里的字段位置来拿字段的值 String jsonStr = callableStatement.getString(columnIndex); return StringUtils.isNotBlank(jsonStr) ? JSON.parseObject(jsonStr, getRawType()) : null; }
}
复制代码
三、枚举转换
/** * <p>作用：将实体类中的枚举 code 映射为数据库的 int</p> * <p>注意：需要在 .yml 配置文件上加上 {@code mybatis:type-handlers-package: 本类所在包路径}</p> * * @param <E> 该泛型即为需要处理的枚举对象，使用上界通配符来保证类型安全 */@MappedTypes(MyEnum.class)public class EnumCodeTypeHandler <E extends MyEnum> extends BaseTypeHandler<E> {
private final Class<E> type;
/** * 记录枚举值和枚举的对应关系 */ private final Map<Integer, E> enumMap = new ConcurrentHashMap<>();
public EnumCodeTypeHandler(Class<E> type) { Assert.notNull(type, "argument cannot be null"); this.type = type; E[] enums = type.getEnumConstants(); if (Objects.nonNull(enums)) { //这里将枚举值和枚举类型存入 enumMap for (E e : enums) { this.enumMap.put(e.toCode(), e); } } }
@Override public void setNonNullParameter(PreparedStatement preparedStatement, int index, E e, JdbcType jdbcType) throws SQLException { //这里将枚举的 code 转为数据库该字段的 int 类型 preparedStatement.setInt(index, e.toCode()); }
@Override public E getNullableResult(ResultSet resultSet, String columnName) throws SQLException { //这里根据字段名来将数据库的 int 转为 Java 的 Integer Integer code = resultSet.getInt(columnName); if (resultSet.wasNull()){ return null; }else { //取出对应的枚举值 return enumMap.get(code); } }
@Override public E getNullableResult(ResultSet resultSet, int columnIndex) throws SQLException { //这里根据字段位置来将数据库的 int 转为 Java 的 Integer Integer code = resultSet.getInt(columnIndex); if (resultSet.wasNull()){ return null; }else { //取出对应的枚举值 return enumMap.get(code); } }
@Override public E getNullableResult(CallableStatement callableStatement, int columnIndex) throws SQLException { //这里根据SQL存储过程里的字段位置将字段的 int 转为 Java 的 Integer Integer code = callableStatement.getInt(columnIndex); if (callableStatement.wasNull()){ return null; }else { //取出对应的枚举值 return enumMap.get(code); } }
}
复制代码
/** * <p>作用：该接口包含了两个枚举操作的抽象方法</p> */public interface MyEnum {
/** * 根据 code 获取枚举实例 * @param code */ MyEnum fromCode(int code);
/** * 获取枚举中的 code */ int toCode();
}
复制代码
@Getter@RequiredArgsConstructorpublic enum StudyStatusEnum implements MyEnum{ ONE(1, "枚举1"), TWO(2, 枚举2"), THREE(3, "枚举3"), FOUR(4, "枚举4"), FIVE(5, "枚举5");
private final Integer code;
private final String desc;
/** * 根据 code 获取枚举实例 */ @Override public MyEnum fromCode(int code) { return Arrays.stream(StudyStatusEnum.values()) .filter(val -> val.getCode().equals(code)) .findFirst().orElse(null); }
/** * 获取枚举中的 code */ @Override public int toCode() { return this.getCode(); }
}
复制代码
四、文章小结
通过内置和自定义的 TypeHandler，我们可以轻松处理各种数据类型转换需求，提高开发效率和代码可维护性。
在 Spring Boot 环境中使用自定义 TypeHandler 更是简化了配置和注册过程，使得我们能够更专注于业务逻辑的实现。
最后，文章如有不足和错误，还请大家指正。或者你有其它想说的，也欢迎大家在评论区交流！
文章转载自：CodeBlogMan
原文链接：https://www.cnblogs.com/CodeBlogMan/p/18353997
体验地址：http://www.jnpfsoft.com/?from=infoq
评论