MyBatis Plus 自动类型转换之TypeHandler

自动类型转换

数据库里有一个ids字段，里面存放的都是一些其他表的主键id，逗号分隔。

按照我一开始的解决思路，转换成String，然后再分割一下，转换成Set。

git提交代码后，老大审查代码，摇头。才了解到MyBatis有一个TypeHandler.

上代码：

@MappedTypes(Set.class)
@MappedJdbcTypes(JdbcType.VARCHAR)
public class SetTypeHandler extends BaseTypeHandler<Set> {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Set parameter, JdbcType jdbcType)
            throws SQLException {
        String param = parameter.toString().replaceAll("\\[|\\]| ", "");
        ps.setString(i, param);
    }

    @Override
    public Set<Integer> getNullableResult(ResultSet rs, String columnName)
            throws SQLException {
        String sqlSet = rs.getString(columnName);
        return getSet(sqlSet);
    }

    @Override
    public Set<Integer> getNullableResult(ResultSet rs, int columnIndex)
            throws SQLException {
        String sqlSet = rs.getString(columnIndex);
        return getSet(sqlSet);
    }

    @Override
    public Set<Integer> getNullableResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        String sqlSet = cs.getString(columnIndex);
        return getSet(sqlSet);
    }

    private Set<Integer> getSet(String sqlSet) {
        if (StringUtils.isNotBlank(sqlSet)) {
            return Arrays.asList(sqlSet.split(",")).stream()
                    .map(Integer::parseInt)
                    .collect(Collectors.toSet());
        }
        return new HashSet();
    }
}

以代码为例解释一下：

1. 实体类要转换的类型：Set
   @MappedTypes(Set.class)
   对应数据库表的字段类型：varchar
   @MappedJdbcTypes(JdbcType.VARCHAR)

2. 一个set，用来给PreparedStatement对象对应的列设置参数，预编译SQL语句，在SQL语句执行之前改变语句。先去掉Set的“[]”就变成了类似“ 1，2，3，4”，然后替换进SQL语句，这是完成了Set到varchar的转换。
   public void setNonNullParameter(PreparedStatement ps, int i, Set parameter, JdbcType jdbcType)

3. 两个大方向get,从ResultSet和CallableStatement获取对应列的值,不同之处是一个是取第几个位置的值，一个是取具体列名所对应的值。
   public Set<Integer> getNullableResult(ResultSet rs, String columnName)
public Set<Integer> getNullableResult(ResultSet rs, int columnIndex)
   上面两个方法是从ResultSet获取值，一个是通过列名，一个是第几个位置
   public Set<Integer> getNullableResult(CallableStatement cs, int columnIndex)

另外：

• Statement接口提供了执行语句和获取结果的基本方法；
• PreparedStatement接口添加了处理输入参数的方法；
• CallableStatement接口添加了调用存储过程核函数以及处理输出参数的方法。

配置好了handler，实体类也需要配置一下。

1. 实体类需要加个注解
   @TableName(autoResultMap = true)
2. 对应对象的属性也需要加注解
   @TableField(typeHandler = SetTypeHandler.class)