快速开始 Mybatis TypeHandler

2023年 9月 6日 52.6k 0

MyBatis TypeHandler是MyBatis框架中的举足轻重的组件之一,用于处理Java对象和数据库中的数据类型之间的转换。

MyBatis TypeHandler 介绍

MyBatis TypeHandler是MyBatis框架中的一个重要组件,用于处理Java对象和数据库中的数据类型之间的转换。在MyBatis中,每个Java类型都需要对应一个TypeHandler,用于将Java类型转换为JDBC类型,或将JDBC类型转换为Java类型。

TypeHandler可以用于以下场景:

  • 数据库中的数据类型与Java类型不匹配,需要进行类型转换。
  • 数据库中的数据类型与Java类型匹配,但需要进行特殊处理,例如将List转换为String,或将Object转换为JSON字符串。
  • 数据库中的数据类型与Java类型匹配,但需要进行自定义转换,例如将Java的枚举类型转换为数据库中的字符串类型。
    使用TypeHandler可以简化代码,提高开发效率,同时也可以避免一些常见的错误,例如空指针异常、类型转换异常等。
  • 在MyBatis中,可以通过实现TypeHandler接口来自定义TypeHandler,也可以使用MyBatis提供的一些默认的TypeHandler,例如StringTypeHandler、IntegerTypeHandler等。

    List转换为String

    
    import org.apache.ibatis.type.JdbcType;
    import org.apache.ibatis.type.TypeHandler;
    import java.sql.CallableStatement;
    import java.sql.PreparedStatement;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import java.util.Collections;
    import java.util.List;
    import java.util.stream.Collectors;
    
    @MappedJdbcTypes(JdbcType.VARCHAR)
    @MappedTypes(List.class)
    @Slf4j
    public class ListToStringTypeHandler implements TypeHandler {
    
        @Override
        public void setParameter(PreparedStatement ps, int i, List parameter, JdbcType jdbcType) throws SQLException {
            if (parameter == null || parameter.isEmpty()) {
                ps.setString(i, "");
            } else {
                String value = parameter.stream().collect(Collectors.joining(","));
                ps.setString(i, value);
            }
        }
    
        @Override
        public List getResult(ResultSet rs, String columnName) throws SQLException {
            String value = rs.getString(columnName);
            return convertToList(value);
        }
    
        @Override
        public List getResult(ResultSet rs, int columnIndex) throws SQLException {
            String value = rs.getString(columnIndex);
            return convertToList(value);
        }
    
        @Override
        public List getResult(CallableStatement cs, int columnIndex) throws SQLException {
            String value = cs.getString(columnIndex);
            return convertToList(value);
        }
    
        private List convertToList(String value) {
            if (value == null || value.isEmpty()) {
                return Collections.emptyList();
            } else {
                return List.of(value.split(","));
            }
        }
    }
    

    代码中,实现了TypeHandler接口,并重写了setParameter、getResult等方法。其中,setParameter方法将List转换为String类型,并设置到PreparedStatement中;getResult方法从ResultSet或CallableStatement中获取String类型的值,并将其转换为List类型。

    需要注意的是,如果List为空或为null,需要将其转换为空字符串或空List,避免出现空指针异常或其他问题。在本示例代码中,使用了Java 8中的Stream API和Collections类来实现List和String之间的转换,可以根据需要进行修改。

    另外,还需要加入注解@MappedJdbcTypes(JdbcType.VARCHAR)
    @MappedTypes(List.class),这样,在MyBatis执行SQL语句时,就会自动调用TypeHandler来处理List和String之间的转换。

    Object 转换成 Json

    import com.fasterxml.jackson.core.JsonProcessingException;
    import com.fasterxml.jackson.databind.ObjectMapper;
    import org.apache.ibatis.type.JdbcType;
    import org.apache.ibatis.type.TypeHandler;
    import java.io.IOException;
    import java.sql.CallableStatement;
    import java.sql.PreparedStatement;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    
    public class ObjectToJsonTypeHandler implements TypeHandler {
    
        private static final ObjectMapper objectMapper = new ObjectMapper();
    
        @Override
        public void setParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
            if (parameter == null) {
                ps.setString(i, null);
            } else {
                try {
                    String value = objectMapper.writeValueAsString(parameter);
                    ps.setString(i, value);
                } catch (JsonProcessingException e) {
                    throw new SQLException("Failed to convert object to JSON", e);
                }
            }
        }
    
        @Override
        public Object getResult(ResultSet rs, String columnName) throws SQLException {
            String value = rs.getString(columnName);
            return convertToObject(value);
        }
    
        @Override
        public Object getResult(ResultSet rs, int columnIndex) throws SQLException {
            String value = rs.getString(columnIndex);
            return convertToObject(value);
        }
    
        @Override
        public Object getResult(CallableStatement cs, int columnIndex) throws SQLException {
            String value = cs.getString(columnIndex);
            return convertToObject(value);
        }
    
        private Object convertToObject(String value) throws SQLException {
            if (value == null) {
                return null;
            } else {
                try {
                    return objectMapper.readValue(value, Object.class);
                } catch (IOException e) {
                    throw new SQLException("Failed to convert JSON to object", e);
                }
            }
        }
    }
    

    代码中,实现了TypeHandler接口,并重写了setParameter、getResult等方法。其中,setParameter方法将Object转换为JSON字符串,并设置到PreparedStatement中;getResult方法从ResultSet或CallableStatement中获取JSON字符串,并将其转换为Object类型。

    需要注意的是,如果Object为null,需要将其转换为null字符串,避免出现空指针异常或其他问题。在本示例代码中,使用了Jackson库来实现Object和JSON字符串之间的转换,可以根据需要进行修改。

    使用

        @TableField(value = "keywords", typeHandler = ListStringTypeHandler.class)
        private List keywords;
    

    使用的方式就是在实体类中想要进行类型转换的字段加上注解@TableField,这是不管是设置值还是获取值的时候就会自动进行类型转换。

    相关文章

    JavaScript2024新功能:Object.groupBy、正则表达式v标志
    PHP trim 函数对多字节字符的使用和限制
    新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
    使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
    为React 19做准备:WordPress 6.6用户指南
    如何删除WordPress中的所有评论

    发布评论