Golang 如何获取当前运行文件的目录

前言

Golang 的开发中，有时候我们需要获取当前正在运行的程序所在的目录路径，这对于读取相对路径下的配置文件或资源文件非常有用。

本文将展示几种在 Golang 中获取当前运行文件目录的方法。

方法一：使用 os 包

Go 语言的 os 包提供了很多与操作系统交互的功能，其中就包括了获取当前执行的程序的绝对路径。

go
package main

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

func main() {
    // 获取当前执行的程序的路径
    exePath, err := os.Executable()
    if err != nil {
        panic(err)
    }
  
    // 将路径转换为绝对路径
    exeAbsPath, err := filepath.Abs(exePath)
    if err != nil {
        panic(err)
    }

    // 获取执行文件所在的目录
    dir := filepath.Dir(exeAbsPath)
  
    fmt.Println("当前运行文件的目录是：", dir)
}

请注意，这里获取的是可执行文件的路径，而不是调用 os.Executable() 的源代码文件路径。

方法二：使用 runtime 包

如果你想获取的是源代码文件所在的目录，可以使用 runtime 包中的 Caller 函数。

go
package main

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

func main() {
    // 获取当前函数调用的程序计数器、文件信息和行号
    _, filename, _, ok := runtime.Caller(0)
    if !ok {
        panic("No caller information")
    }
  
    // 将文件的路径转换为绝对路径
    absPath, err := filepath.Abs(filename)
    if err != nil {
        panic(err)
    }
  
    // 获取执行文件所在的目录
    dir := filepath.Dir(absPath)
  
    fmt.Println("当前源代码文件的目录是：", dir)
}

这段代码会输出当前源代码文件所在的目录路径。runtime.Caller(0) 返回的是当前调用栈的信息，其中 0 表示当前函数，filename 就是当前源文件的路径。

方法三：使用 os 包结合环境变量

有时候，我们可以结合环境变量来确定当前的工作目录，这通常用于获取程序运行时的工作目录，这样可以找到相对于工作目录的文件和资源。

go
package main

import (
    "fmt"
    "os"
)

func main() {
    // 获取程序的工作目录
    dir, err := os.Getwd()
    if err != nil {
        panic(err)
    }
  
    fmt.Println("程序的工作目录是：", dir)
}

这里 os.Getwd() 返回的是程序的当前工作目录，通常是你在哪个目录下执行 go run 或者二进制文件的路径。

总结

在 Golang 中，主要有三种方法来获取当前运行文件或代码的目录。使用 os.Executable 获取的是可执行文件的目录，使用 runtime.Caller 获取的是源代码文件所在的目录，而使用 os.Getwd 则是获得程序执行时的工作目录。根据你的需求选择适合的方法，确保你的程序能够正确地找到它需要的资源。