如何控制Go test输出 为日志添加自定义前缀实现请求ID检索

Go 原生go test命令目前没有提供内置配置项可直接给所有输出行添加统一自定义前缀。针对AWS Lambda场景下按请求ID关联全量测试日志的需求，可通过以下方式实现：

方案1：输出流包装（推荐） #

该方案无需修改测试代码，可覆盖所有输出行（包括框架打印的测试状态行、测试代码内打印的业务日志），完全匹配预期输出效果。
原理是启动go test进程时，捕获其标准输出、标准错误流，按行读取内容后拼接请求ID前缀再输出。

• 若通过Shell脚本启动测试，可直接借助awk实现行处理：

# 替换为运行时动态获取的Lambda请求ID
REQUEST_ID="your-lambda-request-id"
go test -v ./... 2>&1 | awk -v prefix="[$REQUEST_ID]    " '{print prefix $0}'

执行后输出效果如下：

[request-id]    === RUN   SomeTest
[request-id]    --- PASS: SomeTest (0.00s)

• 若通过代码在Lambda中调用exec启动go test进程，直接获取进程的stdout/stderr管道，按行读取内容、拼接前缀后写入CloudWatch日志即可。

方案2：测试代码内配置（仅覆盖业务日志） #

如果仅需给测试代码内主动打印的业务日志加前缀，可通过包级别的TestMain全局配置日志前缀，注意该方案无法覆盖go test框架自动打印的=== RUN/--- PASS这类状态行：

func TestMain(m *testing.M) {
    // 从Lambda运行时注入的环境变量读取请求ID
    reqID := os.Getenv("AWS_LAMBDA_REQUEST_ID")
    // 配置日志前缀
    log.SetFlags(0)
    log.SetPrefix(fmt.Sprintf("[%s]    ", reqID))
    // 执行测试
    os.Exit(m.Run())
}

注意事项 #

执行go test时需要添加-v参数，否则默认仅在测试失败时输出详细日志，添加该参数才会打印所有测试用例的执行状态行，满足全量日志检索需求。

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