在Go中使用相对文件路径作为函数参数的实现难题

目录结构 #

src/
- main.go
data/
- example.csv
csv/
- parse.go
utils/
- mapFields.go

问题背景 #

现有utils/mapFields.go中的MapFields函数定义如下：

MapFields(csvInPath string, csvOutPath string) {...}

调用该函数时希望传入相对于调用者所在文件的相对路径（比如在src/main.go中调用时传入../data/example.csv），但函数内部无法自动基于调用者的文件路径拼接出正确的文件绝对路径，导致文件打开失败。

解决方案 #

利用Go标准库runtime获取调用者的文件路径，再结合path/filepath拼接出绝对路径，具体实现如下：

修改MapFields函数 #

在utils/mapFields.go中更新函数逻辑：

package utils

import (
    "fmt"
    "path/filepath"
    "runtime"
)

func MapFields(csvInPath string, csvOutPath string) error {
    // 获取直接调用者的文件路径（参数1表示跳过当前函数，取上一层调用者）
    _, callerFile, _, ok := runtime.Caller(1)
    if !ok {
        return fmt.Errorf("无法获取调用者文件路径")
    }

    // 提取调用者文件所在的目录
    callerDir := filepath.Dir(callerFile)

    // 将传入的相对路径与调用者目录拼接为绝对路径
    absInPath := filepath.Join(callerDir, csvInPath)
    absOutPath := filepath.Join(callerDir, csvOutPath)

    // 后续业务逻辑使用absInPath和absOutPath操作文件
    // 示例：打开输入文件
    // file, err := os.Open(absInPath)
    // ...

    return nil
}

调用示例 #

在src/main.go中直接传入相对于当前文件的相对路径即可：

package main

import "your-module-path/utils"

func main() {
    err := utils.MapFields("../data/example.csv", "../data/output.csv")
    if err != nil {
        panic(err)
    }
}

注意事项 #

• runtime.Caller(1)中的参数1表示获取当前函数的直接调用者信息，若存在多层函数包装，需要根据调用栈层级调整该数值。
• 若编译时使用-ldflags="-s -w"参数去除调试符号，runtime.Caller将无法获取到文件路径，需避免此编译参数。
• 测试函数调用时，由于测试框架的栈层级，可能需要调整Caller的参数值（比如改为2）来获取正确的测试文件路径。

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