org.voltdb.plannerv2.utils.PostBuildVisitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of voltdb Show documentation
VoltDB server libraries
There is a newer version: 10.1.1
/* This file is part of VoltDB.
 * Copyright (C) 2008-2020 VoltDB Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with VoltDB.  If not, see .
 */

package org.voltdb.plannerv2.utils;

import org.apache.calcite.sql.type.SqlTypeName;
import org.voltdb.expressions.AbstractExpression;
import org.voltdb.expressions.ParameterValueExpression;
import org.voltdb.plannerv2.ColumnTypes;
import org.voltdb.plannerv2.converter.RexConverter;
import org.voltdb.plannodes.AbstractPlanNode;
import org.voltdb.plannodes.LimitPlanNode;

import java.util.HashSet;
import java.util.Set;

/**
 * A Visitor to iterate over a built plan to collect various information about the plan
 */
public class PostBuildVisitor extends AbstractPlanNodeVisitor {

    private boolean m_hasLimitOffset = false;
    // TODO: set the flag instead of default value
    private boolean m_isOrderDeterministic = false;
    private Set m_pveSet = new HashSet<>();

    @Override
    public void visitNode(AbstractPlanNode node) {
        if (node instanceof LimitPlanNode) {
            m_hasLimitOffset = true;
            // add the Limit Parameter
            if (((LimitPlanNode) node).getLimitParameterIndex() >= 0) {
                ParameterValueExpression pve = new ParameterValueExpression();
                pve.setParameterIndex((int) ((LimitPlanNode) node).getLimitParameterIndex());
                // we don't care about precision for ParameterValueExpression
                RexConverter.setType(pve, ColumnTypes.getVoltType(SqlTypeName.INTEGER), 0);
                m_pveSet.add(pve);
            }
            // add the Offset Parameter
            if (((LimitPlanNode) node).getOffsetParameterIndex() >= 0) {
                ParameterValueExpression pve = new ParameterValueExpression();
                pve.setParameterIndex((int) ((LimitPlanNode) node).getOffsetParameterIndex());
                // we don't care about precision for ParameterValueExpression
                RexConverter.setType(pve, ColumnTypes.getVoltType(SqlTypeName.INTEGER), 0);
                m_pveSet.add(pve);
            }
        }
        // Collect pve
        Set pves = new HashSet<>();
        node.findAllExpressionsOfClass(ParameterValueExpression.class, pves);
        for (AbstractExpression pve : pves) {
            m_pveSet.add((ParameterValueExpression) pve);
        }

        super.visitNode(node);
    }

    public boolean hasLimitOffset() {
        return m_hasLimitOffset;
    }

    public boolean isOrderDeterministic() {
        return m_isOrderDeterministic;
    }

    ParameterValueExpression[] getParameterValueExpressions() {
        return m_pveSet.toArray(new ParameterValueExpression[m_pveSet.size()]);
    }
}