src/sqlancer/databend/DatabendToStringVisitor.java - master - NikolayIvkin/sqlancer2

src/sqlancer/databend/DatabendToStringVisitor.java
100 строк3 KB
Yisong Han
fix(databend): mvn verify succeed
09 сен 2022, 19:09
09 сен 2022, 19:095f07a2d
package sqlancer.databend;

import sqlancer.common.ast.newast.NewToStringVisitor;
import sqlancer.common.ast.newast.Node;
import sqlancer.databend.ast.DatabendConstant;
import sqlancer.databend.ast.DatabendExpression;
import sqlancer.databend.ast.DatabendJoin;
import sqlancer.databend.ast.DatabendSelect;

public class DatabendToStringVisitor extends NewToStringVisitor<DatabendExpression> {

    @Override
    public void visitSpecific(Node<DatabendExpression> expr) {
        if (expr instanceof DatabendConstant) {
            visit((DatabendConstant) expr);
        } else if (expr instanceof DatabendSelect) {
            visit((DatabendSelect) expr);
        } else if (expr instanceof DatabendJoin) {
            visit((DatabendJoin) expr);
        } else {
            throw new AssertionError(expr.getClass());
        }
    }

    private void visit(DatabendJoin join) {
        visit(join.getLeftTable());
        sb.append(" ");
        sb.append(join.getJoinType());
        sb.append(" ");
        if (join.getOuterType() != null) {
            sb.append(join.getOuterType());
        }
        sb.append(" JOIN ");
        visit(join.getRightTable());
        if (join.getOnCondition() != null) {
            sb.append(" ON ");
            visit(join.getOnCondition());
        }
    }

    private void visit(DatabendConstant constant) {
        sb.append(constant.toString());
    }

    // private void visitFromList(List<Node<DatabendExpression>> fromList) {
    // for (int i = 0; i < fromList.size(); i++) {
    // if (i != 0) {
    // sb.append(" INNER JOIN ");
    // }
    // visit(fromList.get(i));
    // }
    // }

    private void visit(DatabendSelect select) {
        sb.append("SELECT ");
        if (select.isDistinct()) {
            sb.append("DISTINCT ");
        }
        visit(select.getFetchColumns());
        sb.append(" FROM ");
        visit(select.getFromList());
        if (!select.getFromList().isEmpty() && !select.getJoinList().isEmpty()) {
            sb.append(", ");
        }
        if (!select.getJoinList().isEmpty()) {
            visit(select.getJoinList());
        }
        if (select.getWhereClause() != null) {
            sb.append(" WHERE ");
            visit(select.getWhereClause());
        }
        if (!select.getGroupByExpressions().isEmpty()) {
            sb.append(" GROUP BY ");
            visit(select.getGroupByExpressions());
        }
        if (select.getHavingClause() != null) {
            sb.append(" HAVING ");
            visit(select.getHavingClause());
        }
        if (!select.getOrderByExpressions().isEmpty()) {
            sb.append(" ORDER BY ");
            visit(select.getOrderByExpressions());
        }
        if (select.getLimitClause() != null) {
            sb.append(" LIMIT ");
            visit(select.getLimitClause());
        }
        if (select.getOffsetClause() != null) {
            sb.append(" OFFSET ");
            visit(select.getOffsetClause());
        }
    }

    public static String asString(Node<DatabendExpression> expr) {
        DatabendToStringVisitor visitor = new DatabendToStringVisitor();
        visitor.visit(expr);
        return visitor.get();
    }

}