OrPredicateCreator.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.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, boolean isFullQuery) {
		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));
	}
}