抽象语法树扫描

Question

我正在尝试从go AST提取函数调用

$ cat main.go
package main
import ("fmt"; "go/ast"; "go/token"; "go/parser"; "io/ioutil")

func CalledFuncs(n ast.Node) bool {
    switch n.(type) {
    case *ast.CallExpr:
        l := ast.CallExpr(n).Lparen
        r := ast.CallExpr(n).Right
        fmt.Printf("call %d( ... %d)\n",l,r)
    }
    return true
}

func main() {
    fset := token.NewFileSet()
    src,_ := ioutil.ReadFile("simple.go")
    f, _ := parser.ParseFile(fset,">>",src,0)
    ast.Inspect(f, CalledFuncs)
}

但是go编译器抱怨说我不能进行转换：

$ go build -o main main.go
# command-line-arguments
./main.go:7:26: cannot convert n (type ast.Node) to type ast.CallExpr
./main.go:8:26: cannot convert n (type ast.Node) to type ast.CallExpr

我可能漏掉了什么。

Answer 1

开关

TypeSwitchGuard可能包括一个短变量声明。当使用该表单时，变量将在TypeSwitchCase末尾的每个子句的隐块中声明。在用大小写列出一种类型的子句中，变量具有该类型；否则，变量具有TypeSwitchGuard中表达式的类型。

这意味着您可以执行以下操作：

func CalledFuncs(n ast.Node) bool {
    switch x := n.(type) {
    case *ast.CallExpr:
        l := x.Lparen
        r := x.Rparen
        fmt.Printf("call %d( ... %d)\n", l, r)
    }
    return true
}