go - 在 Golang 中设置绝对文件路径

Question

我是一个包，它的唯一作用是提供从文件中读取的虚拟数据集合。

看起来像这样:

func GetArrayOfSize(n int) []int {
f, _ := os.Open("./arrays.txt")
defer f.Close()

numbers := make([]int, 0)
scanner := bufio.NewScanner(f)

for scanner.Scan() {
    s, _ := strconv.Atoi(scanner.Text())
    numbers = append(numbers, s)
}

    return numbers[0:n]
}

当我在这个包中测试它时它工作正常但是每当我从另一个包调用 GetArrayOfSize 时我得到一个运行时错误:

panic: runtime error: slice bounds out of range [recovered]
    panic: runtime error: slice bounds out of range

我猜这个错误是由于相对路径:'./arrays.txt。如何获取我的dummy 包的绝对路径来解决这个问题？

非常感谢

Answer 1

您应该检查 os.Open 返回的错误并进行处理。

您可以使用 go/build 获取源代码目录的绝对路径包裹:

p, err := build.Default.Import("github.com/user/repo/dummy", "", build.FindOnly)
if err != nil {
    // handle error
}
fname := filepath.Join(p.Dir, "arrays.txt")

此代码假定设置了 GOPATH 环境变量。