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

Question

情况

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

问题

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

即，

以下不是自定义类型

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

但以下是自定义类型

Integer,PhoneNumber,*PhoneNumber

我想我可以使用如下函数来完成此操作，该函数查找完全匹配并可能添加 map ，[] 支持

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