1
0
mirror of https://github.com/golang/go synced 2024-11-25 08:57:58 -07:00

go/ast: add Inspect function for easy AST inspection w/o a visitor

R=rsc
CC=golang-dev
https://golang.org/cl/2770044
This commit is contained in:
Robert Griesemer 2010-11-01 15:06:34 -07:00
parent 035696c59a
commit 9384fdc96a

View File

@ -321,3 +321,22 @@ func Walk(v Visitor, node interface{}) {
v.Visit(nil) v.Visit(nil)
} }
type inspector func(node interface{}) bool
func (f inspector) Visit(node interface{}) Visitor {
if node != nil && f(node) {
return f
}
return nil
}
// Inspect traverses an AST in depth-first order: If node != nil, it
// invokes f(node). If f returns true, inspect invokes f for all the
// non-nil children of node, recursively.
//
func Inspect(ast interface{}, f func(node interface{}) bool) {
Walk(inspector(f), ast)
}