如何使用 go/analysis 查找 Ident 的声明?

zhi*_*ong 5 static-analysis abstract-syntax-tree go go-toolchain

我使用 go/analysis 创建自己的静态分析工具。我仍然不知道如何从ast.Ident中找到def信息。

这是我的测试数据

package randomcheck
func xxx() {
}
func demo()  {
    xxx()
}

Run Code Online (Sandbox Code Playgroud)

还有我自己的分析仪

import (
    "fmt"
    "go/ast"
    "golang.org/x/tools/go/analysis"
    "golang.org/x/tools/go/analysis/passes/inspect"
)

var name string // -name flag
var Analyzer = &analysis.Analyzer{
    Name:     "fft",
    Requires: []*analysis.Analyzer{inspect.Analyzer},
    Run:      run,
}
//pass.Fset.Position(name.Pos())
func run(pass *analysis.Pass) (interface{}, error) {
    for _, f := range pass.Files {
        ast.Inspect(f, func(node ast.Node) bool {
            name,ok := node.(*ast.Ident)
            if !ok {
                return true
            }
            if name == nil {
                return true
            }
            if pass.TypesInfo.Defs[name] != nil {
                fmt.Println("def: " ,name)
            } else {
                fmt.Println("use: ", name)
            }
            return true
        })
    }

    return nil, nil
}


output:

use:  randomcheck
def:  xxx
def:  demo
use:  xxx

Run Code Online (Sandbox Code Playgroud)

我需要直接从use: xxx找到 def info def:xxx ,但我在 pass.TypesInfo 中找不到有用的信息

Eli*_*sky 2

您在寻找ObjectOf方法吗?这是经过一些修改的版本:

func run(pass *analysis.Pass) (interface{}, error) {
    for _, f := range pass.Files {
        ast.Inspect(f, func(node ast.Node) bool {
            name, ok := node.(*ast.Ident)
            if !ok {
                return true
            }
            if name == nil {
                return true
            }

            fmt.Println("ident:", nodeString(node, pass.Fset))
            obj := pass.TypesInfo.ObjectOf(name)
            fmt.Println(obj)
            if obj != nil {
                fmt.Println("  pos:", pass.Fset.Position(obj.Pos()))
            }
            return true
        })
    }

    return nil, nil
}

// nodeString formats a syntax tree in the style of gofmt.
func nodeString(n ast.Node, fset *token.FileSet) string {
    var buf bytes.Buffer
    format.Node(&buf, fset, n)
    return buf.String()
}
Run Code Online (Sandbox Code Playgroud)

当在示例输入文件上运行时,它显示:

ident: randomcheck
<nil>
ident: xxx
func command-line-arguments.xxx()
  pos: /home/eliben/temp/randomcheck.go:3:6
ident: demo
func command-line-arguments.demo()
  pos: /home/eliben/temp/randomcheck.go:5:6
ident: xxx
func command-line-arguments.xxx()
  pos: /home/eliben/temp/randomcheck.go:3:6
Run Code Online (Sandbox Code Playgroud)

请注意,最后一个 idxxx是作为对顶级函数xxx()及其正确位置等的引用而找到的。