mybatis

在Mybatis中,处理枚举类的TypeHandler有两个:

EnumTypeHandler: 用于保存枚举名
EnumOrdinalTypeHandler: 用于保存枚举的序号。
在实际项目中,以上往往不能满足我们的需求。

需求分析
枚举需要包含两个属性,label(用于显示), value(实际的枚举值)。数据库保存枚举值(value)。

这很明显Mybatis提供的两个枚举TypeHandler不能满足我们的需求。此时,我们可以自定义一个通用的枚举TypeHandler来满足我们的需求。

自定义枚举TypeHandler
通用枚举DisplayedEnum
public interface DisplayedEnum {

String DEFAULT_VALUE_NAME = "value";

String DEFAULT_LABEL_NAME = "label";

default Integer getValue() {
    Field field = ReflectionUtils.findField(this.getClass(), DEFAULT_VALUE_NAME);
    if (field == null)
        return null;
    try {
        field.setAccessible(true);
        return Integer.parseInt(field.get(this).toString());
    } catch (IllegalAccessException e) {
        throw new RuntimeException(e);
    }
}

@JsonValue
default String getLabel() {
    Field field = ReflectionUtils.findField(this.getClass(), DEFAULT_LABEL_NAME);
    if (field == null)
        return null;
    try {
        field.setAccessible(true);
        return field.get(this).toString();
    } catch (IllegalAccessException e) {
        throw new RuntimeException(e);
    }
}

static <T extends Enum<T>> T valueOfEnum(Class<T> enumClass, Integer value) {
    if (value == null)
        throw  new IllegalArgumentException("DisplayedEnum value should not be null");
    if (enumClass.isAssignableFrom(DisplayedEnum.class))
        throw new IllegalArgumentException("illegal DisplayedEnum type");
    T[] enums = enumClass.getEnumConstants();
    for (T t: enums) {
        DisplayedEnum displayedEnum = (DisplayedEnum)t;
        if (displayedEnum.getValue().equals(value))
            return (T) displayedEnum;
    }
    throw new IllegalArgumentException("cannot parse integer: " + value + " to " + enumClass.getName());
}

}
说明:
普通枚举类通过实现DisplayedEnum接口,就可以:

通过getValue()获取枚举值。
通过getLabel()获取枚举的label属性。
通过valueOfEnum()将Integer值转换为指定的枚举类型。
普通枚举类
public enum CommonsType implements DisplayedEnum {

NORMAL("正常", 0), INVALID("无效", 1);

String label;

Integer value;

private CommonsType(String label, Integer value) {
    this.label = label;
    this.value = value;
}

}
以上就是一个普通枚举类的示例。

自定义枚举TypeHandler
@MappedJdbcTypes(value = JdbcType.TINYINT, includeNullJdbcType = true)
public class DefaultEnumTypeHandler extends BaseTypeHandler {

private Class<DisplayedEnum> type;

public EnumTypeHandler(){};

public EnumTypeHandler(Class<DisplayedEnum> type) {
    if (type == null) throw new IllegalArgumentException("Type argument cannot be null");
    this.type = type;
}

@Override
public void setNonNullParameter(PreparedStatement ps, int i, DisplayedEnum parameter, JdbcType jdbcType)
        throws SQLException {
    ps.setInt(i, parameter.getValue());
}

@Override
public DisplayedEnum getNullableResult(ResultSet rs, String columnName) throws SQLException {
    return convert(rs.getInt(columnName));
}

@Override
public DisplayedEnum getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
    return convert(rs.getInt(columnIndex));
}

@Override
public DisplayedEnum getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
    return convert(cs.getInt(columnIndex));
}

private DisplayedEnum convert(int status){
    DisplayedEnum[] objs = type.getEnumConstants();
    for(DisplayedEnum em: objs){
        if(em.getValue() == status){
            return  em;
        }
    }
    return null;
}

}
使用我们自定义的DefaultEnumTypeHandler
由于Mybatis默认在处理枚举类型的时候会使用EnumTypeHandler(只保存及转换枚举类型的名字), 因此,我们需要手动指定使用DefaultEnumTypeHandler。示例如下:






即我们需要通过使用typeHandler来指定

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值