Project

General

Profile

Download (7.08 KB) Statistics
| Branch: | Tag: | Revision:
1
/**
2
* Copyright (C) 2007 EDIT
3
* European Distributed Institute of Taxonomy
4
* http://www.e-taxonomy.eu
5
*
6
* The contents of this file are subject to the Mozilla Public License Version 1.1
7
* See LICENSE.TXT at the top of this package for the full license terms.
8
*/
9

    
10
package eu.etaxonomy.cdm.hibernate;
11

    
12
import java.io.Serializable;
13
import java.sql.PreparedStatement;
14
import java.sql.ResultSet;
15
import java.sql.SQLException;
16
import java.sql.Types;
17
import java.util.Properties;
18

    
19
import org.apache.log4j.Logger;
20
import org.hibernate.HibernateException;
21
import org.hibernate.engine.spi.SessionImplementor;
22
import org.hibernate.type.StandardBasicTypes;
23
import org.hibernate.usertype.ParameterizedType;
24
import org.hibernate.usertype.UserType;
25
import org.jadira.usertype.dateandtime.shared.spi.AbstractUserType;
26

    
27
import eu.etaxonomy.cdm.model.common.AuthorityType;
28
import eu.etaxonomy.cdm.model.common.CdmClass;
29
import eu.etaxonomy.cdm.model.description.DescriptionType;
30
import eu.etaxonomy.cdm.model.media.ExternalLinkType;
31
import eu.etaxonomy.cdm.model.metadata.CdmMetaDataPropertyName;
32
import eu.etaxonomy.cdm.model.molecular.SequenceDirection;
33
import eu.etaxonomy.cdm.model.name.NomenclaturalCode;
34
import eu.etaxonomy.cdm.model.name.NomenclaturalCodeEdition;
35
import eu.etaxonomy.cdm.model.name.NomenclaturalStanding;
36
import eu.etaxonomy.cdm.model.name.RankClass;
37
import eu.etaxonomy.cdm.model.name.RegistrationStatus;
38
import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationType;
39
import eu.etaxonomy.cdm.model.permission.CRUD;
40
import eu.etaxonomy.cdm.model.permission.PermissionClass;
41
import eu.etaxonomy.cdm.model.reference.OriginalSourceType;
42
import eu.etaxonomy.cdm.model.reference.ReferenceType;
43
import eu.etaxonomy.cdm.model.taxon.ConceptDefinition;
44
import eu.etaxonomy.cdm.model.taxon.ConceptStatus;
45
import eu.etaxonomy.cdm.model.taxon.TaxonNodeStatus;
46
import eu.etaxonomy.cdm.model.taxon.TaxonType;
47
import eu.etaxonomy.cdm.model.term.IKeyTerm;
48
import eu.etaxonomy.cdm.model.term.TermType;
49

    
50
/**
51
 * User type for IEnumTerm
52
 * Partly copied from http://stackoverflow.com/questions/9839553/hibernate-map-enum-to-varchar
53
 * @author a.mueller
54
 * @since 15-07-2013
55
 */
56
public class EnumUserType<E extends Enum<E>>
57
        extends AbstractUserType
