Hibernate 實現Enum轉換爲Smallint

由於數據庫的原因,需要用smallint來存儲枚舉類型.

Hibernate 3.0 以上也支持 Enum類型的轉換,這裏以Smallint爲例(當然,也可以轉換爲其他類型,如varchar).

 

首先,以下是一個枚舉類型

1. public enum ConsumerType
2. {
3.     Admin,Vistor,VIP;
4.     
5. }

然後,再寫一個模板,實現UserType類型接口,這樣就可以將以後寫的枚舉類型都轉換爲Smallint存入數據庫

 

1. import java.io.Serializable;
2. import java.sql.PreparedStatement;
3. import java.sql.ResultSet;
4. import java.sql.SQLException;
5. import java.sql.Types;
7. import org.hibernate.HibernateException;
8. import org.hibernate.usertype.UserType;
11. public class IntEnumUserType<E extends Enum<E>> implements UserType 
12. { 
13.     private Class<E> clazz = null;
14.     private E[] theEnumValues;
17.     protected IntEnumUserType(Class<E> c, E[] e) 
18.     { 
19.         this.clazz = c; 
20.         this.theEnumValues = e;
21.     } 
23.     private static final int[] SQL_TYPES = {Types.SMALLINT};
26.     public int[] sqlTypes() 
27.     { 
28.         return SQL_TYPES; 
29.     } 
31.     public Class<E> returnedClass() 
32.     { 
33.         return clazz; 
34.     } 
37.     public Object nullSafeGet(ResultSet resultSet, String[] names, Object owner) 
38.     throws HibernateException, SQLException 
39.     { 
40.         final int val = resultSet.getShort(names[0]);
41.         E result = null;
42.         if (!resultSet.wasNull()) 
43.         {
44.             try 
45.             {
46.                 for(int i=0; i < theEnumValues.length && result == null; i++) 
47.                 {
48.                     if (theEnumValues[i].ordinal() == val) 
49.                     {
50.                         result = theEnumValues[i];
51.                     }
52.                 }
53.             } 
54.             catch (SecurityException e) 
55.             {
56.                 result = null;
57.             } 
58.             catch (IllegalArgumentException e) 
59.             {
60.                 result = null;
61.             }
62.         } 
63.         return result; 
64.     } 
67.     public void nullSafeSet(PreparedStatement preparedStatement, 
68.             Object value, int index) throws HibernateException, SQLException 
69.             { 
70.         if (null == value) 
71.         { 
72.             preparedStatement.setNull(index, Types.SMALLINT); 
73.         } 
74.         else 
75.         { 
76.             preparedStatement.setInt(index, ((Enum)value).ordinal()); 
77.         } 
78.             } 
80.     public Object deepCopy(Object value) throws HibernateException
81.     { 
82.         return value; 
83.     } 
85.     public boolean isMutable() 
86.     { 
87.         return false; 
88.     } 
90.     public Object assemble(Serializable cached, Object owner) throws HibernateException 
91.     {
92.         return cached;
93.     } 
95.     public Serializable disassemble(Object value) throws HibernateException 
96.     { 
97.         return (Serializable)value; 
98.     } 
100.     public Object replace(Object original, Object target, Object owner) throws HibernateException 
101.     { 
102.         return original; 
103.     } 
104.     public int hashCode(Object x) throws HibernateException 
105.     { 
106.         return x.hashCode(); 
107.     } 
108.     public boolean equals(Object x, Object y) throws HibernateException
109.     { 
110.         if (x == y) 
111.             return true; 
112.         if (null == x || null == y) 
113.             return false; 
114.         return x.equals(y); 
115.     } 
116. } 

跟着,新建一個類來實現模板

 

1. /** This class is used only in the hibernate XML configuration */
2. public class ConsumerTypeEnum extends IntEnumUserType<ConsumerType> 
3. { 
4.     public ConsumerTypeEnum() 
5.     { 
6.         // we must give the values of the enum to the parent.
7.         super(ConsumerType.class,ConsumerType.values()); 
8.     } 
9. }

最後,在mapping的xml文件中,使用如下代碼

 

1. <property column="USERTPYE" 
2. not-null="true" 
3. name="consumerType"
4. type="ConsumerTypeEnum" />

注意,這裏是使用 ConsumerTypeEnum 而不是 ConsumerType

 

到此,轉換完成~