分析Golang的变量

本文介绍了一种方法，用于解析 Go 代码中的变量，特别是使用点表示法访问的变量。本文使用 go/ast 和 go/parser 包来遍历代码的抽象语法树 (AST)，并识别符合特定条件的变量。这些条件包括：变量使用点表示法访问，并且左侧变量具有特定属性，而右侧变量具有特定值。通过这种方法，可以从 Go 代码中提取变量及其属性和值之间的关系。

我正在尝试静态分析 go 文件。为此，我需要解析以下格式的变量：

shape.color = color.red

我需要找到所有用点表示法访问的变量。例如，我需要知道形状变量具有颜色属性。并且还需要颜色变量具有红色属性。 我尝试使用 go/ast 和 go/parser 包，但无法找到方法。

注意如果它是类似于 shape.color() 的东西，即一个方法，那么它不应该被计算在内

解决方案

啊！以下代码打印所有使用点表示法访问的变量！

package main

import (
    "fmt"
    "go/ast"
    "go/parser"
    "go/token"
    "log"
)

func main() {
    v := visitor{}
    filename := "test.go"

    fset := token.newfileset()
    f, err := parser.parsefile(fset, filename, nil, 0)

    if err != nil {
        log.fatal(err)
    }

    ast.walk(&v, f)
}

type visitor struct {
}

func (v *visitor) visit(n ast.node) ast.visitor {
    if n == nil {
        return v
    }

    if selectorexp, ok := n.(*ast.selectorexpr); ok {
        if x, ok := selectorexp.x.(*ast.ident); ok {
            if x.obj == nil {
                return v
            }  

            fmt.printf("%s.%s\n", x.name, selectorexp.sel.name)
        }
    }
    return v
}

看起来您正在尝试创建自己的 ast，因为您给出的表达式的右侧看起来不像变量，否则我假设它是一个结构体。但是，这也没有意义，因为将名为 red 的字段放入名为 color 的结构中实际上是不合逻辑的。看起来您正在尝试访问包的变量，但这也不起作用，因为小写的首字母意味着该实体未导出。

把所有这些都放在一边，我写了一个小片段只是为了遵守您列出的条件。

• shape 变量具有颜色属性。
• 颜色变量具有红色属性。

https://play.golang.org/p/gIpctQ1XSgT，我只将其改编为一行，并且每当不满足条件时就会感到恐慌。请随意根据您的需要进行调整。

package main

import (
    "go/ast"
    "go/format"
    "go/parser"
    "go/token"
    "os"
)

func main() {
    expr, err := parser.ParseExpr("shape.color==color.red")
    if err != nil {
        panic(err)
    }

    // Checking if the expression was binary.
    bExpr, ok := expr.(*ast.BinaryExpr)
    if !ok {
        panic("expr is not a binary expr.")
    }

    // If the operation is not “==”, die.
    if bExpr.Op != token.EQL {
        panic("the op should have been ==.")
    }

    // Left must be a selector expr, meaning followed with a selector which is “dot” in this case.
    left, ok := bExpr.X.(*ast.SelectorExpr)
    if !ok {
        panic("left should have been a selector expr.")
    }

    // Same as above.
    right, ok := bExpr.Y.(*ast.SelectorExpr)
    if !ok {
        panic("right should have been a selector expr.")
    }

    // Checking for attributes.
    if left.Sel.Name != "color" {
        panic("left should have had a color attr.")
    }

    // Same as above.
    if right.Sel.Name != "red" {
        panic("right should have had a red attr.")
    }

    // Then we finally gofmt the code and print it to stdout.
    if err := format.Node(os.Stdout, token.NewFileSet(), expr); err != nil {
        panic(err)
    }
}

以上就是《分析Golang的变量》的详细内容，更多关于的资料请关注golang学习网公众号！