本篇内容介绍了“MyBatisPlus TypeHandler怎么自定义字段类型转换Handler”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!
因为用到的 ORM 框架是 MybatisPlus,所以首先找的就是有没有官方的支持。
继而就在官网找到一个字段类型处理器,一看才发现,是学过的东西啊,只怪用的太少,知道的太少啊。
然后根据这个线索继续找,就了解到 MyBatis-Plus 字段类型处理器 TypeHandler
这个 TypeHandler 处于的位置,就是应用程序和数据库之间的拦截器,所有的操作,都会走一遍这里。
就翻看源码,想用一个东西,最快的方式就是看一下源码的实现
public interface TypeHandler<T> { /** * 入库前的类型转换 */ void setParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException; /** * 得到结果。 * 查询后的数据处理 */ T getResult(ResultSet rs, String columnName) throws SQLException; T getResult(ResultSet rs, int columnIndex) throws SQLException; T getResult(CallableStatement cs, int columnIndex) throws SQLException; }
找到接口,看一下源码中针对已有属性是如何处理,我们仿写一份,达到我们的要求即可啊.
有这么多,我们直接看一下 BaseTypeHandler
是什么样的处理逻辑,
一方面 base 吗,基础吗,我们就看看基础是什么样的处理啦,另外一方面他是抽象类吗,说明它其他实现类的基类吗。
public abstract class BaseTypeHandler<T> extends TypeReference<T> implements TypeHandler<T> { @Override public void setParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException { if (parameter == null) { if (jdbcType == null) { throw new TypeException("JDBC requires that the JdbcType must be specified for all nullable parameters."); } try { ps.setNull(i, jdbcType.TYPE_CODE); } catch (SQLException e) { throw new TypeException("Error setting null for parameter #" + i + " with JdbcType " + jdbcType + " . " + "Try setting a different JdbcType for this parameter or a different jdbcTypeForNull configuration property. " + "Cause: " + e, e); } } else { try { setNonNullParameter(ps, i, parameter, jdbcType); } catch (Exception e) { throw new TypeException("Error setting non null for parameter #" + i + " with JdbcType " + jdbcType + " . " + "Try setting a different JdbcType for this parameter or a different configuration property. " + "Cause: " + e, e); } } } @Override public T getResult(ResultSet rs, String columnName) throws SQLException { try { return getNullableResult(rs, columnName); } catch (Exception e) { throw new ResultMapException("Error attempting to get column '" + columnName + "' from result set. Cause: " + e, e); } } @Override public T getResult(ResultSet rs, int columnIndex) throws SQLException { try { return getNullableResult(rs, columnIndex); } catch (Exception e) { throw new ResultMapException("Error attempting to get column #" + columnIndex + " from result set. Cause: " + e, e); } } @Override public T getResult(CallableStatement cs, int columnIndex) throws SQLException { try { return getNullableResult(cs, columnIndex); } catch (Exception e) { throw new ResultMapException("Error attempting to get column #" + columnIndex + " from callable statement. Cause: " + e, e); } } // 这里就是设置为 不为 null 时的入库 public abstract void setNonNullParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException; /** * 获取可为空的结果。 */ public abstract T getNullableResult(ResultSet rs, String columnName) throws SQLException; public abstract T getNullableResult(ResultSet rs, int columnIndex) throws SQLException; public abstract T getNullableResult(CallableStatement cs, int columnIndex) throws SQLException; }
看起来好像很长很多的样子:当我们去掉那些判断,精简一下:
public abstract class BaseTypeHandler<T> extends TypeReference<T> implements TypeHandler<T> { @Override public void setParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException { // 设置不为null的参数,进行入库 ,此处是抽象类,下层还有实现类, // 记住这里,待会带你看实现类,你就知道了 setNonNullParameter(ps, i, parameter, jdbcType); } @Override public T getResult(ResultSet rs, String columnName) throws SQLException { // 这里从数据库中获取到数据,然后进行类型的一个设置 return getNullableResult(rs, columnName); } @Override public T getResult(ResultSet rs, int columnIndex) throws SQLException { //这两个抽象方法,给我的感觉是一模一样的,包括下一个也是如此 return getNullableResult(rs, columnIndex); } @Override public T getResult(CallableStatement cs, int columnIndex) throws SQLException { return getNullableResult(cs, columnIndex); } }
public class BigIntegerTypeHandler extends BaseTypeHandler<BigInteger> { @Override public void setNonNullParameter(PreparedStatement ps, int i, BigInteger parameter, JdbcType jdbcType) throws SQLException { // 这里是转为 BigDecimal ,所以这里就算 setBigDecimal, // 那么我们就可以猜测,它还支持其他的方法,Date的话,那就是setDate ps.setBigDecimal(i, new BigDecimal(parameter)); } @Override public BigInteger getNullableResult(ResultSet rs, String columnName) throws SQLException { BigDecimal bigDecimal = rs.getBigDecimal(columnName); // 这里是rs.getBigDecimal ,我们待会去试一下能否getDate就可以了 return bigDecimal == null ? null : bigDecimal.toBigInteger(); } // 这两个暂时没有做了解,Debug的时候,断点没有执行到这,后期再补一块的知识 // 但是为了以防万一,我们待会也会照着它的方式将代码改成这样 @Override public BigInteger getNullableResult(ResultSet rs, int columnIndex) throws SQLException { BigDecimal bigDecimal = rs.getBigDecimal(columnIndex); return bigDecimal == null ? null : bigDecimal.toBigInteger(); } @Override public BigInteger getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { BigDecimal bigDecimal = cs.getBigDecimal(columnIndex); return bigDecimal == null ? null : bigDecimal.toBigInteger(); } }
这个实现类,没什么代码,而且就是set、get ,并没有其他的一些处理逻辑什么的。
那么我们也照这样的方式实现一个。
先明确目标,我们Mysql 中的字段类型 为 BigInt
,Java程序中的属性类型为 Date
,
所以我们在入库的时候就是要将 Date 类型转化为 Long
进行入库,
在从数据库中取出来的时候,要从 Long
类型转化为 Date
映射到 JavaBean中
我们直接copy上面的代码,然后进行一些更改
public class MyDateTypeHandler implements TypeHandler<Date>{ /** * 入库前的类型转换 即执行insert、update方法时会执行 */ @Override public void setParameter(PreparedStatement ps, int i, Date parameter, JdbcType jdbcType) throws SQLException { log.info("setParameter(PreparedStatement ps, int i, Date parameter,JdbcType jdbcType)...."); log.info("[{}],[{}]",parameter,jdbcType); ps.setLong(i, parameter.getTime()); } /** * 查询后的数据处理 * 这就是查询出来,进行映射的时候,会执行这段代码 */ @Override public Date getResult(ResultSet rs, String columnName) throws SQLException { log.info("getResult(ResultSet rs, String columnName)....",columnName); return new Date(rs.getLong(columnName)); } @Override public Date getResult(ResultSet rs, int columnIndex) throws SQLException { log.info("getResult(ResultSet rs, int columnIndex)...."); return new Date(rs.getLong(columnIndex)); } @Override public Date getResult(CallableStatement cs, int columnIndex) throws SQLException { log.info("getResult(CallableStatement cs, int columnIndex)...."); return cs.getDate(columnIndex); } }
咋一眼好像成功啦,但是我们忽略了一个问题,就是MybatisPlus怎么知道它的存在?
那些默认允许进行相互进行类型转换的Handler,它在程序启动的时候,就已经被注册了。
但是我们写了这个类,一方面没有被MybatisPlus知晓,另一方面还没有指明给谁使用,我们又该怎么使用?
基于此,我写了一个小Demo,希望大家能够弄明白,以后遇上也能够解决一些问题
实现目标:
Mysql 中的表的字段为Bigint
,Java程序中为 Date 类型,我们希望还是可以一如既往的使用
MybatisPlus的方法,实现save、list类似这种方法的正常调用,而无需我在保存的时候,将前端传过来的数据手动转换为时间戳,再存放至数据库。查询时亦是如此
数据库
SET NAMES utf8mb4; SET FOREIGN_KEY_CHECKS = 0; -- ---------------------------- -- Table structure for handler_test -- ---------------------------- DROP TABLE IF EXISTS `handler_test`; CREATE TABLE `handler_test` ( `id` int(11) NOT NULL AUTO_INCREMENT, `name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL, `date` bigint(50) NOT NULL COMMENT '存时间戳', PRIMARY KEY (`id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 6 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic; -- ---------------------------- -- Records of handler_test -- ---------------------------- INSERT INTO `handler_test` VALUES (1, '测试数据1', 1659967236); INSERT INTO `handler_test` VALUES (2, '测试数据2', 1659967236); INSERT INTO `handler_test` VALUES (3, '测试插入数据', 1659968162926); INSERT INTO `handler_test` VALUES (4, '测试插入数据', 1659972053771); INSERT INTO `handler_test` VALUES (5, '测试插入数据', 1659972815670); SET FOREIGN_KEY_CHECKS = 1;
我只贴出了相关的代码,其余代码在源码仓库中有,别慌,家人们
service
public interface IHandlerTestService extends IService<HandlerTest> { }
TypeHandler 实现类
/** * @author Ning zaichun */ @Slf4j @MappedJdbcTypes({JdbcType.BIGINT}) //对应数据库类型 @MappedTypes({Date.class}) //java数据类型 public class MyDateTypeHandler implements TypeHandler<Date>{ /** * 入库前的类型转换 */ @Override public void setParameter(PreparedStatement ps, int i, Date parameter, JdbcType jdbcType) throws SQLException { log.info("setParameter(PreparedStatement ps, int i, Date parameter,JdbcType jdbcType)...."); log.info("[{}],[{}]",parameter,jdbcType); ps.setLong(i, parameter.getTime()); } /** * 查询后的数据处理 */ @Override public Date getResult(ResultSet rs, String columnName) throws SQLException { log.info("getResult(ResultSet rs, String columnName)...."); log.info("[{}]",columnName); return new Date(rs.getLong(columnName)); } @Override public Date getResult(ResultSet rs, int columnIndex) throws SQLException { log.info("getResult(ResultSet rs, int columnIndex)...."); return new Date(rs.getLong(columnIndex)); } @Override public Date getResult(CallableStatement cs, int columnIndex) throws SQLException { log.info("getResult(CallableStatement cs, int columnIndex)...."); return cs.getDate(columnIndex); } }
实体类的修改,有两点,
第一点,需要在实体类上加上
@TableName(value = "handler_test",autoResultMap = true)
value 是对应表名,autoResultMap 说的
是否自动构建 resultMap 并使用,
只生效与 mp 自动注入的 method,
如果设置 resultMap 则不会进行 resultMap 的自动构建并注入,
只适合个别字段 设置了 typeHandler 或 jdbcType 的情况
第二点就是要在需要处理的字段上加上
@TableField(typeHandler = MyDateTypeHandler.class)
注解,class就写我们自己编写 Handler.class即可
@Data @TableName(value = "handler_test",autoResultMap = true) @EqualsAndHashCode(callSuper = false) public class HandlerTest implements Serializable { private static final long serialVersionUID = 1L; private String name; /** * 存时间戳 */ @TableField(typeHandler = MyDateTypeHandler.class) @JsonFormat(locale = "zh", timezone = "GMT+8", pattern = "yyyy-MM-dd HH:mm:ss") @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss") private Date date; }
弄完上述这两点,我们还有一个问题,我之前提到一个注册,虽然我们指定了,也写好了,但实际上,还并没有注册到一个存储 TypeHandler 一个 Map 集合中去的,也就是说Mybatis 在遇到的时候,其实还是不知道它的存在的~。
但其实只需要在配置文件中加一行即可,原谅我这么绕圈子,只是希望说明白这是一步步得来的
type-handlers-package
后面填写的是我们Handler 存放的包路径。
有这一步即可。
@RunWith(SpringRunner.class) @SpringBootTest @ContextConfiguration(classes = HandlerApplication.class) public class HandlerServiceTest { @Autowired IHandlerTestService handlerTestService; @Test public void test1(){ List<HandlerTest> list = handlerTestService.list(); list.forEach(System.out::println); } @Test public void test2(){ HandlerTest handlerTest = new HandlerTest(); handlerTest.setDate(new Date()); handlerTest.setName("测试插入数据"); handlerTestService.save(handlerTest); } }
测试插入
==> Preparing: SELECT name,date FROM handler_test ==> Parameters: <== Columns: name, date <== Row: 测试数据1, 1659967236 2022-08-08 23:55:25.854 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659967236 <== Row: 测试数据2, 1659967236 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659967236 <== Row: 测试插入数据, 1659968162926 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659968162926 <== Row: 测试插入数据, 1659972053771 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659972053771 <== Row: 测试插入数据, 1659972815670 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659972815670 <== Row: 测试插入数据, 1659974106847 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659974106847 <== Row: 测试插入数据, 1659974125542 2022-08-08 23:55:25.855 INFO 7368 --- [ main] com.nzc.demo.handler.MyDateTypeHandler : getResult(ResultSet rs, String columnName).... 1659974125542 <== Total: 7 Closing non transactional SqlSession [org.apache.ibatis.session.defaults.DefaultSqlSession@145113f] HandlerTest(name=测试数据1, date=Tue Jan 20 13:06:07 CST 1970) HandlerTest(name=测试数据2, date=Tue Jan 20 13:06:07 CST 1970) HandlerTest(name=测试插入数据, date=Mon Aug 08 22:16:02 CST 2022) HandlerTest(name=测试插入数据, date=Mon Aug 08 23:20:53 CST 2022) HandlerTest(name=测试插入数据, date=Mon Aug 08 23:33:35 CST 2022) HandlerTest(name=测试插入数据, date=Mon Aug 08 23:55:06 CST 2022) HandlerTest(name=测试插入数据, date=Mon Aug 08 23:55:25 CST 2022) 2022-08-08 23:55:25.863 INFO 7368 --- [ionShutdownHook] com.alibaba.druid.pool.DruidDataSource : {dataSource-1} closing ... 2022-08-08 23:55:25.869 INFO 7368 --- [ionShutdownHook] com.alibaba.druid.pool.DruidDataSource : {dataSource-1} closed
“MyBatisPlus TypeHandler怎么自定义字段类型转换Handler”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站,小编将为大家输出更多高质量的实用文章!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。