Блог

Я хочу извлечь сигнатуры функций, чтобы иметь возможность создавать над ними некоторые методы-оболочки. Для этого я использую golang.org/x/tools/go/packages то, что дает мне возможность читать AST.

Например, для определения функции func MyFunc(param int) вы получаете некоторые

 ast.FuncDecl{
    Type: *FieldList{
        List: []*Field{
            {
                Names: []*Ident{ /*...*/ },
                Type:  nil, /*...*/
            },
        },
    },
}
 

Где Тип представляет тип.

Я хочу сгенерировать специальный код для всех int параметров, но int также может быть скрыт с помощью некоторого объявления типа

 type MyType int
 

Как я могу преобразовать ast тип в реальный, который есть у компилятора?

Добавьте packages.NeedTypes и packages.NeedTypesInfo в режим загрузки. При этом у каждого загруженного пакета будет TypesInfo инициализировано свое поле, и тип этого поля *types.Info имеет поле, называемое Types полем, которое сопоставляет выражения ast типам. Вы можете использовать это следующим образом:

 func main() {
    loadConfig := new(packages.Config)
    loadConfig.Mode = packages.NeedSyntax | packages.NeedTypes | packages.NeedTypesInfo
    loadConfig.Fset = token.NewFileSet()
    pkgs, err := packages.Load(loadConfig, "syscall")
    if err != nil {
        panic(err)
    }

    for _, pkg := range pkgs {
        for _, syn := range pkg.Syntax {
            for _, dec := range syn.Decls {
                if fd, ok := dec.(*ast.FuncDecl); ok amp;amp; fd.Name.Name == "Kill" {
                    x1 := fd.Type.Params.List[0].Type // int
                    x2 := fd.Type.Params.List[1].Type // syscall.Signal

                    tv1 := pkg.TypesInfo.Types[x1]
                    tv2 := pkg.TypesInfo.Types[x2]

                    if basic, ok := tv1.Type.(*types.Basic); ok {
                        fmt.Printf("%#vn", basic) // int
                    }

                    if named, ok := tv2.Type.(*types.Named); ok {
                        fmt.Printf("%vn", named.Obj())         // *types.TypeName (Signal)
                        fmt.Printf("%#vn", named.Underlying()) // *types.Basic (int)
                    }
                }
            }
        }
    }
}