58
        implements UserType, ParameterizedType {
59

    
60
    private static final long serialVersionUID = 4641078915907621907L;
61
	@SuppressWarnings("unused")
62
	private static final Logger logger = Logger.getLogger(EnumUserType.class);
63

    
64
    private static final int[] SQL_TYPES = { Types.VARCHAR };
65

    
66
	private Class<E> clazz = null;
67

    
68
	public EnumUserType(){}
69

    
70
    public EnumUserType(Class<E> c) {
71
    	this.clazz = c;
72
    }
73

    
74
	@Override
75
	@SuppressWarnings("unchecked")
76
	public void setParameterValues(Properties parameters) {
77
		try {
78
			this.clazz = (Class<E>) Class.forName(parameters.getProperty("enumClass"));
79
		} catch (ClassNotFoundException e) {
80
			throw new RuntimeException(e);
81
		}
82
	}
83

    
84
	@Override
85
	public Object deepCopy(Object o) throws HibernateException {
86
		return o;
87
	}
88

    
89
	@Override
90
	public Serializable disassemble(Object value) throws HibernateException {
91
	    return (Serializable)value;
92
	}
93

    
94
	@Override
95
	public IKeyTerm nullSafeGet(ResultSet rs, String[] names, SessionImplementor session, Object owner)
96
			throws HibernateException, SQLException {
97
        String val = (String) StandardBasicTypes.STRING.nullSafeGet(rs, names, session, owner);
98

    
99
		if(val == null) {
100
			return null;
101
		} else {
102

    
103
		    return getTerm(clazz, val);
104
		}
105
	}
106

    
107
    public static <E extends Enum<E>> IKeyTerm getTerm(Class<E> clazz, String val) {
108
        // TermType
109
        if (clazz.equals(TermType.class)){
110
        	return TermType.getByKey(val);
111
        //Reference Type
112
        }else if (clazz.equals(ReferenceType.class)){
113
        		return ReferenceType.getByKey(val);
114
        //OriginalSourceType
115
        }else if (clazz.equals(OriginalSourceType.class)){
116
        	return OriginalSourceType.getByKey(val);
117
        //NomenclaturalCode
118
        }else if (clazz.equals(NomenclaturalCode.class)){
119
        	return NomenclaturalCode.getByKey(val);
120
        //NomenclaturalCode
121
        }else if (clazz.equals(NomenclaturalCodeEdition.class)){
122
            return NomenclaturalCodeEdition.getByKey(val);
123
        //RankClass
124
        }else if (clazz.equals(RankClass.class)){
125
        	return RankClass.getByKey(val);
126
        //SpecimenOrObservationType
127
        }else if (clazz.equals(SpecimenOrObservationType.class)){
128
        	return SpecimenOrObservationType.getByKey(val);
129
        //SequenceDirection
130
        }else if (clazz.equals(SequenceDirection.class)){
131
        	return SequenceDirection.getByKey(val);
132
        //RegistrationStatus
133
        }else if (clazz.equals(RegistrationStatus.class)){
134
            return RegistrationStatus.getByKey(val);
135
        //CdmMetaDataPropertyName
136
        }else if (clazz.equals(CdmMetaDataPropertyName.class)){
137
            return CdmMetaDataPropertyName.getByKey(val);
138
        //EntityAuthority
139
        }else if (clazz.equals(AuthorityType.class)){
140
            return AuthorityType.getByKey(val);
141
        //ExternalLinkType
142
        }else if (clazz.equals(ExternalLinkType.class)){
143
            return ExternalLinkType.getByKey(val);
144
        //PermissionClass
145
        }else if (clazz.equals(PermissionClass.class)){
146
            return PermissionClass.getByKey(val);
147
        //CRUD
148
        }else if (clazz.equals(CRUD.class)){
149
            return CRUD.getByKey(val);
150
        //TaxonNodeStatus
151
        }else if (clazz.equals(TaxonNodeStatus.class)){
152
            return TaxonNodeStatus.getByKey(val);
153
        //DescriptionType
154
        }else if (clazz.equals(DescriptionType.class)){
155
            return DescriptionType.getByKey(val);
156
        //NomenclaturalStanding
157
        }else if (clazz.equals(NomenclaturalStanding.class)){
158
            return NomenclaturalStanding.getByKey(val);
159
        //CdmClass
160
        }else if (clazz.equals(CdmClass.class)){
161
            return CdmClass.getByKey(val);
162
        //CdmClass
163
        }else if (clazz.equals(ConceptDefinition.class)){
164
            return ConceptDefinition.getByKey(val);
165
        //CdmClass
166
        }else if (clazz.equals(ConceptStatus.class)){
167
            return ConceptStatus.getByKey(val);
168
        //CdmClass
169
        }else if (clazz.equals(TaxonType.class)){
170
            return TaxonType.getByKey(val);
171
        }else{
172
        	throw new IllegalArgumentException(String.format("EnumType %s not supported by %s.", clazz.getSimpleName(), EnumUserType.class.getSimpleName()));
173
        }
174
    }
175

    
176
	@Override
177
	public void nullSafeSet(PreparedStatement statement, Object value, int index, SessionImplementor session)
178
			throws HibernateException, SQLException {
179
		if (value == null) {
180
            StandardBasicTypes.STRING.nullSafeSet(statement, value, index, session);
181
        } else {
182
        	IKeyTerm term = (IKeyTerm)value;
183
            StandardBasicTypes.STRING.nullSafeSet(statement, term.getKey(), index, session);
184
        }
185
	}
186

    
187
	@Override
188
	public Class<E> returnedClass() {
189
		return clazz;
190
	}
191

    
192
	@Override
193
	public int[] sqlTypes() {
194
		return SQL_TYPES;
195
	}
196
}
(4-4/13)