正文

Go语言os包用法简述-Go语言教程

admin
admin V管理员 /0 评论 /248 阅读
0110
此篇文章发布距今已超过683天，您需要注意文章的内容或图片是否可用！

Go语言os包用法简述

os 标准包，是一个比较重要的包，顾名思义，主要是在服务器上进行系统的基本操作，如文件操作，目录操作，执行命令，信号与中断，进程，系统状态等等。在 os 包下，有 exec，signal，user 三个子包。

启动外部命令和程序

在 os 包中，有很多有趣的功能，比如可以通过变量 Args 来获取命令参数，os.Args 返回一个字符串数组。

fmt.Println(os.Args)

在 os 包中，相关函数名字和作用有较重的 Unix 风，比如：

func Chdir(dir string) error //chdir 将当前工作目录更改为 dir 目录

func Getwd() (dir string, err error) //获取当前目录

func Chmod(name string, mode FileMode) error //更改文件的权限

func Chown(name string, uid, gid int) error //更改文件拥有者 owner

func Chtimes(name string, atime time.Time, mtime time.Time) error func Clearenv() //清除所有环境变量（慎用）

func Environ() []string //返回所有环境变量

func Exit(code int) //系统退出，并返回 code，其中 0 表示执行成功并退出，非 0 表示错误并退出

在 os 包中，有关文件的处理也有很多方法，如：

func Create(name string) (file File, err error) // Create 采用模式 0666 创建一个名为 name 的文件，如果文件已存在会截断它（为空文件）

func Open(name string) (file File, err error) // Open 打开一个文件用于读取。

func (f File) Stat() (fi FileInfo, err error) // Stat 返回描述文件 f 的 FileInfo 类型值

func (f File) Readdir(n int) (fi []FileInfo, err error) // Readdir 读取目录 f 的内容，返回一个有 n 个成员的 []FileInfo

func (f File) Read(b []byte) (n int, err error) // Read 方法从 f 中读取最多 len(b) 字节数据并写入 b

func (f File) WriteString(s string) (ret int, err error) // 向文件中写入字符串

func (f File) Sync() (err error) // Sync 递交文件的当前内容进行稳定的存储。

func (f File) Close() error // Close 关闭文件 f

有关文件的处理，将在《Go语言文件处理》一章中详细说明。


在 os 包中有一个 StartProcess 函数可以调用或启动外部系统命令和二进制可执行文件；它的第一个参数是要运行的进程，第二个参数用来传递选项或参数，第三个参数是含有系统环境基本信息的结构体。


这个函数返回被启动进程的 id(pid)，或者启动失败返回错误。

package main
import (
    "fmt"
    "os"
)
func main() {
    // 1) os.StartProcess //
    /*********************/
    /* Linux: */
    env := os.Environ()
    procAttr := &os.ProcAttr{
        Env: env,
        Files: []*os.File{
            os.Stdin,
            os.Stdout,
            os.Stderr,
        },
    }
    // 1st example: list files
    Pid, err := os.StartProcess("/bin/ls", []string{"ls", "-l"}, procAttr)
    if err != nil {
        fmt.Printf("Error %v starting process!", err) //
        os.Exit(1)
    }
    fmt.Printf("The process id is %v", pid)
}

os/signal 信号处理

一个运行良好的程序在退出（正常退出或者强制退出，如 ctrl+c，kill 等）时是可以执行一段清理代码，将收尾工作做完后再真正退出。一般采用系统 Signal 来通知系统退出，如 kill pid。在程序中针对一些系统信号设置了处理函数，当收到信号后，会执行相关清理程序或通知各个子进程做自清理。


Go语言的系统信号处理主要涉及 os 包、os.signal 包以及 syscall 包。其中最主要的函数是 signal 包中的 Notify 函数：

func Notify(c chan<- os.Signal, sig …os.Signal)

该函数会将进程收到的系统 Signal 转发给 channel c。如果没有传入 sig 参数，那么 Notify 会将系统收到的所有信号转发给 channel c。


Notify 会根据传入的 os.Signal，监听对应 Signal 信号，Notify() 方法会将接收到对应 os.Signal 往一个 channel c 中发送。


下面代码以 syscall.SIGUSR2 信息为例，说明了具体实现：

package main
import (
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)
func main() {
    go signalListen()
    for {
        time.Sleep(10 * time.Second)
    }
}
func signalListen() {
    c := make(chan os.Signal)
    signal.Notify(c, syscall.SIGUSR2)
    for {
        s := <-c
        //收到信号后的处理，这里只是输出信号内容，可以做一些更有意思的事
        fmt.Println("get signal:", s)
    }
}

宙飒天下