Created
March 19, 2015 14:25
-
-
Save drei01/44a09e559e4b370dda17 to your computer and use it in GitHub Desktop.
Hibernate Postgres Json User Type
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import java.io.IOException; | |
import java.io.Serializable; | |
import java.sql.PreparedStatement; | |
import java.sql.ResultSet; | |
import java.sql.SQLException; | |
import java.sql.Types; | |
import java.util.Properties; | |
import org.hibernate.HibernateException; | |
import org.hibernate.engine.spi.SessionImplementor; | |
import org.hibernate.internal.util.ReflectHelper; | |
import org.hibernate.usertype.ParameterizedType; | |
import org.hibernate.usertype.UserType; | |
import com.fasterxml.jackson.core.JsonProcessingException; | |
import com.fasterxml.jackson.databind.ObjectMapper; | |
import com.google.common.base.Objects; | |
/** | |
* Hibernate {@link UserType} implementation to handle JSON objects | |
* | |
* @see https://docs.jboss.org/hibernate/orm/4.1/javadocs/org/hibernate/usertype/ | |
* UserType.html | |
*/ | |
public class JSONUserType implements UserType, ParameterizedType, Serializable { | |
private static final long serialVersionUID = 1L; | |
private static final ObjectMapper MAPPER = new ObjectMapper(); | |
private static final String CLASS_TYPE = "classType"; | |
private static final int[] SQL_TYPES = new int[] { Types.JAVA_OBJECT }; | |
private Class<?> classType; | |
private int sqlType = Types.LONGVARCHAR; // before any guessing | |
@Override | |
public void setParameterValues(Properties params) { | |
String classTypeName = params.getProperty(CLASS_TYPE); | |
try { | |
this.classType = ReflectHelper.classForName(classTypeName, this.getClass()); | |
} catch (ClassNotFoundException cnfe) { | |
throw new HibernateException("classType not found", cnfe); | |
} | |
this.sqlType = Types.OTHER; | |
} | |
@Override | |
public Object assemble(Serializable cached, Object owner) throws HibernateException { | |
return this.deepCopy(cached); | |
} | |
@Override | |
public Object deepCopy(Object value) throws HibernateException { | |
Object copy = null; | |
if (value != null) { | |
try { | |
return MAPPER.readValue(MAPPER.writeValueAsString(value), this.classType); | |
} catch (IOException e) { | |
throw new HibernateException("unable to deep copy object", e); | |
} | |
} | |
return copy; | |
} | |
@Override | |
public Serializable disassemble(Object value) throws HibernateException { | |
try { | |
return MAPPER.writeValueAsString(value); | |
} catch (JsonProcessingException e) { | |
throw new HibernateException("unable to disassemble object", e); | |
} | |
} | |
@Override | |
public boolean equals(Object x, Object y) throws HibernateException { | |
return Objects.equal(x, y); | |
} | |
@Override | |
public int hashCode(Object x) throws HibernateException { | |
return Objects.hashCode(x); | |
} | |
@Override | |
public boolean isMutable() { | |
return true; | |
} | |
@Override | |
public Object nullSafeGet(ResultSet rs, String[] names, SessionImplementor session, Object owner) throws HibernateException, SQLException { | |
Object obj = null; | |
if (!rs.wasNull()) { | |
try { | |
obj = MAPPER.readValue(rs.getString(names[0]), this.classType); | |
} catch (IOException e) { | |
throw new HibernateException("unable to read object from result set", e); | |
} | |
} | |
return obj; | |
} | |
@Override | |
public void nullSafeSet(PreparedStatement st, Object value, int index, SessionImplementor session) throws HibernateException, SQLException { | |
if (value == null) { | |
st.setNull(index, this.sqlType); | |
} else { | |
try { | |
st.setObject(index, MAPPER.writeValueAsString(value), this.sqlType); | |
} catch (JsonProcessingException e) { | |
throw new HibernateException("unable to set object to result set", e); | |
} | |
} | |
} | |
@Override | |
public Object replace(Object original, Object target, Object owner) throws HibernateException { | |
return this.deepCopy(original); | |
} | |
@Override | |
public Class<?> returnedClass() { | |
return this.classType; | |
} | |
@Override | |
public int[] sqlTypes() { | |
return SQL_TYPES; | |
} | |
} |
When registering the type as "typeContributions.contributeType(new JSONUserType(), new String[]{"json"});" in the Dialect, the parameters will not be passed to the instance :/ But I think thats an hibernate issue. Or does someone has a solution for that.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Your nullSafeGet has a bug. The rs.wasNull() function needs to be called after the value was read. The javadoc states:
Reports whether the last column read had a value of SQL NULL. Note that you must first call one of the getter methods on a column to try to read its value and then call the method wasNull to see if the value read was SQL NULL.
However in your code you do not read the column first. To fix simple change this to: