Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.cxf.jaxrs.ext.search.jpa.AbstractJPATypedQueryVisitor Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.cxf.jaxrs.ext.search.jpa;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Stack;
import javax.persistence.EntityManager;
import javax.persistence.TypedQuery;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Expression;
import javax.persistence.criteria.Join;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import org.apache.cxf.jaxrs.ext.search.ConditionType;
import org.apache.cxf.jaxrs.ext.search.OrSearchCondition;
import org.apache.cxf.jaxrs.ext.search.PrimitiveStatement;
import org.apache.cxf.jaxrs.ext.search.SearchCondition;
import org.apache.cxf.jaxrs.ext.search.collections.CollectionCheckInfo;
import org.apache.cxf.jaxrs.ext.search.visitor.AbstractSearchConditionVisitor;
public abstract class AbstractJPATypedQueryVisitor
extends AbstractSearchConditionVisitor {
private EntityManager em;
private Class tClass;
private Class queryClass;
private Root root;
private CriteriaBuilder builder;
private CriteriaQuery cq;
private Stack> predStack = new Stack>();
private boolean criteriaFinalized;
private Set joinProperties;
protected AbstractJPATypedQueryVisitor(EntityManager em, Class tClass) {
this(em, tClass, null, null, null);
}
protected AbstractJPATypedQueryVisitor(EntityManager em, Class tClass, Class queryClass) {
this(em, tClass, queryClass, null, null);
}
protected AbstractJPATypedQueryVisitor(EntityManager em,
Class tClass,
Map fieldMap) {
this(em, tClass, null, fieldMap, null);
}
protected AbstractJPATypedQueryVisitor(EntityManager em,
Class tClass,
List joinProps) {
this(em, tClass, null, null, joinProps);
}
protected AbstractJPATypedQueryVisitor(EntityManager em,
Class tClass,
Map fieldMap,
List joinProps) {
this(em, tClass, null, fieldMap, joinProps);
}
protected AbstractJPATypedQueryVisitor(EntityManager em,
Class tClass,
Class queryClass,
Map fieldMap) {
this(em, tClass, queryClass, fieldMap, null);
}
protected AbstractJPATypedQueryVisitor(EntityManager em,
Class tClass,
Class queryClass,
Map fieldMap,
List joinProps) {
super(fieldMap);
this.em = em;
this.tClass = tClass;
this.queryClass = toQueryClass(queryClass, tClass);
this.joinProperties = joinProps == null ? null : new HashSet(joinProps);
}
@SuppressWarnings("unchecked")
private static Class toQueryClass(Class queryClass, Class tClass) {
return queryClass != null ? queryClass : (Class)tClass;
}
protected EntityManager getEntityManager() {
return em;
}
public void visit(SearchCondition sc) {
if (builder == null) {
builder = em.getCriteriaBuilder();
cq = builder.createQuery(queryClass);
root = cq.from(tClass);
predStack.push(new ArrayList());
}
if (sc.getStatement() != null) {
predStack.peek().add(buildPredicate(sc.getStatement()));
} else {
predStack.push(new ArrayList());
for (SearchCondition condition : sc.getSearchConditions()) {
condition.accept(this);
}
List predsList = predStack.pop();
Predicate[] preds = predsList.toArray(new Predicate[predsList.size()]);
Predicate newPred;
if (sc instanceof OrSearchCondition) {
newPred = builder.or(preds);
} else {
newPred = builder.and(preds);
}
predStack.peek().add(newPred);
}
}
protected CriteriaBuilder getCriteriaBuilder() {
return builder;
}
protected Class getQueryClass() {
return queryClass;
}
public Root getRoot() {
return root;
}
public TypedQuery getTypedQuery() {
return em.createQuery(getCriteriaQuery());
}
public CriteriaQuery getCriteriaQuery() {
if (!criteriaFinalized) {
List predsList = predStack.pop();
cq.where(predsList.toArray(new Predicate[predsList.size()]));
criteriaFinalized = true;
}
return cq;
}
private Predicate buildPredicate(PrimitiveStatement ps) {
String name = ps.getProperty();
name = super.getRealPropertyName(name);
ClassValue cv = getPrimitiveFieldClass(ps,
name,
ps.getValue().getClass(),
ps.getValueType(),
ps.getValue());
CollectionCheckInfo collInfo = cv.getCollectionCheckInfo();
Path path = getPath(root, name, cv, collInfo);
Predicate pred = collInfo == null
? doBuildPredicate(ps.getCondition(), path, cv.getCls(), cv.getValue())
: doBuildCollectionPredicate(ps.getCondition(), path, collInfo);
return pred;
}
@SuppressWarnings({ "unchecked", "rawtypes" })
private Predicate doBuildPredicate(ConditionType ct, Path path, Class valueClazz, Object value) {
Class clazz = (Class)valueClazz;
Expression exp = path.as(clazz);
Predicate pred = null;
switch (ct) {
case GREATER_THAN:
pred = builder.greaterThan(exp, clazz.cast(value));
break;
case EQUALS:
if (clazz.equals(String.class)) {
String theValue = value.toString();
if (theValue.contains("*")) {
theValue = ((String)value).replaceAll("\\*", "");
}
pred = builder.like((Expression)exp, "%" + theValue + "%");
} else {
pred = builder.equal(exp, clazz.cast(value));
}
break;
case NOT_EQUALS:
pred = builder.notEqual(exp, clazz.cast(value));
break;
case LESS_THAN:
pred = builder.lessThan(exp, clazz.cast(value));
break;
case LESS_OR_EQUALS:
pred = builder.lessThanOrEqualTo(exp, clazz.cast(value));
break;
case GREATER_OR_EQUALS:
pred = builder.greaterThanOrEqualTo(exp, clazz.cast(value));
break;
default:
break;
}
return pred;
}
@SuppressWarnings({ "unchecked", "rawtypes" })
private Predicate doBuildCollectionPredicate(ConditionType ct, Path path, CollectionCheckInfo collInfo) {
Predicate pred = null;
Expression exp = builder.size((Expression)path);
Integer value = Integer.valueOf(collInfo.getCollectionCheckValue().toString());
switch (ct) {
case GREATER_THAN:
pred = builder.greaterThan(exp, value);
break;
case EQUALS:
pred = builder.equal(exp, value);
break;
case NOT_EQUALS:
pred = builder.notEqual(exp, value);
break;
case LESS_THAN:
pred = builder.lessThan(exp, value);
break;
case LESS_OR_EQUALS:
pred = builder.lessThanOrEqualTo(exp, value);
break;
case GREATER_OR_EQUALS:
pred = builder.greaterThanOrEqualTo(exp, value);
break;
default:
break;
}
return pred;
}
private Path getPath(Path element, String name, ClassValue cv, CollectionCheckInfo collSize) {
if (name.contains(".")) {
String pre = name.substring(0, name.indexOf('.'));
String post = name.substring(name.indexOf('.') + 1);
return getPath(getNextPath(element, pre, cv, null),
post,
cv,
collSize);
} else {
return getNextPath(element, name, cv, collSize);
}
}
private Path getNextPath(Path element, String name, ClassValue cv, CollectionCheckInfo collSize) {
if (collSize == null
&& (cv.isCollection(name) || isJoinProperty(name)) && (element == root || element instanceof Join)) {
return element == root ? root.join(name) : ((Join)element).join(name);
} else {
return element.get(name);
}
}
private boolean isJoinProperty(String prop) {
return joinProperties == null ? false : joinProperties.contains(prop);
}
}
