用Hibernate映射INT字段到EnumSet

上次提到了把数据库的INT字段映射到Java里的Enum，但这样的字段只表示保存一个值。以前用C时经常会写这样的代码：

const int TYPE_1 = 1;
const int TYPE_2 = 2;
const int TYPE_3 = 4;

int type = TYPE_1 | TYPE_3;

if (type & (TYPE_1 | TYPE_3)) {
   ...
}

这样我可以在type变量里保存多个boolean类型的option。在Java里面类似的功能可以通过EnumSet实现，但是我还是希望在数据库里只用一个INT字段就可以保存整个EnumSet，自然还是通过自定义一个UserType来实现。

以上一篇文章里的IntegerValuedEnumType基础，再增加两个方法，encode用来把EnumSet转换为Integer，decode用来把Integer解码为EnumSet。其他只需要稍加修改即可。

/*
 * Encode the EnumSet into an integer based on bit on/off
 */
private int encode(EnumSet<T> set) {
    int ret = 0;
    for (T val : set) {
        ret |= 1 << val.getCode();
    }
    return ret;
}

/*
 * Decode the integer back to a EnumSet based on bit on/off
 */
private EnumSet<T> decode(int code) {
	Map<Integer, T> codeMap = new HashMap<Integer, T>();
	for (T val : EnumSet.allOf(enumClass)) {
		codeMap.put(val.getCode(), val);
	}
	  
    EnumSet<T> result = EnumSet.noneOf(enumClass);
    while (code != 0) {
        int ordinal = Integer.numberOfTrailingZeros(code);
        code ^= Integer.lowestOneBit(code);
        result.add(codeMap.get(ordinal));
    }
    return result;
}

相关文章：
用Hibernate映射INT字段到枚举类型：http://voidtech.iteye.com/blog/641347