亲宝软件园·资讯

展开

Mybatis TypeHandler接口及继承关系示例解析

念念清晰 人气:0

开篇

JDBC类型与Java类型并不是完全一一对应的。所以在PreparedStatement绑定参数的时候需要把Java类型转为JDBC类型。JDBC类型的枚举值在JdbcType枚举值中存储。

MyBatis中提供了一个接口专用于JDBC类型与Java类型的转换。它就是我们今天的主题:TypeHandler(类型转换器)

TypeHandler接口

TypeHandler是用于JDBC类型与Java类型的转换

我们先来看一下这个接口的定义,它都规范了哪些行为。再来说这些方法的实现类和具体作用。

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;
}

该接口中共有4个方法。这四个方法只定义了两种行为,一是setParameter(设置参数)二是操作结果集获取指定对象

方法名描述
setParameter使用PreparedStatement执行SQL是,设置带?的参数。
getResult获取结果集中指定列名对应的值,或者获取结果集中指定索引对应的值

实际setParameter方法的底层实现就是JDBC操作

PreparedStatement ps = connection.prepareStatement();
ps.setString(1,"value");

getResult方法的底层同样也是JDBC操作

// rs是结果集对象ResultSet
rs.getInt(colName);
rs.getInt(i)

TypeHandler继承体系

在MyBatis中JDBC类型被定义在枚举类JdbcType中。共有41中JDBC类型。每个JDBC类型都对应一个XxxTypeHandler类。每个类都是解决特定的JDBC类型转换

TypeHandler接口值规定了行为,每种JDBC类型,都提供了对应的实现类来完场JDBC到Java类型的转换。

每个TypeHandler的实现都大同小异,而为了避免空指针的问题,TypeHandler还有一个抽象子类,对这4个方法做了模板处理。

BaseTypeHandler的代码比较简单易懂,这列举重要实现:

public void setParameter(PreparedStatement ps, int i, T parameter, JdbcType jdbcType) throws SQLException {
  if (parameter == null) {
    if (jdbcType == null) {抛异常}
    try {
      ps.setNull(i, jdbcType.TYPE_CODE);
    } catch (SQLException e) { 抛异常 }
  } else {
    try {
      // 如果参数不为Null才调用子类方法
      setNonNullParameter(ps, i, parameter, jdbcType);
    } catch (Exception e) { 抛异常 }
  }
}

我们先来剖析下常用的类型。IntegerTypeHandler

IntegerTypeHandler

public class IntegerTypeHandler extends BaseTypeHandler<Integer> {
  @Override
  public void setNonNullParameter(PreparedStatement ps, int i, Integer parameter, JdbcType jdbcType) throws SQLException {
    ps.setInt(i, parameter);
  }
  public Integer getNullableResult(ResultSet rs, String columnName) throws SQLException {
    int result = rs.getInt(columnName);
    return result == 0 && rs.wasNull() ? null : result;
  }
  // 省略另外两个方法
}

可以看到IntegerTypeHandler的底层就是调用了JDBC对象PreparedStatement的方法来设置参数与获取结果集中的记录。这个实现是最简单的一个,就不展开介绍了。

下面我们再来看一下日期类型的映射

DateTypeHandler

public class DateTypeHandler extends BaseTypeHandler<Date> {
  @Override
  public void setNonNullParameter(PreparedStatement ps, int i, Date parameter, JdbcType jdbcType) throws SQLException {
    ps.setTimestamp(i, new Timestamp(parameter.getTime()));
  }
  @Override
  public Date getNullableResult(ResultSet rs, String columnName) throws SQLException {
    Timestamp sqlTimestamp = rs.getTimestamp(columnName);
    if (sqlTimestamp != null) {
      return new Date(sqlTimestamp.getTime());
    }
    return null;
  }
}

注:可能大家在编程JDBC的时候,执行如下语句select * from user where birthday > ?,无论数据库中birthday字段是Date/DateTime/TimeStamp类型,我们都会把会把?的值设置为Date类型,实际上这是不符合JDBC标准的。

其他TypeHandler的实现,就不再看了。逻辑都是一样的。

TypeHandlerRegistry

通过上一小节了解到了每一种JDBC类型,都提供了TypeHandler的实现。那么mybatis什么时候才会用到这些实现类呢?或者说当需要进行类型转换的时候,MyBatis是如何使用这些实现类的。

