zoukankan      html  css  js  c++  java
  • 面试官问:Mybatis中的TypeHandler你用过吗?

    持续原创输出,点击上方蓝字关注我吧

    目录

    • 前言
    • 环境配置
    • 什么是TypeHandler?
      • 如何自定义?
      • 如何将其添加到Mybatis中?
      • XML文件中如何指定TypeHandler?
    • 源码中如何执行TypeHandler?
      • 入参如何转换?
      • 结果如何转换?
      • 总结
    • 总结

    前言

    • 相信大家用Mybatis这个框架至少一年以上了吧,有没有思考过这样一个问题:数据库有自己的数据类型,Java有自己的数据类型,那么Mybatis是如何把数据库中的类型和Java的数据类型对应的呢?

    • 本篇文章就来讲讲Mybatis中的黑匣子TypeHandler(类型处理器),说它是黑匣子一点都不为过,总是在默默的奉献着,但是不为人知。

    环境配置

    • 本篇文章讲的一切内容都是基于Mybatis3.5SpringBoot-2.3.3.RELEASE

    什么是TypeHandler?

    • 顾名思义,类型处理器,将入参和结果转换为所需要的类型,Mybatis中对于内置了许多类型处理器,实际开发中已经足够使用了,如下图:

    • 类型处理器这个接口其实很简单,总共四个方法,一个方法将入参的Java类型的数据转换为JDBC类型,三个方法将返回结果转换为Java类型。源码如下:

    public interface TypeHandler<T> {
      //设置参数,java类型转换为jdbc类型
      void setParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException;
      //将查询的结果转换为java类型
      T getResult(ResultSet rs, String columnName) throws SQLException;
    
      T getResult(ResultSet rs, int columnIndex) throws SQLException;
    
      T getResult(CallableStatement cs, int columnIndex) throws SQLException;
    }
    

    如何自定义并使用TypeHandler?

    • 实际应用开发中的难免会有一些需求要自定义一个TypeHandler,比如这样一个需求:前端传来的年龄是,,但是数据库定义的字段却是int类型(1男2女)。此时可以自定义一个年龄的类型处理器,进行转换。

    如何自定义?

    • 自定义的方式有两种,一种是实现TypeHandler这个接口,另一个就是继承BaseTypeHandler这个便捷的抽象类。
    • 下面直接继承BaseTypeHandler这个抽象类,定义一个年龄的类型处理器,如下:
    @MappedJdbcTypes(JdbcType.INTEGER)
    @MappedTypes(String.class)
    public class GenderTypeHandler extends BaseTypeHandler {
    
        //设置参数,这里将Java的String类型转换为JDBC的Integer类型
        @Override
        public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
            ps.setInt(i, StringUtils.equals(parameter.toString(),"男")?1:2);
        }
    
        //以下三个参数都是将查询的结果转换
        @Override
        public Object getNullableResult(ResultSet rs, String columnName) throws SQLException {
            return rs.getInt(columnName)==1?"男":"女";
        }
    
        @Override
        public Object getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
            return rs.getInt(columnIndex)==1?"男":"女";
        }
    
        @Override
        public Object getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
            return cs.getInt(columnIndex)==1?"男":"女";
        }
    }
    
    • 这里涉及到两个注解,如下:
      • @MappedTypes:指定与其关联的 Java 类型列表。 如果在 javaType 属性中也同时指定,则注解上的配置将被忽略。
      • @MappedJdbcTypes:指定与其关联的 JDBC 类型列表。 如果在 jdbcType 属性中也同时指定,则注解上的配置将被忽略。

    如何将其添加到Mybatis中?

    • Mybatis在与SpringBoot整合之后一切都变得很简单了,其实这里有两种配置方式,下面将会一一介绍。
    • 第一种:只需要在配置文件application.properties中添加一行配置即可,如下:
    ## 设置自定义的Typehandler所在的包,启动的时候会自动扫描配置到Mybatis中
    mybatis.type-handlers-package=cn.cb.demo.typehandler
    
    • 第二种:其实任何框架与Springboot整合之后,只要配置文件中能够配置的,在配置类中都可以配置(除非有特殊定制,否则不要轻易覆盖自动配置)。如下:
    @Bean("sqlSessionFactory")
        public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
            SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
            sqlSessionFactoryBean.setDataSource(dataSource);
            sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(MAPPER_LOCATOIN));
            org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
            // 自动将数据库中的下划线转换为驼峰格式
            configuration.setMapUnderscoreToCamelCase(true);
            configuration.setDefaultFetchSize(100);
            configuration.setDefaultStatementTimeout(30);
            sqlSessionFactoryBean.setConfiguration(configuration);
            //将typehandler注册到mybatis
            GenderTypeHandler genderTypeHandler = new GenderTypeHandler();
            TypeHandler[] typeHandlers=new TypeHandler[]{genderTypeHandler};
            sqlSessionFactoryBean.setTypeHandlers(typeHandlers);
            return sqlSessionFactoryBean.getObject();
        }
    
    • 第二种方式的思想其实就是重写自动配置类MybatisAutoConfiguration中的方法。注意:除非自己有特殊定制,否则不要轻易重写自动配置类中的方法

    XML文件中如何指定TypeHandler?

    • 上面的两个步骤分别是自定义和注入到Mybatis中,那么如何在XML文件中使用呢?
    • 使用其实很简单,分为两种,一种是更新,一种查询,下面将会一一介绍。
    • 更新:删除自不必说了,这里讲的是updateinsert两种,只需要在#{}中指定的属性typeHandler为自定义的全类名即可,代码如下:
    <insert id="insertUser">
            insert into user_info(user_id,his_id,name,gender,password,create_time)
            values(#{userId,jdbcType=VARCHAR},#{hisId,jdbcType=VARCHAR},#{name,jdbcType=VARCHAR},
            #{gender,jdbcType=INTEGER,typeHandler=cn.cb.demo.typehandler.GenderTypeHandler},#{password,jdbcType=VARCHAR},now())
        </insert>
    
    • 查询:查询的时候类型处理会将JDBC类型的转化为Java类型,因此也是需要指定typeHandler,需要在resultMap中指定typeHandler这个属性,值为全类名,如下:
    <resultMap id="userResultMap" type="cn.cb.demo.domain.UserInfo">
            <id column="id" property="id"/>
            <result column="user_id" property="userId"/>
            <result column="his_id" property="hisId"/>
                <!-- 指定typeHandler属性为全类名-->
            <result column="gender" property="gender" typeHandler="cn.cb.demo.typehandler.GenderTypeHandler"/>
            <result column="name" property="name"/>
            <result column="password" property="password"/>
        </resultMap>
    
        <select id="selectList" resultMap="userResultMap">
            select * from user_info where status=1
            and user_id in
            <foreach collection="userIds" item="item" open="(" separator="," close=")" >
                #{item}
            </foreach>
        </select>
    

    源码中如何执行TypeHandler?

    • 既然会使用TypeHandler了,那么肯定要知道其中的执行原理了,在Mybatis中类型处理器是如何在JDBC类型和Java类型进行转换的,下面的将从源码角度详细介绍。

    入参如何转换?

    • 这个肯定是发生在设置参数的过程中,详细的代码在PreparedStatementHandler中的parameterize()方法中,这个方法就是设置参数的方法。源码如下:
     @Override
      public void parameterize(Statement statement) throws SQLException {
        //实际调用的是DefaultParameterHandler
        parameterHandler.setParameters((PreparedStatement) statement);
      }
    
    • 实际执行的是DefaultParameterHandler中的setParameters方法,如下:
    public void setParameters(PreparedStatement ps) {
        ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());
        //获取参数映射
        List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
        //遍历参数映射,一一设置
        if (parameterMappings != null) {
          for (int i = 0; i < parameterMappings.size(); i++) {
            ParameterMapping parameterMapping = parameterMappings.get(i);
            if (parameterMapping.getMode() != ParameterMode.OUT) {
              Object value;
              String propertyName = parameterMapping.getProperty();
              if (boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional params
                value = boundSql.getAdditionalParameter(propertyName);
              } else if (parameterObject == null) {
                value = null;
              } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
                value = parameterObject;
              } else {
                MetaObject metaObject = configuration.newMetaObject(parameterObject);
                value = metaObject.getValue(propertyName);
              }
              //获取类型处理器,如果不存在,使用默认的
              TypeHandler typeHandler = parameterMapping.getTypeHandler();    
              //JdbcType
              JdbcType jdbcType = parameterMapping.getJdbcType();
              if (value == null && jdbcType == null) {
                jdbcType = configuration.getJdbcTypeForNull();
              }
              try {
                //调用类型处理器中的方法设置参数,将Java类型转换为JDBC类型
                typeHandler.setParameter(ps, i + 1, value, jdbcType);
              } catch (TypeException e) {
                throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
              } catch (SQLException e) {
                throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
              }
            }
          }
        }
      }
    
    • 从上面的源码中可以知道这行代码typeHandler.setParameter(ps, i + 1, value, jdbcType);就是调用类型处理器中的设置参数的方法,将Java类型转换为JDBC类型。

    结果如何转换?

    • 这一过程肯定是发生在执行查询语句的过程中,之前也是介绍过Mybatis的六大剑客,其中的ResultSetHandler这个组件就是对查询的结果进行处理的,那么肯定是发生在这一组件中的某个方法。
    • PreparedStatementHandler执行查询结束之后,调用的是ResultSetHandler中的handleResultSets()方法,对结果进行处理,如下:
    public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
        PreparedStatement ps = (PreparedStatement) statement;
        //执行SQL
        ps.execute();
        //处理结果
        return resultSetHandler.handleResultSets(ps);
      }
    
    • 最终的在DefaultResultHandler中的getPropertyMappingValue()方法中调用了TypeHandler中的getResult()方法,如下:
    private Object getPropertyMappingValue(ResultSet rs, MetaObject metaResultObject, ResultMapping propertyMapping, ResultLoaderMap lazyLoader, String columnPrefix)
          throws SQLException {
        if (propertyMapping.getNestedQueryId() != null) {
          return getNestedQueryMappingValue(rs, metaResultObject, propertyMapping, lazyLoader, columnPrefix);
        } else if (propertyMapping.getResultSet() != null) {
          addPendingChildRelation(rs, metaResultObject, propertyMapping);   // TODO is that OK?
          return DEFERRED;
        } else {
          final TypeHandler<?> typeHandler = propertyMapping.getTypeHandler();
          final String column = prependPrefix(propertyMapping.getColumn(), columnPrefix);
          //执行typeHandler中的方法获取结果并且转换为对应的Java类型
          return typeHandler.getResult(rs, column);
        }
      }
    

    总结

    总结

    • 本文详细的介绍了TypeHandler在Mybatis中的应用、自定义使用以及从源码角度分析了类型处理器的执行流程,如果觉得作者写的不错,有所收获的话,不妨点点关注,分享一波。
    如果觉得作者写的好,有所收获的话,点个关注,推荐一波,文章首发于公众号!!!
  • 相关阅读:
    nginx解决跨域问题(临时应急,最好服务端配置)
    TCP与UDP区别
    图表转图片
    kafka-python开发kafka生产者和消费者
    Django restframework 序列化之 ModelSerializer 小记
    django urls 配置小记
    pycharm 使用Git提交代码到Github
    django rest_framework vue 实现用户列表分页
    Django rest framework ---- 权限
    Django rest framework----认证
  • 原文地址:https://www.cnblogs.com/Chenjiabing/p/13677321.html
Copyright © 2011-2022 走看看