快速开始 Mybatis TypeHandler
MyBatis TypeHandler是MyBatis框架中的举足轻重的组件之一,用于处理Java对象和数据库中的数据类型之间的转换。
MyBatis TypeHandler 介绍
MyBatis TypeHandler是MyBatis框架中的一个重要组件,用于处理Java对象和数据库中的数据类型之间的转换。在MyBatis中,每个Java类型都需要对应一个TypeHandler,用于将Java类型转换为JDBC类型,或将JDBC类型转换为Java类型。
TypeHandler可以用于以下场景:
使用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
,这是不管是设置值还是获取值的时候就会自动进行类型转换。