go 标准库未提供直接提取 `go:generate` 注释的 api，需手动扫描源码行并解析以 `//go:generate` 开头的注释行，本文详解实现方法与注意事项。
在 Go 工程中，go:generate 是一种常用代码生成机制，其指令以 //go:generate 形式写在源文件注释中（如 //go:generate go run gen.go）。但标准库 go/doc 包不解析也不暴露这类指令——它仅提取文档注释（如 // Package xxx），忽略所有 go: 前缀的编译器/工具指令。
要可靠提取 go:generate 行，核心思路是：逐行扫描源文件内容，识别符合规范的注释行，并剥离注释前缀后提取指令。关键步骤如下：
- 读取源码字节切片（如通过 os.ReadFile 获取）；
- 使用 bufio.Scanner 按行扫描（避免一次性加载大文件导致内存压力）；
- 对每行调用辅助函数提取注释文本（需处理 // 单行注释、/* */ 块注释内首行等边界情况）；
- 判断是否以 go:generate 开头（注意末尾空格），并提取后续命令。
以下为精简可靠的实现示例：
import (
"bufio"
"bytes"
"strings"
)
// extractGenerateDirectives 从 Go 源码字节中提取所有 go:generate 指令
func extractGenerateDirectives(src []byte) []string {
var directives []string
scanner := bufio.NewScanner(bytes.NewReader(src))
for scanner.Scan() {
line := scanner.Text()
// 提取该行中的注释内容（支持 // 和 /* ... */ 中的首行）
comment := commentText(line)
if strings.HasPrefix(comment, "go:generate ") {
cmd := strings.TrimSpace(strings.TrimPrefix(comment, "go:generate "))
if cmd != "" {
directives = append(directives, cmd)
}
}
}
return directives
}
// commentText 提取 Go 源码行中的注释文本（参考 go/src/cmd/compile/internal/syntax/parser.go）
func commentText(line string) string {
// 跳过行首空白
line = strings.TrimSpace(line)
if len(line) == 0 {
return ""
}
// 处理 // 单行注释
if strings.HasPrefix(line, "//") {
return strings.TrimSpace(strings.TrimPrefix(line, "//"))
}
// 处理 /* ... */ 块注释（仅匹配行内完整块注释，简化版；生产环境建议用 go/scanner）
if idx := strings.Index(line, "/*"); idx >= 0 {
if end := strings.Index(line[idx:], "*/"); end >= 0 {
content := line[idx+2 : idx+end]
return strings.TrimSpace(content)
}
}
return ""
}⚠️ 注意事项：
- 此实现为轻量级方案，适用于大多数场景；若需 100% 符合 Go 语法（如嵌套注释、字符串字面量内误匹配），推荐使用 go/scanner 包进行词法分析；
- go:generate 必须严格位于 // 后、无前置字符（如 // foo //go:generate ... 不合法）；
- 官方 go generate 工具本身即采用类似逻辑（见 src/cmd/go/internal/generate/generate.go），可作为权威参考；
- 不要依赖 go/doc 或 go/ast 直接获取——go/ast 的 CommentGroup 保留原始注释，但需自行遍历节点并正则匹配，反而更复杂。
总结：解析 go:generate 并非难事，关键是理解其语义约束（纯行首注释、固定前缀），并选择合适抽象层级——简单项目用 bufio.Scanner + 字符串处理 足够；大型工具链建议复用 go/scanner 确保语法严谨性。
