sql_ast.go

➜  ast go run select.go
package main

import (
        "strings"
        "fmt"
        "github.com/pingcap/parser/ast"
        "github.com/pingcap/parser"
        _ "github.com/pingcap/tidb/types/parser_driver"
)

type visitor struct{}

var (
        level = 0
        sp    = "++"
)

func (v *visitor) Enter(in ast.Node) (out ast.Node, skipChildren bool) {
        s := strings.Repeat("\t", level)
        fmt.Printf("L-%d %s%T %s\n", level, s, in, in.Text())
        level += 1
        return in, false
}

func (v *visitor) Leave(in ast.Node) (out ast.Node, ok bool) {
        level -= 1
        //s := strings.Repeat("\t", level)
        //fmt.Printf("L%d%s%T\n", level, s, in)
        return in, true
}