Project

General

Profile

Download (6.57 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.TaxonNodeStatus;
44
import eu.etaxonomy.cdm.model.term.IKeyTerm;
45
import eu.etaxonomy.cdm.model.term.TermType;
46

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

    
57
    private static final long serialVersionUID = 4641078915907621907L;
58
	@SuppressWarnings("unused")
59
	private static final Logger logger = Logger.getLogger(EnumUserType.class);
60

    
61
    private static final int[] SQL_TYPES = { Types.VARCHAR };
62

    
63
	private Class<E> clazz = null;
64

    
65
	public EnumUserType(){}
66

    
67
    public EnumUserType(Class<E> c) {
68
    	this.clazz = c;
69
    }
70

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

    
81
	@Override
82
	public Object deepCopy(Object o) throws HibernateException {
83
		return o;
84
	}
85

    
86
	@Override
87
	public Serializable disassemble(Object value) throws HibernateException {
88
	    return (Serializable)value;
89
	}
90

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

    
96
		if(val == null) {
97
			return null;
98
		} else {
99

    
100
		    return getTerm(clazz, val);
101
		}
102
	}
103

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

    
164
	@Override
165
	public void nullSafeSet(PreparedStatement statement, Object value, int index, SessionImplementor session)
166
			throws HibernateException, SQLException {
167
		if (value == null) {
168
            StandardBasicTypes.STRING.nullSafeSet(statement, value, index, session);
169
        } else {
170
        	IKeyTerm term = (IKeyTerm)value;
171
            StandardBasicTypes.STRING.nullSafeSet(statement, term.getKey(), index, session);
172
        }
173
	}
174

    
175
	@Override
176
	public Class<E> returnedClass() {
177
		return clazz;
178
	}
179

    
180
	@Override
181
	public int[] sqlTypes() {
182
		return SQL_TYPES;
183
	}
184
}
(4-4/13)