package eu.dnetlib.openaire.common;

import java.io.Serializable;
import java.sql.*;

import org.hibernate.HibernateException;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.usertype.UserType;

/**
 * Created by claudio on 05/07/2017.
 */
public class GenericArrayUserType<T extends Serializable> implements UserType {

	protected static final int[] SQL_TYPES = { Types.ARRAY };

	private  Class<T> typeParameterClass;

	@Override
	public Object assemble(Serializable cached, Object owner) throws HibernateException {
		return this.deepCopy(cached);
	}

	@Override
	public Object deepCopy(Object value) throws HibernateException {
		return value;
	}

	@SuppressWarnings("unchecked")
	@Override
	public Serializable disassemble(Object value) throws HibernateException {
		return (T) this.deepCopy(value);
	}

	@Override
	public boolean equals(Object x, Object y) throws HibernateException {

		if (x == null) {
			return y == null;
		}
		return x.equals(y);
	}

	@Override
	public int hashCode(Object x) throws HibernateException {
		return x.hashCode();
	}

	@Override
	public Object nullSafeGet(final ResultSet resultSet,
			final String[] names,
			final SharedSessionContractImplementor sharedSessionContractImplementor,
			final Object o)
			throws HibernateException, SQLException {
		if (resultSet.wasNull()) {
			return null;
		}
		if (resultSet.getArray(names[0]) == null) {
			return new Integer[0];
		}

		Array array = resultSet.getArray(names[0]);
		@SuppressWarnings("unchecked")
		T javaArray = (T) array.getArray();
		return javaArray;
	}

	@Override
	public void nullSafeSet(final PreparedStatement statement,
			final Object value,
			final int index,
			final SharedSessionContractImplementor session)
			throws HibernateException, SQLException {
		Connection connection = statement.getConnection();
		if (value == null) {
			statement.setNull(index, SQL_TYPES[0]);
		} else {
			@SuppressWarnings("unchecked")
			T castObject = (T) value;
			Array array = connection.createArrayOf("integer", (Object[]) castObject);
			statement.setArray(index, array);
		}
	}

	@Override
	public boolean isMutable() {
		return true;
	}

	@Override
	public Object replace(Object original, Object target, Object owner) throws HibernateException {
		return original;
	}

	@Override
	public Class<T> returnedClass() {
		return typeParameterClass;
	}

	@Override
	public int[] sqlTypes() {
		return new int[] { Types.ARRAY };
	}


}