如何在 Go 自定义 fatal 函数中保留原始调用行号-Golang-PHP中文网

如何在 Go 自定义 fatal 函数中保留原始调用行号

Go 标准日志的 log.Fatal 默认显示日志语句所在行号，而非错误发生处；通过 runtime.Caller(1) 可获取调用者（如 main 中）的文件名与行号，并结合自定义 logger 实现行号“透传”。

go 标准日志的 `log.fatal` 默认显示日志语句所在行号，而非错误发生处；通过 `runtime.caller(1)` 可获取调用者（如 `main` 中）的文件名与行号，并结合自定义 logger 实现行号“透传”。

在 Go 中，log.Logger 的 Lshortfile 标志会自动插入调用 mylog.Fatal(...) 语句所在的文件与行号。当你封装 fatal(err) 函数时，该行号就变成了 fatal 函数内部的调用位置（如 fatal.go:12），而非真正出错的业务代码行（如 main.go:18）——这严重削弱了错误定位效率。

解决的关键在于：绕过 log 包的自动行号捕获机制，手动注入上游调用栈信息。runtime.Caller(1) 正是为此而生：它返回调用当前函数的上一级栈帧信息（1 表示跳过当前函数，取调用者）。

以下是推荐实现方式：

package main

import (
    "log"
    "os"
    "runtime"
)

var mylog = log.New(os.Stderr, "app: ", log.LstdFlags)

// fatal 检查 err，若非 nil 则以调用者（caller）的文件:行号格式记录并退出
func fatal(err error) {
    if err != nil {
        // 获取调用 fatal() 的位置（即 main.go:18 这类真实错误点）
        _, file, line, ok := runtime.Caller(1)
        if !ok {
            file = "???"
            line = 0
        }
        // 手动构造带 caller 信息的前缀，并禁用 log 自动 shortfile（避免冲突）
        // 注意：此处需移除 log.Lshortfile，否则会叠加两行号
        logger := log.New(os.Stderr, "app: "+file+":"+string(rune(line))+" ", log.LstdFlags)
        logger.Fatal(err)
    }
}

但上述方式每次新建 logger 效率低且前缀格式不统一。更优雅的做法是复用原 logger 并临时修改输出前缀——由于 log.Logger 不支持运行时修改 prefix，我们改用 log.SetPrefix + log.Output 组合实现精确控制：

使用Go语言搭建家庭相册系统-相关课件

下载

func fatal(err error) {
    if err != nil {
        _, file, line, ok := runtime.Caller(1)
        if !ok {
            file = "???"
            line = 0
        }
        // 构造标准格式前缀：YYYY/MM/DD HH:MM:SS file:line:
        // 注意：log.LstdFlags 已含时间戳，我们只补足 file:line 部分
        prefix := "app: " + file + ":" + strconv.Itoa(line) + ": "

        // 使用 log.Output(2, ...) 跳过 fatal 和本层，直接输出到 caller 的位置
        // 第一个参数 2 表示跳过 runtime.Caller(1) 和 fatal 本身，指向原始调用点
        log.SetPrefix(prefix)
        log.Output(2, err.Error()) // 注意：Output 不会自动加换行，Fatal 会 exit
        os.Exit(1)
    }
}

不过最简洁、生产可用的方案是完全接管日志输出格式，避开 log.Lshortfile 冲突：

package main

import (
    "fmt"
    "log"
    "os"
    "runtime"
    "time"
)

var mylog = log.New(os.Stderr, "", 0) // 关闭所有内置 flag

func fatal(err error) {
    if err != nil {
        _, file, line, ok := runtime.Caller(1)
        if !ok {
            file = "???"
            line = 0
        }
        // 手动拼接完整日志：时间 + 前缀 + 文件:行 + 错误
        now := time.Now().Format("2006/01/02 15:04:05")
        msg := fmt.Sprintf("app: %s %s:%d: %v\n", now, file, line, err)
        os.Stderr.WriteString(msg)
        os.Exit(1)
    }
}

func main() {
    _, err := os.Open("some-crazy-non-existent-file")
    fatal(err) // 输出示例：app: 2024/05/20 14:30:22 main.go:22: open some-crazy-non-existent-file: no such file or directory
}

✅ 注意事项：