it.amattioli.dominate.hibernate.specifications.HqlEnumSpecification Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dominate-hibernate Show documentation
The newest version!
package it.amattioli.dominate.hibernate.specifications;

import java.util.Properties;

import org.hibernate.Hibernate;
import org.hibernate.Query;
import org.hibernate.type.Type;
import org.hibernate.usertype.UserType;

import it.amattioli.dominate.Entity;
import it.amattioli.dominate.hibernate.types.EnumUserType;
import it.amattioli.dominate.specifications.Assembler;
import it.amattioli.dominate.specifications.EnumSpecification;

import static it.amattioli.dominate.hibernate.specifications.HqlUtils.*;

public class HqlEnumSpecification, U extends Enum> extends EnumSpecification {
	private String alias;
	private Type hibernateType;
	
	public HqlEnumSpecification() {
		
	}
	
	public HqlEnumSpecification(String propertyName, Class enumClass) {
		super(propertyName, enumClass);
	}
	
	public HqlEnumSpecification(String propertyName, Class enumClass, String alias) {
		this(propertyName, enumClass);
		this.alias = alias;
	}

	public HqlEnumSpecification(String propertyName, Class enumClass, Class hibernateType) {
		super(propertyName, enumClass);
		this.hibernateType = Hibernate.custom(hibernateType);
	}
	
	public HqlEnumSpecification(String propertyName, Class enumClass, Class hibernateType, String alias) {
		this(propertyName, enumClass, hibernateType);
		this.alias = alias;
	}
	
	@Override
	protected void setEnumClass(Class enumClass) {
		super.setEnumClass(enumClass);
		Properties props = new Properties();
		props.put(EnumUserType.ENUM_CLASS_PARAM, enumClass.getName());
		this.hibernateType = Hibernate.custom(EnumUserType.class, props);
	}
	
	@Override
	public void itselfAssembleQuery(Assembler assembler) {
		addHqlCondition((HqlAssembler)assembler);
		setHqlParam((HqlAssembler)assembler);
	}
	
	public boolean itselfSupportsAssembler(Assembler assembler) {
		return assembler instanceof HqlAssembler;
	}

	public void addHqlCondition(HqlAssembler assembler) {
		if (wasSet()) {
	        assembler.newCriteria();
	        if (alias != null) {
	        	assembler.append(alias).append(".");
	        } else {
            	assembler.append(assembler.getAliasPrefix());
            }
	        assembler.append(hqlPropertyName(getPropertyName()))
	                 .append(" like :")
	                 .append(normalizedPropertyName(getPropertyName()))
	                 .append(" ");
		}
    }
	
	public void setHqlParam(HqlAssembler assembler) {
		if (wasSet()) {
	    	assembler.addParameter(new HqlAssembler.ParameterSetter() {
	
				@Override
				public void setParameter(Query query) {
					query.setParameter(normalizedPropertyName(getPropertyName()), getValue(), hibernateType);
				}
	    		
	    	});
		}
    }

}