Mybatis 枚举自定义转换

r囧r小猫 2022-05-27 00:48 272阅读 0赞

定义Hander 实现

import java.sql.CallableStatement;
    import java.sql.PreparedStatement;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import org.apache.ibatis.type.BaseTypeHandler;
    import org.apache.ibatis.type.JdbcType;
    import org.apache.ibatis.type.MappedJdbcTypes;
    import org.apache.ibatis.type.MappedTypes;
    import xxx.CouponProvider;
    
    @MappedJdbcTypes(value = {JdbcType.TINYINT})
    @MappedTypes({CouponProvider.class})
    public class EnumTypeHander extends BaseTypeHandler<CouponProvider> {
    
      @Override
      public void setNonNullParameter(PreparedStatement ps, int i, CouponProvider parameter,
          JdbcType jdbcType) throws SQLException {
        // 枚举类转换到number类型，设置到Statement中
        ps.setInt(i, parameter.getNumber());
      }
    
      @Override
      public CouponProvider getNullableResult(ResultSet rs, String columnName) throws SQLException {
        // 将Number数值转到到枚举类型
        return CouponProvider.forNumber(rs.getInt(columnName));
      }
    
      @Override
      public CouponProvider getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        // 将Number数值转到到枚举类型
        return CouponProvider.forNumber(rs.getInt(columnIndex));
      }
    
      @Override
      public CouponProvider getNullableResult(CallableStatement cs, int columnIndex)
          throws SQLException {
        // 将Number数值转到到枚举类型
        return CouponProvider.forNumber(cs.getInt(columnIndex));
      }
    
    }

Mybatis-mapper.xml 定义

<insert id="save" parameterType="xxx.Model">
    	INSERT INTO promotion (name,type)
    	VALUES(#{name},#{type,javaType=xxx.EnumType,jdbcType=TINYINT,typeHandler=xxx.EnumTypeHander})
    	</insert>
    	<resultMap type="xxx.model.xxModel" id="resultMap">
    		<id column="id" property="id" />
    		<result column="type" property="type" typeHandler="xxx.EnumTypeHander" />
    	</resultMap>
    	<select id="getById" resultMap="resultMap">
    		SELECT * FROM xxx where id=#{id}
    	</select>