com.espertech.esper.epl.parse.ASTGroupByHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of esper Show documentation
Complex event processing and event series analysis component
There is a newer version: 7.1.0
/*
 * *************************************************************************************
 *  Copyright (C) 2006-2015 EsperTech, Inc. All rights reserved.                       *
 *  http://www.espertech.com/esper                                                     *
 *  http://www.espertech.com                                                           *
 *  ---------------------------------------------------------------------------------- *
 *  The software in this package is published under the terms of the GPL license       *
 *  a copy of which has been included with this distribution in the license.txt file.  *
 * *************************************************************************************
 */

package com.espertech.esper.epl.parse;

import com.espertech.esper.epl.expression.core.ExprNode;
import com.espertech.esper.epl.generated.EsperEPL2GrammarParser;
import com.espertech.esper.epl.spec.*;
import org.antlr.v4.runtime.tree.Tree;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class ASTGroupByHelper {
    public static void walkGroupBy(EsperEPL2GrammarParser.GroupByListExprContext ctx, Map astExprNodeMap, List groupByExpressions) {
        List choices = ctx.groupByListChoice();
        for (EsperEPL2GrammarParser.GroupByListChoiceContext choice : choices) {
            GroupByClauseElement element = walkChoice(choice, astExprNodeMap);
            groupByExpressions.add(element);
        }
    }

    private static GroupByClauseElement walkChoice(EsperEPL2GrammarParser.GroupByListChoiceContext choice, Map astExprNodeMap) {
        if (choice.e1 != null) {
            ExprNode expr = ASTExprHelper.exprCollectSubNodes(choice.e1, 0, astExprNodeMap).get(0);
            return new GroupByClauseElementExpr(expr);
        }
        if (choice.groupByCubeOrRollup() != null) {
            return walkCubeOrRollup(choice.groupByCubeOrRollup(), astExprNodeMap);
        }
        return walkGroupingSets(choice.groupByGroupingSets().groupBySetsChoice(), astExprNodeMap);

    }

    private static GroupByClauseElement walkCubeOrRollup(EsperEPL2GrammarParser.GroupByCubeOrRollupContext ctx, Map astExprNodeMap) {
        boolean cube = ctx.CUBE() != null;
        List combinables = walkCombinables(ctx.groupByCombinableExpr(), astExprNodeMap);
        return new GroupByClauseElementRollupOrCube(cube, combinables);
    }

    private static List walkCombinables(List ctxs, Map astExprNodeMap) {
        List elements = new ArrayList();
        for (EsperEPL2GrammarParser.GroupByCombinableExprContext ctx : ctxs) {
            elements.add(walkCombinable(ctx, astExprNodeMap));
        }
        return elements;
    }

    private static GroupByClauseElement walkCombinable(EsperEPL2GrammarParser.GroupByCombinableExprContext ctx, Map astExprNodeMap) {
        if (ctx.e1 != null && ctx.LPAREN() == null) {
            ExprNode expr = ASTExprHelper.exprCollectSubNodes(ctx.e1, 0, astExprNodeMap).get(0);
            return new GroupByClauseElementExpr(expr);
        }
        List combined = ASTExprHelper.exprCollectSubNodes(ctx, 0, astExprNodeMap);
        return new GroupByClauseElementCombinedExpr(combined);
    }

    private static GroupByClauseElementGroupingSet walkGroupingSets(List ctxs, Map astExprNodeMap) {
        List elements = new ArrayList();
        for (EsperEPL2GrammarParser.GroupBySetsChoiceContext ctx : ctxs) {
            if (ctx.groupByCubeOrRollup() != null) {
                elements.add(walkCubeOrRollup(ctx.groupByCubeOrRollup(), astExprNodeMap));
            }
            else {
                elements.add(walkCombinable(ctx.groupByCombinableExpr(), astExprNodeMap));
            }
        }
        return new GroupByClauseElementGroupingSet(elements);
    }
}