HHH-5736 Fixed problem with "not" function of CriteriaBuilder

This commit is contained in:
Vyacheslav Dimitrov 2010-12-06 18:13:36 +03:00 committed by Hardy Ferentschik
parent e5da3bf149
commit 0c05f66f8d
3 changed files with 97 additions and 2 deletions

View File

@ -251,7 +251,10 @@ public Order desc(Expression<?> x) {
// predicates ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // predicates ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
public Predicate wrap(Expression<Boolean> expression) { public Predicate wrap(Expression<Boolean> expression) {
if ( Predicate.class.isInstance( expression ) ) { if ( CompoundPredicate.class.isInstance( expression ) ) {
return (( CompoundPredicate ) expression);
}
else if(Predicate.class.isInstance(expression)) {
return ( ( Predicate ) expression ); return ( ( Predicate ) expression );
} }
else if ( PathImplementor.class.isInstance( expression ) ) { else if ( PathImplementor.class.isInstance( expression ) ) {

View File

@ -44,7 +44,7 @@
public class CompoundPredicate public class CompoundPredicate
extends AbstractPredicateImpl extends AbstractPredicateImpl
implements Serializable { implements Serializable {
private final BooleanOperator operator; private BooleanOperator operator;
private final List<Expression<Boolean>> expressions = new ArrayList<Expression<Boolean>>(); private final List<Expression<Boolean>> expressions = new ArrayList<Expression<Boolean>>();
/** /**
@ -148,4 +148,29 @@ private String operatorTextWithSeparator() {
public String renderProjection(CriteriaQueryCompiler.RenderingContext renderingContext) { public String renderProjection(CriteriaQueryCompiler.RenderingContext renderingContext) {
return render( renderingContext ); return render( renderingContext );
} }
/**
* Create negation of compound predicate by using logic rules:
* 1. not (x || y) is (not x && not y)
* 2. not (x && y) is (not x || not y)
*
*/
@Override
public Predicate not() {
if (this.operator == BooleanOperator.AND) {
this.operator = BooleanOperator.OR;
} else {
this.operator = BooleanOperator.AND;
}
for (Expression expr : this.getExpressions()) {
if (Predicate.class.isInstance(expr)) {
( (Predicate) expr ).not();
}
else if(CompoundPredicate.class.isInstance(expr)) {
( (CompoundPredicate) expr ).not();
}
}
return this;
}
} }

View File

@ -27,6 +27,7 @@
import javax.persistence.criteria.CriteriaBuilder; import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery; import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.persistence.criteria.Predicate;
import org.hibernate.ejb.metamodel.AbstractMetamodelSpecificTest; import org.hibernate.ejb.metamodel.AbstractMetamodelSpecificTest;
import org.hibernate.ejb.metamodel.Order; import org.hibernate.ejb.metamodel.Order;
@ -64,4 +65,70 @@ public void testEmptyDisjunction() {
em.getTransaction().commit(); em.getTransaction().commit();
em.close(); em.close();
} }
/**
* Check simple not.
*/
public void testSimpleNot() {
// yes this is a retarded case, but explicitly allowed in the JPA spec
CriteriaBuilder builder = factory.getCriteriaBuilder();
EntityManager em = getOrCreateEntityManager();
em.getTransaction().begin();
CriteriaQuery<Order> orderCriteria = builder.createQuery(Order.class);
Root<Order> orderRoot = orderCriteria.from(Order.class);
orderCriteria.select(orderRoot);
orderCriteria.where( builder.not( builder.equal(orderRoot.get("id"), 1) ) );
em.createQuery( orderCriteria ).getResultList();
em.getTransaction().commit();
em.close();
}
/**
* Check complicated not.
*/
public void testComplicatedNotOr() {
// yes this is a retarded case, but explicitly allowed in the JPA spec
CriteriaBuilder builder = factory.getCriteriaBuilder();
EntityManager em = getOrCreateEntityManager();
em.getTransaction().begin();
CriteriaQuery<Order> orderCriteria = builder.createQuery(Order.class);
Root<Order> orderRoot = orderCriteria.from(Order.class);
orderCriteria.select(orderRoot);
Predicate p1 = builder.equal(orderRoot.get("id"), 1);
Predicate p1 = builder.equal(orderRoot.get("id"), 2);
orderCriteria.where( builder.not( builder.or (p1, p2) ) );
em.createQuery( orderCriteria ).getResultList();
em.getTransaction().commit();
em.close();
}
/**
* Check complicated not.
*/
public void testComplicatedNotAND() {
// yes this is a retarded case, but explicitly allowed in the JPA spec
CriteriaBuilder builder = factory.getCriteriaBuilder();
EntityManager em = getOrCreateEntityManager();
em.getTransaction().begin();
CriteriaQuery<Order> orderCriteria = builder.createQuery(Order.class);
Root<Order> orderRoot = orderCriteria.from(Order.class);
orderCriteria.select(orderRoot);
Predicate p1 = builder.equal(orderRoot.get("id"), 1);
Predicate p1 = builder.equal(orderRoot.get("id"), 2);
orderCriteria.where( builder.not( builder.and (p1, p2) ) );
em.createQuery( orderCriteria ).getResultList();
em.getTransaction().commit();
em.close();
}
} }