golang 获取调用者的方法名及所在源码行数

package main

import (
    "fmt"
    "runtime"
    "strings"
)

// CallerInfo 存储调用者信息的结构体
type CallerInfo struct {
    FuncName string // 方法名
    File     string // 文件名
    Line     int    // 行号
    Ok       bool   // 是否成功获取
}

// GetCallerInfo 获取调用者的完整信息（方法名、文件、行号）
func GetCallerInfo(skip int) CallerInfo {
    var info CallerInfo
    // 获取调用栈信息，skip 由外部指定，更灵活
    pc, file, line, ok := runtime.Caller(skip)
    if !ok {
        info.Ok = false
        return info
    }

    // 通过 pc 获取方法信息
    funcInfo := runtime.FuncForPC(pc)
    if funcInfo == nil {
        info.Ok = false
        return info
    }

    // 处理方法名：完整格式是 "包名.方法名"，直接保留即可；如果是匿名函数，会返回类似 "main.main.func1" 的格式
    funcName := funcInfo.Name()
    // 可选：如果只想保留方法名，去掉包名（根据需求选择）
    // parts := strings.Split(funcName, ".")
    // if len(parts) > 0 {
    //     funcName = parts[len(parts)-1]
    // }

    // 处理文件名：只保留最后一段
    shortFile := file
    for i := len(file) - 1; i >= 0; i-- {
        if file[i] == '/' {
            shortFile = file[i+1:]
            break
        }
    }

    info = CallerInfo{
        FuncName: funcName,
        File:     shortFile,
        Line:     line,
        Ok:       true,
    }
    return info
}

// 模拟日志工具函数
func LogInfo(msg string) {
    // skip=2：跳过 LogInfo（1）和 GetCallerInfo（2），获取调用 LogInfo 的函数信息
    caller := GetCallerInfo(2)
    if caller.Ok {
        fmt.Printf("[%s:%d %s] %s\n", caller.File, caller.Line, caller.FuncName, msg)
    } else {
        fmt.Printf("[未知位置] %s\n", msg)
    }
}

// 模拟业务模块
func UserService() {
    LogInfo("用户查询成功")
}

func main() {
    UserService()
}

package main

import (
    "fmt"
    "runtime"
    "strings"
)

// CallerOption 自定义配置选项
type CallerOption struct {
    Skip        int  // 跳过的层级
    SimplifyFunc bool // 是否简化方法名（去掉包名）
}

// GetCaller 获取调用者信息，支持自定义配置
func GetCaller(opt CallerOption) (funcName, file string, line int, ok bool) {
    // 基础校验：skip 不能小于 1（至少跳过当前函数）
    if opt.Skip < 1 {
        opt.Skip = 1
    }

    pc, file, line, ok := runtime.Caller(opt.Skip)
    if !ok {
        return "", "", 0, false
    }

    funcInfo := runtime.FuncForPC(pc)
    if funcInfo == nil {
        return "", "", 0, false
    }

    funcName = funcInfo.Name()
    // 简化方法名：去掉包名部分
    if opt.SimplifyFunc {
        parts := strings.Split(funcName, ".")
        if len(parts) > 0 {
            funcName = parts[len(parts)-1]
        }
    }

    // 简化文件名：只保留最后一段
    shortFile := file
    lastSlashIdx := strings.LastIndex(file, "/")
    if lastSlashIdx != -1 {
        shortFile = file[lastSlashIdx+1:]
    }

    return funcName, shortFile, line, true
}

// 跨包调用场景模拟（实际项目中可放在单独工具包）
func ToolFunc() {
    // 配置：跳过 1 层（ToolFunc），简化方法名
    funcName, file, line, ok := GetCaller(CallerOption{
        Skip:        1,
        SimplifyFunc: true,
    })
    if ok {
        fmt.Printf("工具函数调用者：%s（%s:%d）\n", funcName, file, line)
    }
}

// 匿名函数场景测试
func AnonymousFuncTest() {
    // 匿名函数中调用工具
    func() {
        ToolFunc()
    }()
}

func main() {
    // 1. 普通调用场景
    funcName, file, line, ok := GetCaller(CallerOption{
        Skip:        1,
        SimplifyFunc: false,
    })
    if ok {
        fmt.Printf("普通调用者：%s（%s:%d）\n", funcName, file, line)
    }

    // 2. 跨包调用场景（这里模拟，实际是同包，效果一致）
    ToolFunc()

    // 3. 匿名函数场景
    AnonymousFuncTest()
}