package rewards.internal.restaurant;
import java.io.Serializable;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import org.hibernate.HibernateException;
import org.hibernate.usertype.UserType;
public class BenefitAvailabilityPolicyUserType implements UserType {
public Object assemble(Serializable cached, Object owner) throws HibernateException {
return getBenefitAvailabiltyPolicy((String) cached);
}
public Object deepCopy(Object value) throws HibernateException {
return value;
}
public Serializable disassemble(Object value) throws HibernateException {
BenefitAvailabilityPolicy policy = (BenefitAvailabilityPolicy) value;
return getPolicyCode(policy);
}
public boolean equals(Object x, Object y) throws HibernateException {
return x.equals(y);
}
public int hashCode(Object x) throws HibernateException {
return x.hashCode();
}
public boolean isMutable() {
return false;
}
public Object nullSafeGet(ResultSet rs, String[] names, Object owner) throws HibernateException, SQLException {
String policyCode = rs.getString(names[0]);
return getBenefitAvailabiltyPolicy(policyCode);
}
public void nullSafeSet(PreparedStatement st, Object value, int index) throws HibernateException, SQLException {
if (value == null) {
st.setNull(index, Types.VARCHAR);
} else {
BenefitAvailabilityPolicy policy = (BenefitAvailabilityPolicy) value;
st.setString(index, getPolicyCode(policy));
}
}
public Object replace(Object original, Object target, Object owner) throws HibernateException {
return original;
}
@SuppressWarnings("rawtypes")
public Class returnedClass() {
return BenefitAvailabilityPolicy.class;
}
public int[] sqlTypes() {
return new int[] { Types.VARCHAR };
}
private BenefitAvailabilityPolicy getBenefitAvailabiltyPolicy(String policyCode) {
if ("A".equals(policyCode)) {
return AlwaysAvailable.INSTANCE;
} else if ("N".equals(policyCode)) {
return NeverAvailable.INSTANCE;
} else {
throw new IllegalArgumentException("Not a supported policy code " + policyCode);
}
}
private String getPolicyCode(BenefitAvailabilityPolicy policy) {
if (AlwaysAvailable.INSTANCE.equals(policy)) {
return "A";
} else if (NeverAvailable.INSTANCE.equals(policy)) {
return "N";
} else {
throw new IllegalArgumentException("Not a supported policy " + policy.toString());
}
}
}