Method VisitQuery

query/sql/sql_parser.go:210–251

********************** Visit SQL grammar starts ******************** ********************** query expressions ******************** VisitQuery visits the node

(ctx *antlrgen.QueryContext)

// VisitQuery visits the node
func (v *ASTBuilder) VisitQuery(ctx *antlrgen.QueryContext) interface{} {
	v.Logger.Debugf("VisitQuery: %s", ctx.GetText())

	location := v.getLocation(ctx)
	level, levelWith, levelQuery := v.getCtxLevels(v.SQL2AqlCtx)
	if levelQuery >= maxLevelQuery {
		panic(fmt.Errorf("only support %v level subquery at (line:%d, col:%d)",
			maxLevelQuery, location.Line, location.CharPosition))
	}

	// handle with
	v.setCtxLevels(v.SQL2AqlCtx, level, levelWith, levelQuery)
	classWith := reflect.TypeOf((*tree.With)(nil))
	with := v.visitIfPresent(ctx.With(), classWith).(*tree.With)

	// handle queryNoWith
	v.setCtxLevels(v.SQL2AqlCtx, level, levelWith, levelQuery)
	body, _ := v.Visit(ctx.QueryNoWith()).(*tree.Query)
	if body == nil {
		panic(fmt.Errorf("missing queryNoWith body at (line:%d, col:%d)", location.Line, location.CharPosition))
	}

	if levelQuery == 0 {
		if valid, err := v.isValidWithOrSubQuery(v.SQL2AqlCtx); !valid || err != nil {
			panic(fmt.Errorf("line:%d, col:%d isValidWithOrSubQuery: %v, reason :%v",
				location.Line, location.CharPosition, valid, err))
		}
	}

	query := tree.NewQuery(
		v.getLocation(ctx),
		with,
		body.QueryBody,
		body.OrderBy,
		body.Limit,
	)
	query.SetValue(fmt.Sprintf("Query: (%s)", v.getText(ctx.BaseParserRuleContext)))

	// reset SQL2AqlContext
	v.setCtxLevels(v.SQL2AqlCtx, level-1, levelWith, levelQuery)
	return query
}

// VisitWith visits the node
func (v *ASTBuilder) VisitWith(ctx *antlrgen.WithContext) interface{} {