如何使用 AST 解析器提取 Golang 函数文档？

go get golang.org/x/tools/go/ast

import (
    "go/ast"
    "go/parser"
    "go/token"
)

func ParseFile(filePath string) (*ast.File, error) {
    fset := token.NewFileSet()
    return parser.ParseFile(fset, filePath, nil, parser.ParseComments)
}

func ExtractFunctionDocs(file *ast.File) map[string]string {
    docs := make(map[string]string)
    for _, decl := range file.Decls {
        if f, ok := decl.(*ast.FuncDecl); ok {
            if f.Doc != nil {
                docs[f.Name.Name] = f.Doc.Text()
            }
        }
    }
    return docs
}

package main

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

func main() {
    file, err := ParseFile("my_code.go")
    if err != nil {
        fmt.Println(err)
        return
    }
    docs := ExtractFunctionDocs(file)
    fmt.Println("Documentation for function 'greet':", docs[" greet"])
}

// greet says hello
func greet(name string) string {
    return "Hello, " + name
}

Documentation for function 'greet': // greet says hello