如何打印手写的AST?

use*_*814 1 code-generation abstract-syntax-tree go

我有一组网络协议的 XML 描述,我试图从中生成 Go 代码,因此我没有任何现有的 Go 代码可以使用。所有使用go/ast(例如go fmt)的示例都采用现有代码,对 AST 进行一些转换,然后将它们写回。因为我只有 XML 文件,所以我正在使用的 AST 必须手工编写。问题是我无法让我的手写 AST 输出。

例子

package main

import (
    "go/ast"
    "go/printer"
    "go/token"
    "os"
)

func main() {
    f := ast.File{
        Name: ast.NewIdent("foo"),
        Decls: []ast.Decl{
            &ast.GenDecl{
                Tok: token.TYPE,
                Specs: []ast.Spec{
                    &ast.TypeSpec{
                        Name: ast.NewIdent("Bar"),
                        Type: ast.NewIdent("uint32"),
                    },
                },
            },
        },
    }
    fset := token.NewFileSet()
    printer.Fprint(os.Stdout, fset, f)
}
Run Code Online (Sandbox Code Playgroud)

预期输出:

package foo

type Bar uint32
Run Code Online (Sandbox Code Playgroud)

实际输出:无

如何让 AST 打印?

Dav*_*e C 5

不要忽略错误!

添加:

err := printer.Fprint(os.Stdout, fset, f)
if err != nil {
    log.Fatal(err)
}
Run Code Online (Sandbox Code Playgroud)

给出:“ go/printer: unsupported node type ast.File ”这应该足以为您指明正确的方向。

printer.Fprint的最后一个参数是interface{}这样编译器接受任何东西。然而,就像parser.ParseFile返回一个*ast.File(而不是ast.File)一样,它需要一个指向节点的指针。

传递一个指针给出你想要的输出(注意&ast.File):

package main

import (
    "go/ast"
    "go/printer"
    "go/token"
    "log"
    "os"
)

func main() {
    f := &ast.File{
        Name: ast.NewIdent("foo"),
        Decls: []ast.Decl{
            &ast.GenDecl{
                Tok: token.TYPE,
                Specs: []ast.Spec{
                    &ast.TypeSpec{
                        Name: ast.NewIdent("Bar"),
                        Type: ast.NewIdent("uint32"),
                    },
                },
            },
        },
    }
    fset := token.NewFileSet()
    err := printer.Fprint(os.Stdout, fset, f)
    if err != nil {
        log.Fatal(err)
    }
}
Run Code Online (Sandbox Code Playgroud)

playground