AndPredicateCreator.java

/*
 * Copyright Codecrew Ry
 * 
 * All rights reserved.
 * 
 * This license applies to any software containing a notice placed by the 
 * copyright holder. Such software is herein referred to as the Software. 
 * This license covers modification, distribution and use of the Software. 
 * 
 * Any distribution and use in source and binary forms, with or without 
 * modification is not permitted without explicit written permission from the 
 * copyright owner. 
 * 
 * A non-exclusive royalty-free right is granted to the copyright owner of the 
 * Software to use, modify and distribute all modifications to the Software in 
 * future versions of the Software. 
 * 
 */
package fi.codecrew.moya.facade.callbacks;

import java.util.Collections;
import java.util.List;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Predicate;
import javax.persistence.metamodel.SingularAttribute;

import fi.codecrew.moya.utilities.jpa.FacadeCallback;
import fi.codecrew.moya.utilities.jpa.ModelInterface;

public class AndPredicateCreator<A, T extends ModelInterface> implements FacadeCallback<T> {
	private final A searchval;
	private final List<SingularAttribute<T, A>> attributes;

	public AndPredicateCreator(A search, List<SingularAttribute<T, A>> attrs) {
		searchval = search;
		attributes = attrs;
	}

	public AndPredicateCreator(A search, SingularAttribute<T, A> attr) {
		searchval = search;
		attributes = Collections.singletonList(attr);
	}

	@Override
	public void exec(CriteriaBuilder cb, CriteriaQuery<?> cq, Path<T> root, List<Predicate> predicates) {
		if (searchval == null || attributes == null || attributes.isEmpty()) {
			return;
		}
		if (attributes.size() > 1) {
			Predicate[] preds = new Predicate[attributes.size()];
			int i = 0;
			for (SingularAttribute<T, A> attr : attributes) {
				preds[i++] = cb.equal(root.get(attr), searchval);
			}
			predicates.add(cb.and(preds));
		} else {
			predicates.add(cb.equal(root.get(attributes.get(0)), searchval));
		}
	}
}