mybatis枚举支持通过value映射

ゞ浴缸里的玫瑰 2022-10-11 15:58 145阅读 0赞

背景  
之前ebean简单的 @DbEnumValue 注解就能支持通过value映射。  
mybatis也有支持，是 @EnumValue注解。分别支持二种映射规则。  
EnumOrdinalTypeHandler和EnumTypeHandler  
EnumTypeHandler(默认)作用：

@Override
      public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException {
        
        if (jdbcType == null) {
        
          ps.setString(i, parameter.name());
        } else {
        
          ps.setObject(i, parameter.name(), jdbcType.TYPE_CODE); // see r3589
        }
      }

将枚举对象的name当成数据库value传入。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70]  
可以看到PreparedStatement  
![在这里插入图片描述][20210701191549889.png]

![在这里插入图片描述][20210701191501670.png]  
而EnumOrdinalTypeHandler则是将枚举对象的ordinal传入。  
![在这里插入图片描述][20210701191658673.png]

公司的枚举规则是name大写，value小写。这里自定义枚举映射规则

public class MyEnumTypeHandler<E extends Enum<E>> extends EnumTypeHandler<E> {
        
        private final Class<E> type;
        public MyEnumTypeHandler(Class type) {
        
            super(type);
            this.type=type;
        }
        @Override
        public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException {
        
            if (jdbcType == null) {
        
                ps.setString(i, parameter.name().toLowerCase());
            } else {
        
                // see r3589
                ps.setObject(i, parameter.name().toLowerCase(), jdbcType.TYPE_CODE);
            }
        }
    
        @Override
        public E getNullableResult(ResultSet rs, String columnName) throws SQLException {
        
            String s = rs.getString(columnName);
            return s == null ? null : Enum.valueOf(type, s.toUpperCase());
        }
    }

然后apoll配置

![在这里插入图片描述][20210701191825132.png]  
亲测可用。

注意我这里的枚举类要求，name和value一样。但是name是大写，value是小写。偷个懒，懒得写反射。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70]: /images/20221005/a7c366b59555420d84b024583844ebfe.png
[20210701191549889.png]: /images/20221005/1ce48eec7dfb4d8ebd248106684b1c40.png
[20210701191501670.png]: /images/20221005/e9d584a7a05d4f269b03cd04fe451189.png
[20210701191658673.png]: /images/20221005/49f4a95c71e84ed9b0cdd80fb842388b.png
[20210701191825132.png]: /images/20221005/cddb935447c44a93b444f0c68fd62a09.png