golang面经必学之os包的用法和场景

os包

Go语言的os包提供了与操作系统交互的功能，包括文件操作、环境变量访问、进程管理等。以下是os包的一些常用功能及其用法和场景用例：

1. 文件操作

os包可以用来创建、读取、写入和删除文件。

用法示例（读取文件）：

package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	file, err := os.Open("example.txt")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		fmt.Println(scanner.Text())
	}
}

用法示例（写入文件）：

package main

import (
	"fmt"
	"os"
)

func main() {
	file, err := os.Create("output.txt")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	_, err = file.WriteString("Hello, World!")
	if err != nil {
		panic(err)
	}
}

2. 环境变量

os包可以读取和设置环境变量。

用法示例：

package main

import (
	"fmt"
	"os"
)

func main() {
	// 获取环境变量
	user := os.Getenv("USER")
	fmt.Println("Current User:", user)

	// 设置环境变量
	err := os.Setenv("MY_VAR", "12345")
	if err != nil {
		panic(err)
	}

	// 获取刚才设置的环境变量
	value := os.Getenv("MY_VAR")
	fmt.Println("MY_VAR:", value)
}

3. 命令行参数

os包可以访问命令行参数。

用法示例：

package main

import (
	"fmt"
	"os"
)

func main() {
	// 访问命令行参数
	args := os.Args
	fmt.Println("Arguments:", args)

	// 访问第一个参数（程序名称）
	fmt.Println("Program Name:", args[0])

	// 访问第二个参数（第一个命令行参数）
	if len(args) > 1 {
		fmt.Println("First Argument:", args[1])
	}
}

4. 执行外部命令

os包可以执行外部命令并获取输出。

用法示例：

package main

import (
	"fmt"
	"os/exec"
)

func main() {
	cmd := exec.Command("ls", "-l")
	output, err := cmd.CombinedOutput()
	if err != nil {
		panic(err)
	}
	fmt.Println(string(output))
}

5. 进程管理

os包可以创建和管理进程。可以使用os包的Exec和os.Process来获取和管理系统进程。

用法示例（创建进程）：

package main

import (
	"fmt"
	"os"
	"os/exec"
)

func main() {
	cmd := exec.Command("/bin/bash", "-c", "echo Hello, World!")
	cmd.Stdin = os.Stdin
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	err := cmd.Run()
	if err != nil {
		fmt.Println("Error:", err)
	}
}

用法示例（获取当前进程的CPU时间）：

package main

import (
	"fmt"
	"os"
	"os/exec"
	"syscall"
)

func main() {
	cmd := exec.Command("ps", "-o", "%cpu", "-o", "%mem", "-p", fmt.Sprint(os.Getpid()))
	output, err := cmd.CombinedOutput()
	if err != nil {
		panic(err)
	}
	fmt.Println(string(output))
}

场景用例

1. 文件处理：读取配置文件、日志文件或任何文本数据。
2. 数据持久化：将程序生成的数据写入文件进行存储。
3. 环境配置：读取环境变量以配置应用程序的行为。
4. 命令行工具：开发命令行工具时，解析命令行参数。
5. 系统集成：与其他系统集成，执行外部命令或调用系统脚本。
6. 进程控制：启动和管理外部进程，监控进程状态。

os包是与操作系统交互的基础，提供了丰富的功能来处理文件、环境变量、命令行参数等，是开发各种应用程序时不可或缺的一部分。