你遇到的问题是因为Go的ast解析器会将原结构体上方的注释关联到对应的GenDecl节点，但插入新结构体声明后，printer模块会根据注释的位置信息（token.Pos）错误地将注释插入到新结构体声明的中间——这是因为新生成的AST节点没有设置有效的位置信息，导致注释的归属关系被打乱。

Go标准库提供了ast.CommentMap来明确注释与AST节点的关联关系，通过它可以确保注释始终属于原结构体。以下是修改后的代码：

package main

import (
    "bytes"
    "fmt"
    "go/ast"
    "go/parser"
    "go/printer"
    "go/token"
)

func main() {
    src := `package demo

// Original comment
type Original struct {
    Name string
}`

    fset := token.NewFileSet()
    file, err := parser.ParseFile(fset, "input.go", src, parser.ParseComments)
    if err != nil {
        panic(err)
    }

    original := "Original"
    newName := "Embed"

    var originalGen *ast.GenDecl
    // 找到原结构体的GenDecl
    for i, decl := range file.Decls {
        gen, ok := decl.(*ast.GenDecl)
        if !ok || gen.Tok != token.TYPE {
            continue
        }

        for _, spec := range gen.Specs {
            ts, ok := spec.(*ast.TypeSpec)
            if !ok || ts.Name.Name != original {
                continue
            }
            originalGen = gen

            // 创建新的嵌入结构体声明
            newDecl := &ast.GenDecl{
                Tok: token.TYPE,
                Specs: []ast.Spec{
                    &ast.TypeSpec{
                        Name: ast.NewIdent(newName),
                        Type: &ast.StructType{
                            Fields: &ast.FieldList{
                                List: []*ast.Field{
                                    {Type: ast.NewIdent(original)},
                                },
                            },
                        },
                    },
                },
            }

            // 插入新声明到原结构体之前
            file.Decls = append(file.Decls[:i], append([]ast.Decl{newDecl}, file.Decls[i:]...)...)
            goto done
        }
    }

done:
    // 使用CommentMap确保原注释仅关联到原结构体的GenDecl
    cm := ast.NewCommentMap(fset, file, file.Comments)
    // 移除新声明与任何注释的关联
    delete(cm, newDecl)
    // 确保原注释只属于原GenDecl
    for node := range cm {
        if node != originalGen {
            filtered := []*ast.CommentGroup{}
            for _, cg := range cm[node] {
                if cg != originalGen.Doc {
                    filtered = append(filtered, cg)
                }
            }
            cm[node] = filtered
        }
    }
    // 将更新后的注释映射应用回文件
    file.Comments = cm.Comments()

    var buf bytes.Buffer
    if err := printer.Fprint(&buf, fset, file); err != nil {
        panic(err)
    }

    fmt.Println(buf.String())
    /*
        正确输出：
        package demo

        type Embed struct {
            Original
        }

        // Original comment
        type Original struct {
            Name string
        }
    */
}

1. 始终使用parser.ParseComments解析代码：确保注释被保留在AST中。
2. 用ast.CommentMap管理注释归属：这是标准库提供的官方工具，能精准控制注释与节点的关联，避免打印时出现位置错乱。
3. 手动设置节点的Doc/Comment字段：如果需要给新生成的节点添加注释，直接创建ast.CommentGroup并赋值给节点的Doc（上方注释）或Comment（后方注释）字段，同时通过CommentMap关联到节点。
4. 修改AST后更新CommentMap：插入、删除或移动节点后，必须更新CommentMap并重新应用到文件，确保注释的位置和归属正确。

内容的提问来源于stack exchange，提问作者Delta