Mybatis之typeHandler类型转换器

在JDBC中,需要通过PreparedStatement对象中设置那些已经预编译过的SQL语句的参数,执行SQL后,会通过ResultSet对象获取得到数据库的数据,而这些MyBatis是根据数据的类型通过typeHandler来实现的,在typeHandler中,分为jdbcType和javaType,其中jdbcType用于定义数据库类型,而javaType用于定义java类型,那么typeHandler的作用就是承担jdbcType和javaType之间的相互转换,
在很多中情况下,我们并不需要去配置typeHandler,jdbctype,javatype。因为MyBatis会探测应该使用什么类型的typeHandler进行配置,但是有些场景就无法探测到,对于那些需要使用自定义的枚举的场景,或者数据库使用特殊数据类型的场景,可以使用自定义的typeHandler去处理类型之间的转换问题。

源码分析

在MyBatis中typeHandler都要实现接口org.apache.ibatis.type.TypeHandler,首先让我们先看看这个接口的定义

/**
 * @author Clinton Begin
 */
public interface TypeHandler {
 
  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;
 
}

这里我们稍微说明一下他们的定义

  • 其中T是泛型,专指javaType,比如我们需要String的时候,那么实现类可以写为implements TypeHandler
  • setParameter方法,是使用typeHandler通过PreparedStatement对象设置SQL参数的时候使用的具体方法,其中i是参数在SQL的下标,parameter是参数,jdbcType是数据库类型
  • 其中有三个getResult的方法,它的作用是从JDBC结果集中获取数据进行转换,要么使用列名,(columnname)要么使用下标(columnIndex)获取数据库的数据,其中最后一个getResult方法是存储过程使用的。
    TypeHandler源码实现
public abstract class BaseTypeHandler extends TypeReference implements TypeHandler {
 
  protected Configuration configuration;
 
  public void setConfiguration(Configuration c) {
    this.configuration = c;
  }
 
  @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 {
    T result;
    try {
      result = getNullableResult(rs, columnName);
    } catch (Exception e) {
      throw new ResultMapException("Error attempting to get column '" + columnName + "' from result set.  Cause: " + e, e);
    }
    if (rs.wasNull()) {
      return null;
    } else {
      return result;
    }
  }
 
  @Override
  public T getResult(ResultSet rs, int columnIndex) throws SQLException {
    T result;
    try {
      result = getNullableResult(rs, columnIndex);
    } catch (Exception e) {
      throw new ResultMapException("Error attempting to get column #" + columnIndex+ " from result set.  Cause: " + e, e);
    }
    if (rs.wasNull()) {
      return null;
    } else {
      return result;
    }
  }
 
  @Override
  public T getResult(CallableStatement cs, int columnIndex) throws SQLException {
    T result;
    try {
      result = getNullableResult(cs, columnIndex);
    } catch (Exception e) {
      throw new ResultMapException("Error attempting to get column #" + columnIndex+ " from callable statement.  Cause: " + e, e);
    }
    if (cs.wasNull()) {
      return null;
    } else {
      return result;
    }
  }
 
  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;
 
}

简单分析一下BaseTypeHandler的源码

  • BaseTypeHandler是一个抽象类,它实现了TypeHandler的所有接口,需要子类去实现它自定义的4个抽象方法
  • setParameter方法中,当parameter和jdbcType都为空时将抛出异常,如果能明确jdbcType,将会进行空设置(PreparedStatement的setNull方法);如果paramter不为空,将使用setNonNullParameter去设置参数(该方法需要子类去实现)
  • getResult方法,非空结果集是通过getNullableResult方法去获取的,该方法需要子类去实现,同样是针对下标、列名以及存储过程三种的实现
  • getNullableParameter方法用于存储过程

Mybatis使用最多的是typeHandler之一是——StringTypeHandler.它用于字符串转换、

/**
 * @author Clinton Begin
 */
public class StringTypeHandler extends BaseTypeHandler {
 
  @Override
  public void setNonNullParameter(PreparedStatement ps, int i, String parameter, JdbcType jdbcType)
      throws SQLException {
    ps.setString(i, parameter);
  }
 
  @Override
  public String getNullableResult(ResultSet rs, String columnName)
      throws SQLException {
    return rs.getString(columnName);
  }
 
  @Override
  public String getNullableResult(ResultSet rs, int columnIndex)
      throws SQLException {
    return rs.getString(columnIndex);
  }
 
