com.blazebit.persistence.impl.MultipleSubqueryInitiatorImpl 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!
package com.blazebit.persistence.impl;

import com.blazebit.persistence.MultipleSubqueryInitiator;
import com.blazebit.persistence.SubqueryInitiator;
import com.blazebit.persistence.impl.builder.expression.ExpressionBuilder;
import com.blazebit.persistence.impl.builder.expression.ExpressionBuilderEndedListener;
import com.blazebit.persistence.impl.expression.Expression;
import com.blazebit.persistence.impl.expression.SubqueryExpression;

public class MultipleSubqueryInitiatorImpl extends SubqueryBuilderListenerImpl implements MultipleSubqueryInitiator, ExpressionBuilder {

    private final T result;
    private final ExpressionBuilderEndedListener listener;
    private final SubqueryInitiatorFactory subqueryInitFactory;
    private Expression expression;
    private String subqueryAlias;
    private SubqueryInitiator subqueryStartMarker;
    
    public MultipleSubqueryInitiatorImpl(T result, Expression expression, ExpressionBuilderEndedListener listener, SubqueryInitiatorFactory subqueryInitFactory) {
        this.result = result;
        this.expression = expression;
        this.listener = listener;
        this.subqueryInitFactory = subqueryInitFactory;
    }

    @Override
    @SuppressWarnings("unchecked")
    public SubqueryInitiator> with(String subqueryAlias) {
        verifySubqueryBuilderEnded();
        this.subqueryAlias = subqueryAlias;
        // The cast with the type parameter sucks but I don't want to spend too much time with that right now 
        return startSubqueryInitiator(subqueryInitFactory.createSubqueryInitiator((MultipleSubqueryInitiator) this, (SubqueryBuilderListener>) this));
    }

    @Override
    public T end() {
        if (listener != null) {
            listener.onBuilderEnded(this);
        }
        
        return result;
    }

    @Override
    public Expression getExpression() {
        return expression;
    }

    @Override
    public void onBuilderEnded(SubqueryInternalBuilder builder) {
        super.onBuilderEnded(builder);
        expression = ExpressionUtils.replaceSubexpression(expression, subqueryAlias, new SubqueryExpression(builder));
        subqueryStartMarker = null;
        subqueryAlias = null;
    }

    @Override
    public void verifySubqueryBuilderEnded() {
        if (subqueryStartMarker != null) {
            throw new BuilderChainingException("A builder was not ended properly.");
        }
        super.verifySubqueryBuilderEnded();
    }

    public  SubqueryInitiator startSubqueryInitiator(SubqueryInitiator subqueryInitiator) {
        this.subqueryStartMarker = subqueryInitiator;
        return subqueryInitiator;
    }

}