Go语言os标准库
Go语言的os标准库提供了丰富的功能来与操作系统交互，涵盖文件操作、目录管理、环境变量、进程控制等。以下是一些常用函数及示例：
一、文件操作
1. 打开/关闭文件
os.Open()：打开文件（只读）。os.Create()：创建文件（若存在则清空）。
// 打开文件
file, err := os.Open("test.txt")
if err != nil {
log.Fatal(err)
}
defer file.Close()
// 创建文件
newFile, err := os.Create("newfile.txt")
if err != nil {
log.Fatal(err)
}
defer newFile.Close()
2. 读写文件
os.ReadFile()：读取整个文件。os.WriteFile()：写入整个文件（覆盖）。
// 读取文件
data, err := os.ReadFile("test.txt")
if err != nil {
log.Fatal(err)
}
fmt.Println(string(data))
// 写入文件
content := []byte("Hello, Go!")
err = os.WriteFile("output.txt", content, 0644) // 权限 0644
if err != nil {
log.Fatal(err)
}
3. 文件信息
os.Stat()：获取文件信息（大小、权限等）。
info, err := os.Stat("test.txt")
if err != nil {
log.Fatal(err)
}
fmt.Println("文件名:", info.Name())
fmt.Println("大小:", info.Size(), "bytes")
fmt.Println("修改时间:", info.ModTime())
4. 文件操作扩展
os.OpenFile()：更灵活地打开文件（支持读写、追加等模式）。
// 追加写入文件
file, err := os.OpenFile("log.txt", os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
if err != nil {
log.Fatal(err)
}
defer file.Close()
_, err = file.WriteString("New log entry\n")
if err != nil {
log.Fatal(err)
}
二、目录操作
1. 创建目录
os.Mkdir()：创建单级目录。os.MkdirAll()：创建多级目录。
err := os.Mkdir("mydir", 0755) // 权限 0755
if err != nil {
log.Fatal(err)
}
err = os.MkdirAll("parent/child/grandchild", 0755) // 创建嵌套目录
if err != nil {
log.Fatal(err)
}
2. 删除文件/目录
os.Remove()：删除文件或空目录。os.RemoveAll()：递归删除目录及其内容。
err := os.Remove("oldfile.txt")
if err != nil {
log.Fatal(err)
}
err = os.RemoveAll("obsolete_dir") // 强制删除非空目录
if err != nil {
log.Fatal(err)
}
3. 遍历目录
os.ReadDir()：读取目录内容。
entries, err := os.ReadDir(".")
if err != nil {
log.Fatal(err)
}
for _, entry := range entries {
fmt.Println(entry.Name())
}
三、环境变量
1. 获取/设置环境变量
os.Getenv()：获取环境变量。os.Setenv()：设置环境变量（当前进程有效）。
path := os.Getenv("PATH")
fmt.Println("PATH:", path)
err := os.Setenv("TEMP_DIR", "/tmp")
if err != nil {
log.Fatal(err)
}
四、进程与参数
1. 命令行参数
os.Args：获取命令行参数。
// 运行示例：go run main.go arg1 arg2
fmt.Println("程序名:", os.Args[0])
fmt.Println("参数:", os.Args[1:])
2. 进程退出
os.Exit()：终止进程并返回状态码。
fmt.Println("程序即将退出")
os.Exit(1) // 非0状态码表示错误
五、其他实用函数
1. 工作目录
os.Getwd()：获取当前目录。os.Chdir()：修改当前目录。
dir, err := os.Getwd()
if err != nil {
log.Fatal(err)
}
fmt.Println("当前目录:", dir)
err = os.Chdir("/tmp") // 切换到 /tmp
if err != nil {
log.Fatal(err)
}
2. 标准输入/输出
os.Stdin,os.Stdout,os.Stderr：标准流。
// 从标准输入读取并输出到标准输出
io.Copy(os.Stdout, os.Stdin)
六、文件权限
os.Chmod()：修改文件权限。
err := os.Chmod("secret.txt", 0600) // 仅所有者可读写
if err != nil {
log.Fatal(err)
}