Перебор переменных Голанга

Question

Я пытаюсь статически анализировать файлы Go.Для этого мне нужно проанализировать переменные в следующем формате:

shape.color = color.red

Мне нужно найти все переменные, к которым обращаются точки.Например, мне нужно знать, что переменная формы имеет атрибут цвета.А также нужно, чтобы переменная цвета имела красный атрибут.Я пытаюсь использовать пакет go / ast и go / parser, но не могу найти способ сделать это.

NB Если это что-то вроде shape.color (), то есть метод, то это не должно бытьподсчитывали

Answer 1

Если вы сравниваете две переменные go, которые вы не знаете заранее, вам нужно будет использовать отражение.Это позволит вам рефлексивно сравнить два поля:

type color struct {
    Red string
}

type shape struct {
    Color string
}

func main() {
    color := color{Red: "red"}
    shape := shape{Color: "red"}

    colorVal := reflect.ValueOf(color)
    shapeVal := reflect.ValueOf(shape)

    colorRedField := colorVal.FieldByName("Red")
    shapeColorField := shapeVal.FieldByName("Color")

    fmt.Println(colorRedField)
    fmt.Println(shapeColorField)
    fmt.Println(colorRedField.Interface() == shapeColorField.Interface())
}

https://play.golang.org/p/gvTJYwStP1O

Answer 2

Ах!Следующий код распечатывает все переменные, к которым обращаются точки:! 1001 *

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
}

Answer 3

Похоже, вы пытаетесь создать свой собственный AST, поскольку правая часть выражения, которое вы дали, не похожа на переменную, в противном случае я предполагаю, что это структура.Однако это тоже не имеет смысла, поскольку было бы буквально нелогично помещать поле с именем red в структуру с именем color.Также кажется, что вы пытаетесь получить доступ к переменной пакета, но это также не сработает, потому что первая буква в нижнем регистре означает, что объект не экспортируется.

Оставив все в стороне, я написал небольшой фрагментпросто для соблюдения перечисленных вами условий.

• переменная формы имеет атрибут color.
• переменная цвета имеет атрибут red.

---

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)
    }
}