com.blazebit.persistence.impl.HavingManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of blaze-persistence-core-impl Show documentation
The newest version!
/*
 * Copyright 2014 Blazebit.
 *
 * Licensed 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 com.blazebit.persistence.impl;

import java.util.Set;

import com.blazebit.persistence.impl.builder.predicate.HavingOrBuilderImpl;
import com.blazebit.persistence.impl.expression.Expression;
import com.blazebit.persistence.impl.expression.ExpressionFactory;

/**
 *
 * @author Moritz Becker
 * @since 1.0
 */
public class HavingManager extends PredicateManager {

    HavingManager(ResolvingQueryGenerator queryGenerator, ParameterManager parameterManager, SubqueryInitiatorFactory subqueryInitFactory, ExpressionFactory expressionFactory) {
        super(queryGenerator, parameterManager, subqueryInitFactory, expressionFactory);
    }

    @Override
    protected String getClauseName() {
        return "HAVING";
    }

    @Override
    public ClauseType getClauseType() {
        return ClauseType.HAVING;
    }

    @SuppressWarnings("unchecked")
    HavingOrBuilderImpl havingOr(AbstractCommonQueryBuilder builder) {
        return rootPredicate.startBuilder(new HavingOrBuilderImpl((T) builder, rootPredicate, subqueryInitFactory, expressionFactory, parameterManager));
    }

    void buildGroupByClauses(Set clauses) {
        if (rootPredicate.getPredicate().getChildren().isEmpty()) {
            return;
        }

        // TODO: No idea yet how to actually handle this
        GroupByExpressionGatheringVisitor visitor = new GroupByExpressionGatheringVisitor();
        rootPredicate.getPredicate().accept(visitor);
        
        StringBuilder sb = new StringBuilder();
        queryGenerator.setQueryBuffer(sb);
        SimpleQueryGenerator.BooleanLiteralRenderingContext oldBooleanLiteralRenderingContext = queryGenerator.setBooleanLiteralRenderingContext(SimpleQueryGenerator.BooleanLiteralRenderingContext.PREDICATE);
        
        for (Expression expr : visitor.getExpressions()) {
        	expr.accept(queryGenerator);
        	clauses.add(sb.toString());
        	sb.setLength(0);
        }
        
        queryGenerator.setBooleanLiteralRenderingContext(oldBooleanLiteralRenderingContext);
    }
}