在MyBatis初始化的时候,会加载TypeHandlerRegistry这个类,这个类有在构造方法里会调用一系列的register方法把TypeHandler所有实现类都注册到TypeHandlerRegistry中。下面我们先来看一下TypeHandlerRegistry中的重要属性

public final class TypeHandlerRegistry {
  private final Map<JdbcType, TypeHandler<?>> jdbcTypeHandlerMap = new EnumMap<>(JdbcType.class);
  private final Map<Type, Map<JdbcType, TypeHandler<?>>> typeHandlerMap = new ConcurrentHashMap<>();
  private final TypeHandler<Object> unknownTypeHandler;
  private final Map<Class<?>, TypeHandler<?>> allTypeHandlersMap = new HashMap<>();
}

TypeHandlerRegistry#register方法

上面介绍了TypeHandlerRegistry中几个重要的属性。在mybatis启动时,就会把所有的TypeHandler都注册到如上的4个数据结构中。具体的实现在TypeHandlerRegistry的构造方法中,调用register方法进行注册TypeHandler

public TypeHandlerRegistry(Configuration configuration) {
  register(Boolean.class, new BooleanTypeHandler());
  register(boolean.class, new BooleanTypeHandler());
  register(JdbcType.BOOLEAN, new BooleanTypeHandler());
  register(JdbcType.BIT, new BooleanTypeHandler());
  register(Byte.class, new ByteTypeHandler());
  register(byte.class, new ByteTypeHandler());
  register(JdbcType.TINYINT, new ByteTypeHandler());
  register(Short.class, new ShortTypeHandler());
  register(short.class, new ShortTypeHandler());
  register(JdbcType.SMALLINT, new ShortTypeHandler());
  // ...省略其他register
}

这些register最终将如上的3个Map数据结构填充。通过名字也能看出来,注册Class/JdbcType与TypeHandler的映射关系。而这些register分为两大类,一类是注册JDBC-TypeHandler映射关系,一类是注册Java-TypeHandler的映射关系。我们先来看第一种

public void register(JdbcType jdbcType, TypeHandler<?> handler) {
  jdbcTypeHandlerMap.put(jdbcType, handler);
}

它的实现很简单,就是直接把jdbcType作为key,TypeHandler作为value,存入jdbcTypeHandlerMap即可。接下来我们再来看第二种:注册Java-TypeHandler的映射关系

private <T> void register(Type javaType, TypeHandler<? extends T> typeHandler) {
  MappedJdbcTypes mappedJdbcTypes = typeHandler.getClass().getAnnotation(MappedJdbcTypes.class);
  if (mappedJdbcTypes != null) {
    for (JdbcType handledJdbcType : mappedJdbcTypes.value()) {
      register(javaType, handledJdbcType, typeHandler);
    }
    if (mappedJdbcTypes.includeNullJdbcType()) {
      register(javaType, null, typeHandler);
    }
  } else {
    register(javaType, null, typeHandler);
  }
}

我们来分析下 注册步骤:

下面就来看一下这个重载register方法

private void register(Type javaType, JdbcType jdbcType, TypeHandler<?> handler) {
  if (javaType != null) {
    Map<JdbcType, TypeHandler<?>> map = typeHandlerMap.get(javaType);
    if (map == null || map == NULL_TYPE_HANDLER_MAP) {
      map = new HashMap<>();
    }
    map.put(jdbcType, handler);
    typeHandlerMap.put(javaType, map);
  }
  allTypeHandlersMap.put(handler.getClass(), handler);
}

代码中它主要做了两件事:

至此TypeHandlerRegistry中就有了最重要的两种映射关系

TypeHandlerRegistry也提供了一系列的get*方法,可以根据指定的信息返回需要的TypeHandler类

总结

TypeHandler是类型处理器,它用来解析Java类型与Jdbc类型之间的相互转换。

TypeHandlerRegistry是一个注册器,其中注册了JDBC与TypeHandler的映射关系、Java类型与TypeHandler的映射关系。

那么由此我们可以想象到,在mybatis执行SQL的过程中,一定会在某处调用TypeHandlerRegistry并通过参数的Java类型获取对应的TypeHandler对象为PreparedStatement设置参数。

也一定会在解析结果集的过程中,调用TypeHandlerRegistry并通过结果集数据的Jdbc类型获取对应的TypeHandler对象为来解析结果集中的记录

加载全部内容

相关教程
猜你喜欢
用户评论