parsing - 使用 "go/parser"检查表达式是否为自定义类型

Question

情况

编写一个代码生成器来检查结构的字段并使用结构标签添加验证函数

问题

这里我需要检查type字段是否为自定义类型

即，

以下不是自定义类型

int, []int,*int,[]Integer,map[string]PhoneNumber 

但以下是自定义类型

Integer,PhoneNumber,*PhoneNumber

我想我可以使用像下面这样的函数来查找完全匹配并可以添加映射，[] 支持

func isBuiltInType(typ string) bool {
    switch typ {
            case "bool", "byte", "complex128", "complex64", "error": 
            case "float32", "float64":
            case "int", "int16", "int32", "int64", "int8":
            case "rune", "string":
            case "uint", "uint16", "uint32", "uint64", "uint8", "uintptr":
            default:
                return false
    }
    return true
}

但是有没有更好的方法使用 parse.ParseExpr 等等

Answer 1

对于任何一种可靠的结果，您都需要使用 go/types 包来使用 Go 的类型检查器。使用起来并不简单，但在 https://golang.org/s/types-tutorial 处有一篇有用的介绍性文章。 .

我拼凑了一个示例程序，因此您可以看到会发生什么。重要的一点是 isBasic 函数，其余只是使其可执行的样板。特别是，AST 遍历是针对特定示例源代码进行硬编码的。我假设您已经为此准备了自己的代码。

关键是 types.Info 结构包含实现您自己的逻辑所需的所有类型信息。

我找到了 github.com/fatih/astrewrite和 golang.org/x/tools/go/loader在处理代码生成和/或解析时很有帮助(加载程序包是完整类型信息所必需的)。

https://play.golang.org/p/A9hdPy-Oy-

package main

import (
    "bufio"
    "fmt"
    "go/ast"
    "go/parser"
    "go/token"
    "go/types"
    "log"
    "strings"
)

var src = strings.TrimSpace(`
package main

type T struct{}

func f() {
    var _ T
    var _ *T

    var _ int
    var _ *int
    var _ **int
    var _ []int
    var _ []T
    var _ map[string]int
    var _ map[string]T
}
`)

func main() {
    // Parse source
    fset := token.NewFileSet()
    f, err := parser.ParseFile(fset, "src.go", src, 0)
    if err != nil {
            log.Fatal(err)
    }

    // Run type checker
    info := types.Info{Types: make(map[ast.Expr]types.TypeAndValue)}

    _, err = (&types.Config{}).Check("mypkg", fset, []*ast.File{f}, &info)
    if err != nil {
            log.Fatal(err)
    }

    printSrc()

    // Inspect variable types in f()
    for _, varDecl := range f.Decls[1].(*ast.FuncDecl).Body.List {
            value := varDecl.(*ast.DeclStmt).Decl.(*ast.GenDecl).Specs[0].(*ast.ValueSpec)

            pos := fset.Position(value.Type.Pos())
            typ := info.Types[value.Type].Type

            fmt.Println(pos, "basic:", isBasic(typ))
    }
}

func isBasic(t types.Type) bool {
    switch x := t.(type) {
    case *types.Basic:
            return true
    case *types.Slice:
            return true
    case *types.Map:
            return true
    case *types.Pointer:
            return isBasic(x.Elem())
    default:
            return false
    }
}

func printSrc() {
    s := bufio.NewScanner(strings.NewReader(src))
    for i := 1; s.Scan(); i++ {
            fmt.Printf("%2d: %s\n", i, s.Text())
    }
    fmt.Println("")
}