0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
22 lines
418 B
Go
22 lines
418 B
Go
package astutils
|
|
|
|
import "github.com/sqlc-dev/sqlc/internal/sql/ast"
|
|
|
|
type nodeSearch struct {
|
|
list *ast.List
|
|
check func(ast.Node) bool
|
|
}
|
|
|
|
func (s *nodeSearch) Visit(node ast.Node) Visitor {
|
|
if s.check(node) {
|
|
s.list.Items = append(s.list.Items, node)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func Search(root ast.Node, f func(ast.Node) bool) *ast.List {
|
|
ns := &nodeSearch{check: f, list: &ast.List{}}
|
|
Walk(ns, root)
|
|
return ns.list
|
|
}
|