/*******************************************************************************
* 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.olingo.odata2.jpa.processor.core.jpql;
import java.util.Iterator;
import java.util.List;
import java.util.Map.Entry;
import org.apache.olingo.odata2.jpa.processor.api.access.JPAJoinClause;
import org.apache.olingo.odata2.jpa.processor.api.exception.ODataJPARuntimeException;
import org.apache.olingo.odata2.jpa.processor.api.jpql.JPQLContextType;
import org.apache.olingo.odata2.jpa.processor.api.jpql.JPQLContextView;
import org.apache.olingo.odata2.jpa.processor.api.jpql.JPQLJoinContextView;
import org.apache.olingo.odata2.jpa.processor.api.jpql.JPQLStatement;
import org.apache.olingo.odata2.jpa.processor.api.jpql.JPQLStatement.JPQLStatementBuilder;
public class JPQLJoinStatementBuilder extends JPQLStatementBuilder {
JPQLStatement jpqlStatement;
private JPQLJoinContextView context;
public JPQLJoinStatementBuilder(final JPQLContextView context) {
this.context = (JPQLJoinContextView) context;
}
@Override
public JPQLStatement build() throws ODataJPARuntimeException {
jpqlStatement = createStatement(createJPQLQuery());
return jpqlStatement;
}
private String createJPQLQuery() throws ODataJPARuntimeException {
StringBuilder jpqlQuery = new StringBuilder();
StringBuilder joinWhereCondition = null;
jpqlQuery.append(JPQLStatement.KEYWORD.SELECT).append(JPQLStatement.DELIMITER.SPACE);
if (context.getType().equals(JPQLContextType.JOIN_COUNT)) {// $COUNT
jpqlQuery.append(JPQLStatement.KEYWORD.COUNT).append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(JPQLStatement.DELIMITER.PARENTHESIS_LEFT).append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(context.getSelectExpression()).append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(JPQLStatement.DELIMITER.PARENTHESIS_RIGHT).append(JPQLStatement.DELIMITER.SPACE);
} else { // Normal
jpqlQuery.append(context.getSelectExpression()).append(JPQLStatement.DELIMITER.SPACE);
}
jpqlQuery.append(JPQLStatement.KEYWORD.FROM).append(JPQLStatement.DELIMITER.SPACE);
if (context.getJPAJoinClauses() != null && context.getJPAJoinClauses().size() > 0) {
List<JPAJoinClause> joinClauseList = context.getJPAJoinClauses();
JPAJoinClause joinClause = joinClauseList.get(0);
String joinCondition = joinClause.getJoinCondition();
joinWhereCondition = new StringBuilder();
if (joinCondition != null) {
joinWhereCondition.append(joinCondition);
}
String relationShipAlias = null;
joinClause = joinClauseList.get(1);
jpqlQuery.append(joinClause.getEntityName()).append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(joinClause.getEntityAlias());
int i = 1;
int limit = joinClauseList.size();
relationShipAlias = joinClause.getEntityAlias();
while (i < limit) {
jpqlQuery.append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(JPQLStatement.KEYWORD.JOIN).append(JPQLStatement.DELIMITER.SPACE);
joinClause = joinClauseList.get(i);
jpqlQuery.append(relationShipAlias).append(JPQLStatement.DELIMITER.PERIOD);
jpqlQuery.append(joinClause.getEntityRelationShip()).append(JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(joinClause.getEntityRelationShipAlias());
relationShipAlias = joinClause.getEntityRelationShipAlias();
i++;
joinCondition = joinClause.getJoinCondition();
if (joinCondition != null) {
joinWhereCondition.append(JPQLStatement.DELIMITER.SPACE + JPQLStatement.Operator.AND
+ JPQLStatement.DELIMITER.SPACE);
joinWhereCondition.append(joinCondition);
}
}
} else {
throw ODataJPARuntimeException.throwException(ODataJPARuntimeException.JOIN_CLAUSE_EXPECTED, null);
}
String whereExpression = context.getWhereExpression();
if (whereExpression != null || joinWhereCondition.length() > 0) {
jpqlQuery.append(JPQLStatement.DELIMITER.SPACE).append(JPQLStatement.KEYWORD.WHERE).append(
JPQLStatement.DELIMITER.SPACE);
if (whereExpression != null) {
jpqlQuery.append(whereExpression);
if (joinWhereCondition != null) {
jpqlQuery.append(JPQLStatement.DELIMITER.SPACE + JPQLStatement.Operator.AND + JPQLStatement.DELIMITER.SPACE);
}
}
if (joinWhereCondition != null) {
jpqlQuery.append(joinWhereCondition.toString());
}
}
if (context.getOrderByCollection() != null && context.getOrderByCollection().size() > 0) {
StringBuilder orderByBuilder = new StringBuilder();
Iterator<Entry<String, String>> orderItr = context.getOrderByCollection().entrySet().iterator();
int i = 0;
while (orderItr.hasNext()) {
if (i != 0) {
orderByBuilder.append(JPQLStatement.DELIMITER.SPACE).append(JPQLStatement.DELIMITER.COMMA).append(
JPQLStatement.DELIMITER.SPACE);
}
Entry<String, String> entry = orderItr.next();
orderByBuilder.append(entry.getKey()).append(JPQLStatement.DELIMITER.SPACE);
orderByBuilder.append(entry.getValue());
i++;
}
jpqlQuery.append(JPQLStatement.DELIMITER.SPACE).append(JPQLStatement.KEYWORD.ORDERBY).append(
JPQLStatement.DELIMITER.SPACE);
jpqlQuery.append(orderByBuilder);
}
return jpqlQuery.toString();
}
}