All Downloads are FREE. Search and download functionalities are using the official Maven repository.

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

Go to download 

/*
 * Copyright 2014 - 2020 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.List;
import java.util.Map;

import com.blazebit.persistence.SelectCTECriteriaBuilder;
import com.blazebit.persistence.SelectRecursiveCTECriteriaBuilder;
import com.blazebit.persistence.parser.expression.ExpressionCopyContext;

/**
 * @author Christian Beikov
 * @since 1.2.0
 */
public class RecursiveCTECriteriaBuilderImpl extends AbstractCTECriteriaBuilder, SelectCTECriteriaBuilder, Void> implements SelectRecursiveCTECriteriaBuilder, CTEBuilderListener {

    protected final Class clazz;
    protected boolean done;
    protected boolean unionAll;
    protected SelectCTECriteriaBuilderImpl recursiveCteBuilder;

    public RecursiveCTECriteriaBuilderImpl(MainQuery mainQuery, QueryContext queryContext, CTEManager.CTEKey cteKey, Class clazz, Y result, final CTEBuilderListener listener) {
        super(mainQuery, queryContext, cteKey, false, clazz, result, listener, null, null, null);
        this.clazz = clazz;
    }

    public RecursiveCTECriteriaBuilderImpl(RecursiveCTECriteriaBuilderImpl builder, MainQuery mainQuery, QueryContext queryContext, Map joinManagerMapping, ExpressionCopyContext copyContext) {
        super(builder, mainQuery, queryContext, joinManagerMapping, copyContext);
        this.clazz = builder.clazz;
        this.done = builder.done;
        this.unionAll = builder.unionAll;
        this.recursiveCteBuilder = builder.recursiveCteBuilder.copy(queryContext, joinManagerMapping, copyContext);
    }

    @Override
    AbstractCommonQueryBuilder, SelectCTECriteriaBuilder, Void, BaseFinalSetOperationCTECriteriaBuilderImpl> copy(QueryContext queryContext, Map joinManagerMapping, ExpressionCopyContext copyContext) {
        return new RecursiveCTECriteriaBuilderImpl<>(this, queryContext.getParent().mainQuery, queryContext, joinManagerMapping, copyContext);
    }

    @Override
    public SelectCTECriteriaBuilderImpl union() {
        verifyBuilderEnded();
        unionAll = false;
        recursiveCteBuilder = new SelectCTECriteriaBuilderImpl(mainQuery, queryContext, cteKey, clazz, result, this, !mainQuery.dbmsDialect.supportsJoinsInRecursiveCte());
        return recursiveCteBuilder;
    }

    @Override
    public SelectCTECriteriaBuilderImpl unionAll() {
        verifyBuilderEnded();
        unionAll = true;
        recursiveCteBuilder = new SelectCTECriteriaBuilderImpl(mainQuery, queryContext, cteKey, clazz, result, this, !mainQuery.dbmsDialect.supportsJoinsInRecursiveCte());
        return recursiveCteBuilder;
    }

    @Override
    public void onReplaceBuilder(CTEInfoBuilder oldBuilder, CTEInfoBuilder newBuilder) {
        // Don't care about that
    }

    @Override
    public void onBuilderStarted(CTEInfoBuilder builder) {
        // Don't care about that
    }

    @Override
    public void onBuilderEnded(CTEInfoBuilder builder) {
        done = true;
        listener.onBuilderEnded(this);
    }
    
    public void verifyBuilderEnded() {
        if (recursiveCteBuilder != null && !done) {
            throw new BuilderChainingException("A builder was not ended properly.");
        }
    }

    @Override
    public CTEInfo createCTEInfo() {
        verifyBuilderEnded();
        List attributes = prepareAndGetAttributes();
        List columns = prepareAndGetColumnNames();
        
        // As a side effect, this will reorder selects according to attribute order
        recursiveCteBuilder.createCTEInfo();
        CTEInfo info = new CTEInfo(cteKey.getName(), cteKey.getOwner(), inline, cteType, attributes, columns, true, unionAll, this, recursiveCteBuilder);
        return info;
    }

}
 















© 2015 - 2024 Weber Informatics LLC | Privacy Policy