List of usage examples for javax.persistence PersistenceException PersistenceException
public PersistenceException(Throwable cause)
PersistenceException exception with the specified cause. From source file:org.batoo.jpa.core.impl.criteria.AbstractCriteriaQueryImpl.java
/** * Ensures that there is a valid selection. * //w ww .j a v a 2 s . com * @return the selection * * @since 2.0.0 */ protected AbstractSelection<T> ensureSelection() { if (this.selection == null) { if (this.getRoots().size() == 1) { return this.selection = (AbstractSelection<T>) this.getRoots().iterator().next(); } else { throw new PersistenceException("Selection is not specified"); } } return this.selection; }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Creates the from fragment of the query. * /*from w w w . j a v a 2 s . c om*/ * @param cb * the criteria builder * @param q * the query * @param from * the from metadata * * @since 2.0.0 */ private void constructFrom(CriteriaBuilderImpl cb, AbstractQuery<?> q, Tree froms) { for (int i = 0; i < froms.getChildCount(); i++) { final Tree from = froms.getChild(i); // root query from if (from.getType() == JpqlParser.ST_FROM) { final Aliased fromDef = new Aliased(from.getChild(0)); final EntityTypeImpl<Object> entity = this.getEntity(fromDef.getQualified().toString()); final RootImpl<Object> r = (RootImpl<Object>) q.from(entity); r.alias(fromDef.getAlias()); this.putAlias((BaseQuery<?>) q, from, fromDef, r); this.constructJoins(cb, q, r, from.getChild(1)); if (from.getChild(from.getChildCount() - 1).getType() == JpqlParser.LALL_PROPERTIES) { for (final AssociationMappingImpl<?, ?, ?> association : entity.getAssociations()) { if (!association.isEager()) { final Iterator<String> pathIterator = Splitter.on(".").split(association.getPath()) .iterator(); // Drop the root part pathIterator.next(); Fetch<?, ?> fetch = null; while (pathIterator.hasNext()) { fetch = fetch == null ? r.fetch(pathIterator.next()) : fetch.fetch(pathIterator.next()); } } } } } // in collection form else if (from.getType() == JpqlParser.ST_COLL) { final Aliased aliased = new Aliased(from.getChild(1)); AbstractFrom<?, ?> parent = this.getAliased(q, from.getChild(0).getText()); int depth = 0; for (final String segment : aliased.getQualified().getSegments()) { if ((depth > 0) && (parent instanceof PluralJoin)) { throw new PersistenceException( "Cannot qualify, only embeddable joins within the path allowed, " + "line " + from.getLine() + ":" + from.getCharPositionInLine()); } parent = parent.join(segment, JoinType.LEFT); depth++; } parent.alias(aliased.getAlias()); this.putAlias((BaseQueryImpl<?>) q, from.getChild(1), aliased, parent); } // sub query from else { final Aliased fromDef = new Aliased(from); final EntityTypeImpl<Object> entity = this.getEntity(fromDef.getQualified().toString()); final RootImpl<Object> r = (RootImpl<Object>) q.from(entity); r.alias(fromDef.getAlias()); this.putAlias((BaseQuery<?>) q, from, fromDef, r); } } }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Creates the from fragment of the query. * /* w w w . j ava 2s . c o m*/ * @param cb * the criteria builder * @param q * the query join * @param r * the root * @param joins * the joins metadata * * @since 2.0.0 */ private void constructJoins(CriteriaBuilderImpl cb, AbstractQuery<?> q, RootImpl<Object> r, Tree joins) { for (int i = 0; i < joins.getChildCount(); i++) { final Tree join = joins.getChild(i); JoinType joinType = JoinType.INNER; final int joinSpecification = join.getChild(0).getType(); int offset = 0; if (joinSpecification == JpqlParser.INNER) { offset = 1; joinType = JoinType.INNER; } else if (joinSpecification == JpqlParser.LEFT) { offset = 1; joinType = JoinType.LEFT; } if (join.getChildCount() == (offset + 3)) { FetchParent<?, ?> parent = this.getAliased(q, join.getChild(offset).getText()); final Qualified qualified = new Qualified(join.getChild(offset + 1).getChild(0)); for (final String segment : qualified.getSegments()) { parent = parent.fetch(segment, joinType); } } else { AbstractFrom<?, ?> parent = this.getAliased(q, join.getChild(offset).getText()); final Aliased aliased = new Aliased(join.getChild(offset + 1)); int depth = 0; for (final String segment : aliased.getQualified().getSegments()) { if ((depth > 0) && (parent instanceof PluralJoin)) { throw new PersistenceException( "Cannot qualify, only embeddable joins within the path allowed, " + "line " + join.getLine() + ":" + join.getCharPositionInLine()); } parent = parent.join(segment, joinType); depth++; } parent.alias(aliased.getAlias()); this.putAlias((BaseQuery<?>) q, join.getChild(1), aliased, parent); } } }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Constructs the order by fragment of the query. * //from w ww .ja va2 s. c o m * @param cb * the criteria builder * @param q * the query * @param selections * the selections * @param orderBy * the order by definitions * * @since 2.0.0 */ private void constructOrder(CriteriaBuilderImpl cb, CriteriaQueryImpl<?> q, List<Selection<?>> selections, Tree orderBy) { final List<Order> orders = Lists.newArrayList(); for (int i = 0; i < orderBy.getChildCount(); i++) { final Tree orderByItem = orderBy.getChild(i); final Tree orderItem = orderByItem.getChild(0); Order order = null; if (orderItem.getType() == JpqlParser.ID) { final String alias = orderItem.getText(); for (final Selection<?> selection : selections) { if (alias.equals(selection.getAlias())) { order = orderByItem.getChildCount() == 2 ? // cb.desc((Expression<?>) selection) : // cb.asc((Expression<?>) selection); break; } } if (order == null) { throw new PersistenceException("Alias is not bound: " + alias); } } else { order = orderByItem.getChildCount() == 2 ? // cb.desc(this.getExpression(cb, q, orderItem, null)) : // cb.asc(this.getExpression(cb, q, orderByItem.getChild(0), null)); } orders.add(order); } q.orderBy(orders); }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Creates a single select item./*from w w w . jav a2 s. c o m*/ * * @param cb * the criteria builder * @param q * the query * @param selectDef * the select definition * * @since 2.0.0 */ private AbstractSelection<?> constructSingleSelect(CriteriaBuilderImpl cb, CriteriaQueryImpl<?> q, Tree selectDef) { // constructor select if (selectDef.getType() == JpqlParser.NEW) { final String className = new Qualified(selectDef.getChild(0)).toString(); final List<Selection<?>> childSelections = Lists.newArrayList(); final Tree arguments = selectDef.getChild(1); for (int i = 0; i < arguments.getChildCount(); i++) { final Tree argumentDef = arguments.getChild(i); childSelections.add(this.getExpression(cb, q, argumentDef, null)); } try { final Class<?> clazz = this.metamodel.getEntityManagerFactory().getClassloader() .loadClass(className); return cb.construct(clazz, childSelections.toArray(new Selection[childSelections.size()])); } catch (final ClassNotFoundException e) { throw new PersistenceException("Cannot load class: " + className + ", line " + selectDef.getLine() + ":" + selectDef.getCharPositionInLine()); } } // object type if (selectDef.getType() == JpqlParser.OBJECT) { final String alias = selectDef.getChild(0).getText(); return this.getAliased(q, alias); } return this.getExpression(cb, q, selectDef, null); }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Constructs an update query./*from w w w . j a va2 s . c o m*/ * * @param cb * the criteria builder * @param tree * the tree * @return the update query constructed * * @since 2.0.0 */ @SuppressWarnings({ "rawtypes", "unchecked" }) private CriteriaUpdateImpl<?> constructUpdateQuery(CriteriaBuilderImpl cb, CommonTree tree) { final CriteriaUpdateImpl<?> q = new CriteriaUpdateImpl(this.metamodel, this.qlString); final Tree updateDef = tree.getChild(0); final Tree aliasedDef = updateDef.getChild(0); final Aliased aliased = new Aliased(aliasedDef); final EntityTypeImpl entity = this.getEntity(aliased.getQualified().toString()); final RootImpl<?> r = (RootImpl<?>) q.from(entity); this.putAlias(q, aliasedDef, aliased, r); if (StringUtils.isNotBlank(aliased.getAlias())) { r.alias(aliased.getAlias()); } final Tree setDefs = updateDef.getChild(1); for (int i = 0; i < setDefs.getChildCount(); i++) { final Tree setDef = setDefs.getChild(i); final Tree leftDef = setDef.getChild(0); final Tree rightDef = setDef.getChild(1); AbstractExpression left = null; if (leftDef.getType() == JpqlParser.LQUALIFIED) { for (int j = 0; j < leftDef.getChildCount(); j++) { final Tree child = leftDef.getChild(j); if ((left == null) && child.getText().equals(r.getAlias())) { left = r; } else if (left == null) { left = r.getExpression(child.getText()); } else if (left instanceof SingularJoin) { left = ((SingularJoin<?, ?>) left).getExpression(child.getText()); } else if (left instanceof RootImpl) { left = ((RootImpl<?>) left).getExpression(child.getText()); } else { throw new PersistenceException("cannot dereference: " + leftDef.getText()); } } } if (left instanceof BasicPath) { final AbstractExpression right = this.getExpression(cb, q, rightDef, left.getJavaType()); q.set((BasicPath<?>) left, right); } else { throw new PersistenceException("Path does not resolve to a state field: " + leftDef.getText()); } } if (updateDef.getChildCount() == 3) { q.where(this.constructJunction(cb, q, updateDef.getChild(2))); } return q; }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
private AbstractFrom<?, ?> getAliased(Object q, String alias) { final Map<String, AbstractFrom<?, ?>> aliasMap = this.aliasMap.get(q); if (aliasMap != null) { final AbstractFrom<?, ?> from = aliasMap.get(alias); if (from != null) { return from; }// ww w .j a va 2s . c o m } if (q instanceof Subquery) { final SubqueryImpl<?> s = (SubqueryImpl<?>) q; final AbstractFrom<?, ?> aliased = this.getAliased(s.getParent(), alias); if (aliased instanceof RootImpl) { s.correlate((RootImpl<?>) aliased); } return aliased; } throw new PersistenceException("Alias is not bound: " + alias); }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
private EntityTypeImpl<Object> getEntity(String entityName) { final EntityTypeImpl<Object> entity = this.metamodel.entity(entityName); if (entity == null) { throw new PersistenceException("Type is not managed: " + entityName); }//from www . ja v a 2 s . c o m return entity; }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
/** * Constructs and returns the expression. * //from ww w . jav a2 s .co m * @param cb * the criteria builder * @param exprDef * the definition of the expression * @return the expression * * @since 2.0.0 */ @SuppressWarnings({ "unchecked", "rawtypes" }) private <X, C extends Collection<E>, E> AbstractExpression<X> getExpression(CriteriaBuilderImpl cb, Object q, Tree exprDef, Class<X> javaType) { if ((exprDef.getType() == JpqlParser.Equals_Operator) // || (exprDef.getType() == JpqlParser.Not_Equals_Operator) // || (exprDef.getType() == JpqlParser.Greater_Than_Operator) // || (exprDef.getType() == JpqlParser.Greater_Or_Equals_Operator) // || (exprDef.getType() == JpqlParser.Less_Than_Operator) // || (exprDef.getType() == JpqlParser.Less_Or_Equals_Operator) // || (exprDef.getType() == JpqlParser.BETWEEN)) { final AbstractExpression<X> left; final AbstractExpression<X> right; if ((exprDef.getChild(0).getType() == JpqlParser.ST_SUBQUERY) || (exprDef.getChild(1).getType() == JpqlParser.ST_SUBQUERY)) { // left side is sub query if (exprDef.getChild(0).getType() == JpqlParser.ST_SUBQUERY) { right = this.getExpression(cb, q, exprDef.getChild(1), null); left = (AbstractExpression<X>) this.constructSubquery(cb, q, exprDef.getChild(0), right.getJavaType()); // right side is sub query } else if (exprDef.getChild(1).getType() == JpqlParser.ST_SUBQUERY) { left = this.getExpression(cb, q, exprDef.getChild(0), null); right = (AbstractExpression<X>) this.constructSubquery(cb, q, exprDef.getChild(1), left.getJavaType()); } else { throw new PersistenceException("Both sides of the comparison cannot be sub query, line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } } else { final Tree leftExpr = exprDef.getChild(0); final Tree rightExpr = exprDef.getChild(1); if ((leftExpr.getType() == JpqlParser.Named_Parameter) // || (leftExpr.getType() == JpqlParser.Ordinal_Parameter) // || (leftExpr.getType() == JpqlParser.Question_Sign)) { left = (AbstractExpression<X>) this.getExpression(cb, q, rightExpr, null); right = (AbstractExpression<X>) this.getExpression(cb, q, leftExpr, left.getJavaType()); } else { left = this.getExpression(cb, q, leftExpr, null); right = (AbstractExpression<X>) this.getExpression(cb, q, rightExpr, left.getJavaType()); } } switch (exprDef.getType()) { case JpqlParser.Equals_Operator: return (AbstractExpression<X>) cb.equal(left, right); case JpqlParser.Not_Equals_Operator: return (AbstractExpression<X>) cb.notEqual(left, right); case JpqlParser.Greater_Than_Operator: if (Comparable.class.isAssignableFrom(left.getJavaType())) { return (AbstractExpression<X>) cb.greaterThan((Expression<Comparable>) left, (Expression<Comparable>) right); } else { return (AbstractExpression<X>) cb.gt((Expression<? extends Number>) left, (Expression<? extends Number>) right); } case JpqlParser.Greater_Or_Equals_Operator: if (Comparable.class.isAssignableFrom(left.getJavaType())) { return (AbstractExpression<X>) cb.greaterThanOrEqualTo((Expression<Comparable>) left, (Expression<Comparable>) right); } else { return (AbstractExpression<X>) cb.ge((Expression<? extends Number>) left, (Expression<? extends Number>) right); } case JpqlParser.Less_Than_Operator: if (Comparable.class.isAssignableFrom(left.getJavaType())) { return (AbstractExpression<X>) cb.lessThan((Expression<Comparable>) left, (Expression<Comparable>) right); } else { return (AbstractExpression<X>) cb.lt((Expression<? extends Number>) left, (Expression<? extends Number>) right); } case JpqlParser.Less_Or_Equals_Operator: if (Comparable.class.isAssignableFrom(left.getJavaType())) { return (AbstractExpression<X>) cb.lessThanOrEqualTo((Expression<Comparable>) left, (Expression<Comparable>) right); } else { return (AbstractExpression<X>) cb.le((Expression<? extends Number>) left, (Expression<? extends Number>) right); } case JpqlParser.BETWEEN: final AbstractExpression<?> right2 = this.getExpression(cb, q, exprDef.getChild(2), left.getJavaType()); final PredicateImpl between = cb.between((AbstractExpression) left, (AbstractExpression) right, (AbstractExpression) right2); if (exprDef.getChildCount() == 4) { return (AbstractExpression<X>) between.not(); } return (AbstractExpression<X>) between; } } if (exprDef.getType() == JpqlParser.LIKE) { final AbstractExpression<String> inner = this.getExpression(cb, q, exprDef.getChild(0), String.class); final AbstractExpression<String> pattern = this.getExpression(cb, q, exprDef.getChild(1), String.class); if ((exprDef.getChildCount() > 2) && (exprDef.getChild(2).getType() == JpqlParser.STRING_LITERAL)) { final Expression<Character> escape = this.getExpression(cb, q, exprDef.getChild(2), Character.class); if (exprDef.getChild(exprDef.getChildCount() - 1).getType() == JpqlParser.NOT) { return (AbstractExpression<X>) cb.notLike(inner, pattern, escape); } else { return (AbstractExpression<X>) cb.like(inner, pattern, escape); } } else { if (exprDef.getChild(exprDef.getChildCount() - 1).getType() == JpqlParser.NOT) { return (AbstractExpression<X>) cb.notLike(inner, pattern); } else { return (AbstractExpression<X>) cb.like(inner, pattern); } } } if (exprDef.getType() == JpqlParser.ST_IN) { final boolean notIn = (exprDef.getChildCount() > 2) && (exprDef.getChild(2).getType() == JpqlParser.NOT); AbstractExpression<X> left = null; if ((exprDef.getChild(0).getType() != JpqlParser.Named_Parameter) // && (exprDef.getChild(0).getType() != JpqlParser.Ordinal_Parameter) // && (exprDef.getChild(0).getType() != JpqlParser.Question_Sign)) { left = this.getExpression(cb, q, exprDef.getChild(0), null); } final List<AbstractExpression<X>> expressions = Lists.newArrayList(); final Tree inDefs = exprDef.getChild(1); if ((inDefs.getType() == JpqlParser.Named_Parameter) // || (inDefs.getType() == JpqlParser.Ordinal_Parameter)// || (inDefs.getType() == JpqlParser.Question_Sign)) { return notIn ? // (AbstractExpression<X>) left.notIn(this.getExpression(cb, q, inDefs, left.getJavaType()))// : (AbstractExpression<X>) left.in(this.getExpression(cb, q, inDefs, left.getJavaType())); } if (inDefs.getType() == JpqlParser.ID) { this.getAliased(q, inDefs.getText()); } if (inDefs.getType() == JpqlParser.ST_SUBQUERY) { final SubqueryImpl<? extends X> subquery = this.constructSubquery(cb, q, inDefs, left.getJavaType()); return notIn ? // (AbstractExpression<X>) left.notIn(subquery) : // (AbstractExpression<X>) left.in(subquery); } else { for (int i = 0; i < inDefs.getChildCount(); i++) { expressions.add((AbstractExpression<X>) this.getExpression(cb, q, inDefs.getChild(i), left != null ? left.getJavaType() : null)); } } if (left == null) { left = (AbstractExpression<X>) this.getExpression(cb, q, exprDef.getChild(0), expressions.get(0).getJavaType()); } return notIn ? // (AbstractExpression<X>) left.notIn(expressions) : // (AbstractExpression<X>) left.in(expressions); } if (exprDef.getType() == JpqlParser.ST_NULL) { final AbstractExpression<Object> expr = this.getExpression(cb, q, exprDef.getChild(0), null); if (exprDef.getChildCount() == 2) { return (AbstractExpression<X>) cb.isNotNull(expr); } return (AbstractExpression<X>) cb.isNull(expr); } // identification variable if (exprDef.getType() == JpqlParser.ID) { return (AbstractExpression<X>) this.getAliased(q, exprDef.getText()); } // single valued state field expression if (exprDef.getType() == JpqlParser.ST_PARENTED) { AbstractSelection<?> expression = this.getAliased(q, exprDef.getChild(0).getText()); final Qualified qualified = new Qualified(exprDef.getChild(1)); final Iterator<String> i = qualified.getSegments().iterator(); while (i.hasNext()) { final String segment = i.next(); if (expression instanceof ParentPath) { expression = ((ParentPath<?, ?>) expression).getExpression(segment); } else { throw new PersistenceException("Cannot dereference: " + segment + ", line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } } return (AbstractExpression<X>) expression; } // negation if (exprDef.getType() == JpqlParser.ST_NEGATION) { return (AbstractExpression<X>) cb .neg(this.<Number, Collection<Object>, Object>getExpression(cb, q, exprDef.getChild(0), null)); } if (exprDef.getType() == JpqlParser.Named_Parameter) { final String paramName = exprDef.getText().substring(1); ParameterExpressionImpl<?> expr = this.namedParamMap.get(paramName); if (expr != null) { return (AbstractExpression<X>) expr; } expr = cb.parameter(javaType, paramName); this.namedParamMap.put(paramName, expr); return (AbstractExpression<X>) expr; } if ((exprDef.getType() == JpqlParser.Ordinal_Parameter) // || (exprDef.getType() == JpqlParser.Question_Sign)) { final String strPos = exprDef.getText().substring(1); try { final int position = Integer.parseInt(strPos); AbstractParameterExpressionImpl<X> parameter = null; Object q2 = q; while (q2 instanceof SubqueryImpl) { q2 = ((SubqueryImpl<?>) q2).getParent(); } if (q2 instanceof CriteriaQueryImpl) { parameter = (AbstractParameterExpressionImpl<X>) ((CriteriaQueryImpl<?>) q2) .getParameter(position); } else if (q2 instanceof CriteriaDeleteImpl) { parameter = (AbstractParameterExpressionImpl<X>) ((CriteriaDeleteImpl<?>) q2) .getParameter(position); } else { parameter = (AbstractParameterExpressionImpl<X>) ((CriteriaUpdateImpl<?>) q2) .getParameter(position); } if (parameter == null) { parameter = new ParameterExpressionImpl<X>((BaseQueryImpl<?>) q2, this.metamodel.type(javaType), javaType, position); } return parameter; } catch (final NumberFormatException e) { throw new PersistenceException("Invalid ordinal query parameter declaration: " + strPos); } } // arithmetic operation if ((exprDef.getType() == JpqlParser.Plus_Sign) // || (exprDef.getType() == JpqlParser.Minus_Sign) // || (exprDef.getType() == JpqlParser.Multiplication_Sign) // || (exprDef.getType() == JpqlParser.Division_Sign)) { final AbstractExpression<Number> left = this.getExpression(cb, q, exprDef.getChild(0), Number.class); final AbstractExpression<? extends Number> right = this.getExpression(cb, q, exprDef.getChild(1), left.getJavaType()); switch (exprDef.getType()) { case JpqlParser.Plus_Sign: return (AbstractExpression<X>) cb.sum(left, right); case JpqlParser.Minus_Sign: return (AbstractExpression<X>) cb.diff(left, right); case JpqlParser.Multiplication_Sign: return (AbstractExpression<X>) cb.prod(left, right); case JpqlParser.Division_Sign: return (AbstractExpression<X>) cb.quot(left, right); } } if (exprDef.getType() == JpqlParser.ST_BOOLEAN) { return (AbstractExpression<X>) this.getExpression(cb, q, exprDef, Boolean.class); } if (exprDef.getType() == JpqlParser.NUMERIC_LITERAL) { return (AbstractExpression<X>) new SimpleConstantExpression<Long>( this.metamodel.createBasicType(Long.class), Long.valueOf(exprDef.getText())); } // string literal if (exprDef.getType() == JpqlParser.STRING_LITERAL) { if (javaType == Character.class) { return (AbstractExpression<X>) new SimpleConstantExpression<Character>( this.metamodel.type(Character.class), // exprDef.getText().substring(1, 2).toCharArray()[0]); } return (AbstractExpression<X>) new SimpleConstantExpression<String>(this.metamodel.type(String.class), // exprDef.getText().substring(1, exprDef.getText().length() - 1)); } // functions returning string if ((exprDef.getType() == JpqlParser.UPPER) // || (exprDef.getType() == JpqlParser.LOWER) // || (exprDef.getType() == JpqlParser.SUBSTRING)) { final AbstractExpression<String> argument = this.getExpression(cb, q, exprDef.getChild(0), null); switch (exprDef.getType()) { case JpqlParser.UPPER: return (AbstractExpression<X>) cb.upper(argument); case JpqlParser.LOWER: return (AbstractExpression<X>) cb.lower(argument); case JpqlParser.SUBSTRING: final AbstractExpression<Integer> start = this.getExpression(cb, q, exprDef.getChild(1), Integer.class); final AbstractExpression<Integer> end = exprDef.getChildCount() == 3 ? // this.getExpression(cb, q, exprDef.getChild(2), Integer.class) : null; return (AbstractExpression<X>) new SubstringExpression(argument, start, end); } } // concat function if (exprDef.getType() == JpqlParser.CONCAT) { final List<Expression<String>> arguments = Lists.newArrayList(); for (int i = 0; i < exprDef.getChildCount(); i++) { arguments.add(this.getExpression(cb, q, exprDef.getChild(i), String.class)); } return (AbstractExpression<X>) new ConcatExpression( arguments.toArray(new Expression[arguments.size()])); } // trim function if (exprDef.getType() == JpqlParser.TRIM) { Trimspec trimspec = null; Expression<Character> trimChar = null; Expression<String> inner = null; int i = 0; final int type = exprDef.getChild(i).getType(); // trim spec if (type == JpqlParser.BOTH) { trimspec = Trimspec.BOTH; i++; } else if (type == JpqlParser.LEADING) { trimspec = Trimspec.LEADING; i++; } else if (type == JpqlParser.TRAILING) { trimspec = Trimspec.TRAILING; i++; } if (exprDef.getChildCount() > (i + 1)) { trimChar = this.getExpression(cb, q, exprDef.getChild(i), Character.class); inner = this.getExpression(cb, q, exprDef.getChild(i + 1), String.class); } else { inner = this.getExpression(cb, q, exprDef.getChild(i), String.class); } return (AbstractExpression<X>) new TrimExpression(trimspec, trimChar, inner); } // type functions if ((exprDef.getType() == JpqlParser.TYPE) || (exprDef.getType() == JpqlParser.ST_ENTITY_TYPE)) { switch (exprDef.getType()) { case JpqlParser.TYPE: final AbstractExpression<?> inner = this.getExpression(cb, q, exprDef.getChild(0), null); return (AbstractExpression<X>) ((AbstractPath<?>) inner).type(); case JpqlParser.ST_ENTITY_TYPE: final EntityTypeImpl<?> entity = this.getEntity(exprDef.getChild(0).getText()); if (entity.getRootType().getInheritanceType() == null) { throw new PersistenceException("Entity does not have inheritence: " + entity.getName() + ", line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } return (AbstractExpression<X>) new SimpleConstantExpression<String>(null, entity.getDiscriminatorValue()); } } // date time functions switch (exprDef.getType()) { case JpqlParser.CURRENT_DATE: return (AbstractExpression<X>) cb.currentDate(); case JpqlParser.CURRENT_TIME: return (AbstractExpression<X>) cb.currentTime(); case JpqlParser.CURRENT_TIMESTAMP: return (AbstractExpression<X>) cb.currentTimestamp(); case JpqlParser.SECOND: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.SECOND, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.MINUTE: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.MINUTE, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.HOUR: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.HOUR, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.DAY: case JpqlParser.DAYOFMONTH: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.DAYOFMONTH, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.DAYOFWEEK: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.DAYOFWEEK, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.DAYOFYEAR: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.DAYOFYEAR, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.MONTH: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.MONTH, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.WEEK: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.WEEK, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); case JpqlParser.YEAR: return (AbstractExpression<X>) cb.dateTimeExpression(DateTimeFunctionType.YEAR, this.getExpression(cb, q, exprDef.getChild(0), Date.class)); } // arithmetic functions switch (exprDef.getType()) { case JpqlParser.ABS: return (AbstractExpression<X>) cb.abs(this.getExpression(cb, q, exprDef.getChild(0), Number.class)); case JpqlParser.SQRT: return (AbstractExpression<X>) cb.sqrt(this.getExpression(cb, q, exprDef.getChild(0), Number.class)); case JpqlParser.MOD: return (AbstractExpression<X>) cb.mod(// this.getExpression(cb, q, exprDef.getChild(0), Integer.class), // this.getExpression(cb, q, exprDef.getChild(1), Integer.class)); case JpqlParser.LOCATE: if (exprDef.getChildCount() == 3) { return (AbstractExpression<X>) cb.locate(// this.getExpression(cb, q, exprDef.getChild(0), String.class), // this.getExpression(cb, q, exprDef.getChild(1), String.class), // this.getExpression(cb, q, exprDef, Integer.class)); } return (AbstractExpression<X>) cb.locate(// this.getExpression(cb, q, exprDef.getChild(0), String.class), // this.getExpression(cb, q, exprDef.getChild(1), String.class)); case JpqlParser.LENGTH: return (AbstractExpression<X>) cb.length(this.getExpression(cb, q, exprDef.getChild(0), String.class)); } // aggregate functions switch (exprDef.getType()) { case JpqlParser.AVG: return (AbstractExpression<X>) cb.avg(this.getExpression(cb, q, exprDef.getChild(0), Number.class)); case JpqlParser.SUM: return (AbstractExpression<X>) cb.sum(this.getExpression(cb, q, exprDef.getChild(0), Long.class)); case JpqlParser.MAX: return (AbstractExpression<X>) cb.max(this.getExpression(cb, q, exprDef.getChild(0), Number.class)); case JpqlParser.MIN: return (AbstractExpression<X>) cb.min(this.getExpression(cb, q, exprDef.getChild(0), Number.class)); } // count function if (exprDef.getType() == JpqlParser.COUNT) { if (exprDef.getChildCount() == 2) { return (AbstractExpression<X>) new CountExpression( this.getExpression(cb, q, exprDef.getChild(1), null), true); } return (AbstractExpression<X>) new CountExpression(this.getExpression(cb, q, exprDef.getChild(0), null), false); } // all or any operator if (exprDef.getType() == JpqlParser.ST_ALL_OR_ANY) { // all, any, some expressions switch (exprDef.getChild(0).getType()) { case JpqlParser.ALL: return new AllAnyExpression<X>(true, this.constructSubquery(cb, q, exprDef.getChild(1), javaType)); case JpqlParser.ANY: case JpqlParser.SOME: return new AllAnyExpression<X>(false, this.constructSubquery(cb, q, exprDef.getChild(1), javaType)); } } // exists operator if (exprDef.getType() == JpqlParser.EXISTS) { return (AbstractExpression<X>) new ExistsExpression( this.constructSubquery(cb, q, exprDef.getChild(0), javaType)); } // not operator if (exprDef.getType() == JpqlParser.NOT) { final Tree innerExpression = exprDef.getChild(0); final AbstractExpression<Boolean> expression = innerExpression.getType() == JpqlParser.LOR ? this.constructJunction(cb, q, innerExpression) : this.getExpression(cb, q, innerExpression, Boolean.class); return (AbstractExpression<X>) new PredicateImpl(true, BooleanOperator.AND, expression); } // general case if (exprDef.getType() == JpqlParser.ST_GENERAL_CASE) { final CaseImpl<Object> caseExpr = cb.selectCase(); for (int i = 0; i < exprDef.getChildCount(); i++) { final Tree caseDef = exprDef.getChild(i); if (caseDef.getType() == JpqlParser.WHEN) { caseExpr.when(this.constructJunction(cb, q, caseDef.getChild(0)), this.getExpression(cb, q, caseDef.getChild(1), null)); } else { caseExpr.otherwise(this.getExpression(cb, q, caseDef, null)); } } return (AbstractExpression<X>) caseExpr; } // simple case if (exprDef.getType() == JpqlParser.CASE) { final AbstractExpression<X> expression = this.getExpression(cb, q, exprDef.getChild(0), null); final SimpleCaseImpl<X, Object> caseExpr = cb.selectCase(expression); for (int i = 1; i < exprDef.getChildCount(); i++) { final Tree caseDef = exprDef.getChild(i); if (caseDef.getType() == JpqlParser.WHEN) { final AbstractExpression<Object> result = this.getExpression(cb, q, caseDef.getChild(1), null); final AbstractExpression<X> condition; if (exprDef.getChild(0).getType() == JpqlParser.TYPE) { final EntityTypeImpl<Object> entity = this.getEntity(caseDef.getChild(0).getText()); if (entity.getRootType().getInheritanceType() == null) { throw new PersistenceException("Entity does not have inheritence: " + entity.getName() + ", line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } condition = (AbstractExpression<X>) new SimpleConstantExpression<String>(null, entity.getDiscriminatorValue()); } else { condition = this.getExpression(cb, q, caseDef.getChild(0), null); } caseExpr.when(condition, result); } else { caseExpr.otherwise(this.getExpression(cb, q, caseDef, null)); } } return (AbstractExpression<X>) caseExpr; } // nullif function if (exprDef.getType() == JpqlParser.NULLIF) { final AbstractExpression<X> left = this.getExpression(cb, q, exprDef.getChild(0), null); final AbstractExpression<?> right = this.getExpression(cb, q, exprDef.getChild(1), null); return new NullIfExpression<X>(left, right); } // coalesce function if (exprDef.getType() == JpqlParser.ST_COALESCE) { final CoalesceExpression<X> coalesce = cb.coalesce(); for (int i = 0; i < exprDef.getChildCount(); i++) { coalesce.value(this.getExpression(cb, q, exprDef.getChild(i), javaType)); } return coalesce; } // db func if (exprDef.getType() == JpqlParser.FUNC) { final List<AbstractExpression<?>> arguments = Lists.newArrayList(); final String function = exprDef.getChild(0).getText(); for (int i = 1; i < exprDef.getChildCount(); i++) { arguments.add(this.getExpression(cb, q, exprDef.getChild(i), null)); } return new FunctionExpression<X>((Class<X>) (javaType != null ? javaType : Object.class), // function, arguments.toArray(new Expression<?>[arguments.size()])); } // index expression if (exprDef.getType() == JpqlParser.INDEX) { final AbstractExpression<Object> expression = this.getExpression(cb, q, exprDef.getChild(0), null); if (expression instanceof ListJoinImpl) { return (AbstractExpression<X>) ((ListJoinImpl<?, ?>) expression).index(); } throw new PersistenceException("Reference is not a list join, line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } // empty operation if (exprDef.getType() == JpqlParser.ST_EMPTY) { AbstractExpression<?> expression = this.getExpression(cb, q, exprDef.getChild(0), null); if (expression instanceof MapExpression) { expression = ((MapExpression<Map<?, ?>, ?, ?>) expression).values(); } if (!(expression instanceof CollectionExpression<?, ?>)) { throw new PersistenceException("Reference is not a collection, line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } if (exprDef.getChildCount() == 2) { return (AbstractExpression<X>) cb.isNotEmpty((Expression<Collection<?>>) expression); } else { return (AbstractExpression<X>) cb.isEmpty((Expression<Collection<?>>) expression); } } // member of operation if (exprDef.getType() == JpqlParser.ST_MEMBER) { final AbstractExpression<?> expression = this.getExpression(cb, q, exprDef.getChild(1), null); if (!(expression instanceof CollectionExpression)) { throw new PersistenceException("Member of expression must evaluate to a collection expression, " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } final CollectionExpression<C, E> collection = (CollectionExpression<C, E>) expression; final PluralAttributeImpl<?, C, E> attribute = (PluralAttributeImpl<?, C, E>) collection.getMapping() .getAttribute(); final AbstractExpression<E> elem = this.getExpression(cb, q, exprDef.getChild(0), attribute.getElementType().getJavaType()); if (exprDef.getChildCount() == 3) { return (AbstractExpression<X>) cb.isNotMember(elem, collection); } else { return (AbstractExpression<X>) cb.isMember(elem, collection); } } // size operation if (exprDef.getType() == JpqlParser.SIZE) { final AbstractExpression<?> expression = this.getExpression(cb, q, exprDef.getChild(0), null); if (!(expression instanceof CollectionExpression)) { throw new PersistenceException("Member of expression must evaluate to a collection expression, " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); } final CollectionExpression<C, E> collection = (CollectionExpression<C, E>) expression; return (AbstractExpression<X>) cb.size(collection); } if (exprDef.getType() == JpqlParser.CAST) { final AbstractExpression<?> left = this.getExpression(cb, q, exprDef.getChild(0), null); Class<?> clazz = null; switch (exprDef.getChild(1).getType()) { case JpqlParser.BYTE: clazz = Byte.class; break; case JpqlParser.SHORT: clazz = Short.class; break; case JpqlParser.INT: case JpqlParser.INTEGER: clazz = Integer.class; break; case JpqlParser.LONG: clazz = Long.class; break; case JpqlParser.FLOAT: clazz = Float.class; break; case JpqlParser.DOUBLE: clazz = Double.class; break; default: clazz = String.class; } return (AbstractExpression<X>) cb.cast(left, clazz); } if (exprDef.getType() == JpqlParser.TRUE) { return (AbstractExpression<X>) new SimpleConstantExpression<Boolean>(null, Boolean.TRUE); } if (exprDef.getType() == JpqlParser.FALSE) { return (AbstractExpression<X>) new SimpleConstantExpression<Boolean>(null, Boolean.FALSE); } throw new PersistenceException("Unhandled expression: " + exprDef.toStringTree() + ", line " + exprDef.getLine() + ":" + exprDef.getCharPositionInLine()); }
From source file:org.batoo.jpa.core.impl.criteria.jpql.JpqlQuery.java
private CommonTree parse(String query) { try {// w w w . ja v a 2 s .c o m final JpqlLexer lexer = new JpqlLexer(new ANTLRStringStream(query)); final CommonTokenStream tokenStream = new CommonTokenStream(lexer); final JpqlParser parser = new JpqlParser(tokenStream); final ql_statement_return ql_statement = parser.ql_statement(); final CommonTree tree = (CommonTree) ql_statement.getTree(); final List<String> errors = parser.getErrors(); if (errors.size() > 0) { final String errorMsg = Joiner.on("\n\t").join(errors); JpqlQuery.LOG.error("Cannot parse query: {0}", // JpqlQuery.LOG.boxed(query, // new Object[] { "\n\t" + errorMsg, "\n\n" + tree.toStringTree() + "\n" })); throw new PersistenceException("Cannot parse the query:\n " + errorMsg + ".\n" + query); } return tree; } catch (final PersistenceException e) { throw e; } catch (final Exception e) { throw new PersistenceException("Cannot parse the query:\n " + e.getMessage() + ".\n" + query, e); } }