  @Override
  public String getNullableResult(CallableStatement cs, int columnIndex)
      throws SQLException {
    return cs.getString(columnIndex);
  }
}

显然它实现了BaseTypeHandler的4个抽象方法,代码也非常简单。
在这里,MyBatis把javaType和jdbcType相互转换,那么他们是如何进行注册的呢?在MyBatis中采用org.apache.ibatis.type.TypeHandlerRegistry类对象的register方法进行注册

public TypeHandlerRegistry() {
    register(Boolean.class, new BooleanTypeHandler());
    register(boolean.class, new BooleanTypeHandler());
    register(JdbcType.BOOLEAN, new BooleanTypeHandler());
    register(JdbcType.BIT, new BooleanTypeHandler());
}

这样就实现用代码的形式注册typeHandler,注意,自定义的typeHandler一般不会使用代码注册,而是通过配置或者扫描。0

自定义typeHandler

从系统定义的typeHandler可以知道,要实现typeHandler就需要去实现接口typeHandler,或者继承BaseTypeHandler(实际上BassseTypeHandler实现了typehandler的接口)

package com.learn.ssm.chapter4.typehandler;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.TypeHandler;
import org.apache.log4j.Logger;

public class MyTypeHandler implements TypeHandler {

    Logger logger=Logger.getLogger(MyTypeHandler.class);
//  这个就是一个日志的对象,记录和这个类对象进行的一切操作,记录用户的操作
    
    @Override
    public String getResult(ResultSet rs, String columnName) throws SQLException {
        // TODO Auto-generated method stub
        String result=rs.getString(columnName);
        logger.info("读取string参数1【"+result+"】");
        return result;
    }

    @Override
    public String getResult(ResultSet rs, int columnIndex) throws SQLException {
        // TODO Auto-generated method stub
        String result=rs.getString(columnIndex);
        logger.info("读取string参数2【"+result+"】");
        return result;
    }

    @Override
    public String getResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        // TODO Auto-generated method stub
        String result=cs.getString(columnIndex);
        logger.info("读取string参数3【"+result+"】");
        return result;
    }

    @Override
    public void setParameter(PreparedStatement ps, int i, String parameter,
            JdbcType jdbcType) throws SQLException {
        // TODO Auto-generated method stub
        logger.info("设置string参数["+parameter+"]");
        ps.setString(i, parameter);
    }

}

定义的typeHandler泛型为String,显然我们要把数据库的数据类型转换为String型,然后实现设置参数和获取结果集方法
配置typeHandler,配置文件在mybatis-config.xml



          


    

配置完成后系统才会读取它,这样注册后,当jdbcType和javaType能与MyTypeHandler对应的时候,它就会启动MyTypeHandler.有时候还可以显示启用typeHandler,一般而言启用这个typeHandler有 两种方式。





    
        
        
        
    

    

    

    
    

注意要么指定了与自定义typeHandler一致的jdbcType和javaType,要么直接使用typeHandler指定具体的实现类,在一些因为数据库返回为空导致无法断定采用哪个typeHandler来处理,而又没有注册对应的javaType的typeHandler时,MyBatis无法知道使用哪个typeHandler

  • 补充 resultMap是Mybatis最强大的元素,它可以将查询到的复杂数据(比如查询到几个表中数据)映射到一个结果集当中


  
  
  
    
    
  
  
  
    
      
  

运行

package com.learn.ssm.chapter4.main;

import org.apache.ibatis.session.SqlSession;
import org.apache.log4j.Logger;

import com.learn.ssm.chapter3.mapper.RoleMapper;

import com.learn.ssm.chapter3.pojo.Role;

import com.learn.ssm.chapter3.utils.SqlSessionFactoryUtils;

public class chapter4Main {

    public static void main(String[] args) {
        testRoleMapper();
//      testTypeHandler();
    }

    private static void testRoleMapper() {
        Logger log = Logger.getLogger(chapter4Main.class);
        SqlSession sqlSession = null;
        try {
            sqlSession = SqlSessionFactoryUtils.openSqlSession();
            RoleMapper roleMapper = sqlSession.getMapper(RoleMapper.class);
            Role role = roleMapper.getRole(1L);
    
            log.info(role.getRoleName());
    
        } finally {
            if (sqlSession != null) {
                sqlSession.close();
            }
        }
    }

你可能感兴趣的:(Mybatis之typeHandler类型转换器)