OrPredicateCreator.java 1.3 KB
package fi.codecrew.moya.facade.callbacks;

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 org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

public class OrPredicateCreator<A, T extends ModelInterface> implements FacadeCallback<T> {
	private final A searchstr;
	private final List<SingularAttribute<T, A>> attributes;
	private static final Logger logger = LoggerFactory.getLogger(OrPredicateCreator.class);

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

	@Override
	public void exec(CriteriaBuilder cb, CriteriaQuery<?> cq, Path<T> root, List<Predicate> predicates) {
		if (searchstr == null || attributes == null || attributes.isEmpty()) {
			return;
		}

		logger.info("Adding searchstring {}", searchstr);
		Predicate[] preds = new Predicate[attributes.size()];
		int i = 0;
		for (SingularAttribute<T, A> attr : attributes) {
			preds[i++] = cb.equal(root.get(attr), searchstr);
		}
		predicates.add(cb.or(preds));
	}
}