快捷導(dǎo)航

Go語言os包用法詳解

更新時(shí)間：2023年04月03日 14:42:24 作者：C語言中文網(wǎng)

本文主要介紹了Go語言os包用法詳解，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

Go語言的 os 包中提供了操作系統(tǒng)函數(shù)的接口，是一個(gè)比較重要的包。顧名思義，os 包的作用主要是在服務(wù)器上進(jìn)行系統(tǒng)的基本操作，如文件操作、目錄操作、執(zhí)行命令、信號(hào)與中斷、進(jìn)程、系統(tǒng)狀態(tài)等等。

os 包中的常用函數(shù)

1) Hostname

函數(shù)定義:

func Hostname() (name string, err error)

Hostname 函數(shù)會(huì)返回內(nèi)核提供的主機(jī)名。

2) Environ

函數(shù)定義:

func Environ() []string

Environ 函數(shù)會(huì)返回所有的環(huán)境變量，返回值格式為“key=value”的字符串的切片拷貝。

3) Getenv

函數(shù)定義:

func Getenv(key string) string

Getenv 函數(shù)會(huì)檢索并返回名為 key 的環(huán)境變量的值。如果不存在該環(huán)境變量則會(huì)返回空字符串。

4) Setenv

函數(shù)定義:

func Setenv(key, value string) error

Setenv 函數(shù)可以設(shè)置名為 key 的環(huán)境變量，如果出錯(cuò)會(huì)返回該錯(cuò)誤。

5) Exit

函數(shù)定義:

func Exit(code int)

Exit 函數(shù)可以讓當(dāng)前程序以給出的狀態(tài)碼 code 退出。一般來說，狀態(tài)碼 0 表示成功，非 0 表示出錯(cuò)。程序會(huì)立刻終止，并且 defer 的函數(shù)不會(huì)被執(zhí)行。

6) Getuid

函數(shù)定義:

func Getuid() int

Getuid 函數(shù)可以返回調(diào)用者的用戶 ID。

7) Getgid

函數(shù)定義:

func Getgid() int

Getgid 函數(shù)可以返回調(diào)用者的組 ID。

8) Getpid

函數(shù)定義:

func Getpid() int

Getpid 函數(shù)可以返回調(diào)用者所在進(jìn)程的進(jìn)程 ID。

9) Getwd

函數(shù)定義:

func Getwd() (dir string, err error)

Getwd 函數(shù)可以返回一個(gè)對(duì)應(yīng)當(dāng)前工作目錄的根路徑。如果當(dāng)前目錄可以經(jīng)過多條路徑抵達(dá)（因?yàn)橛叉溄樱?，Getwd 會(huì)返回其中一個(gè)。

10) Mkdir

函數(shù)定義:

func Mkdir(name string, perm FileMode) error

Mkdir 函數(shù)可以使用指定的權(quán)限和名稱創(chuàng)建一個(gè)目錄。如果出錯(cuò)，會(huì)返回 *PathError 底層類型的錯(cuò)誤。

11) MkdirAll

函數(shù)定義:

func MkdirAll(path string, perm FileMode) error

MkdirAll 函數(shù)可以使用指定的權(quán)限和名稱創(chuàng)建一個(gè)目錄，包括任何必要的上級(jí)目錄，并返回 nil，否則返回錯(cuò)誤。權(quán)限位 perm 會(huì)應(yīng)用在每一個(gè)被該函數(shù)創(chuàng)建的目錄上。如果 path 指定了一個(gè)已經(jīng)存在的目錄，MkdirAll 不做任何操作并返回 nil。

12) Remove

函數(shù)定義:

func Remove(name string) error

Remove 函數(shù)會(huì)刪除 name 指定的文件或目錄。如果出錯(cuò)，會(huì)返回 *PathError 底層類型的錯(cuò)誤。

RemoveAll 函數(shù)跟 Remove 用法一樣，區(qū)別是會(huì)遞歸的刪除所有子目錄和文件。

在 os 包下，有 exec，signal，user 三個(gè)子包，下面來分別介紹一下。

os/exec 執(zhí)行外部命令

exec 包可以執(zhí)行外部命令，它包裝了 os.StartProcess 函數(shù)以便更容易的修正輸入和輸出，使用管道連接 I/O，以及作其它的一些調(diào)整。

func LookPath(file string) (string, error)

在環(huán)境變量 PATH 指定的目錄中搜索可執(zhí)行文件，如果 file 中有斜杠，則只在當(dāng)前目錄搜索。返回完整路徑或者相對(duì)于當(dāng)前目錄的一個(gè)相對(duì)路徑。

示例代碼如下：

package main
import (
    "fmt"
    "os/exec"
)
func main() {
    f, err := exec.LookPath("main")
    if err != nil {
        fmt.Println(err)
    }
    fmt.Println(f)
}

運(yùn)行結(jié)果如下：

main.exe

os/user 獲取當(dāng)前用戶信息

可以通過 os/user 包中的 Current() 函數(shù)來獲取當(dāng)前用戶信息，該函數(shù)會(huì)返回一個(gè) User 結(jié)構(gòu)體，結(jié)構(gòu)體中的 Username、Uid、HomeDir、Gid 分別表示當(dāng)前用戶的名稱、用戶 id、用戶主目錄和用戶所屬組 id，函數(shù)原型如下：

func Current() (*User, error)

示例代碼如下：

package main
import (
    "log"
    "os/user"
)
func main() {
    u, _ := user.Current()
    log.Println("用戶名：", u.Username)
    log.Println("用戶id", u.Uid)
    log.Println("用戶主目錄：", u.HomeDir)
    log.Println("主組id：", u.Gid)
    // 用戶所在的所有的組的id
    s, _ := u.GroupIds()
    log.Println("用戶所在的所有組：", s)
}

運(yùn)行結(jié)果如下：

2019/12/13 15:12:14 用戶名： LENOVO-PC\Administrator
2019/12/13 15:12:14 用戶id S-1-5-21-711400000-2334436127-1750000211-000
2019/12/13 15:12:14 用戶主目錄： C:\Users\Administrator
2019/12/13 15:12:14 主組id： S-1-5-22-766000000-2300000100-1050000262-000
2019/12/13 15:12:14 用戶所在的所有組： [S-1-5-32-544 S-1-5-22-000 S-1-5-21-777400999-2344436111-1750000262-003]

os/signal 信號(hào)處理

一個(gè)運(yùn)行良好的程序在退出（正常退出或者強(qiáng)制退出，如 Ctrl+C，kill 等）時(shí)是可以執(zhí)行一段清理代碼的，將收尾工作做完后再真正退出。一般采用系統(tǒng) Signal 來通知系統(tǒng)退出，如 kill pid，在程序中針對(duì)一些系統(tǒng)信號(hào)設(shè)置了處理函數(shù)，當(dāng)收到信號(hào)后，會(huì)執(zhí)行相關(guān)清理程序或通知各個(gè)子進(jìn)程做自清理。

Go語言中對(duì)信號(hào)的處理主要使用 os/signal 包中的兩個(gè)方法，一個(gè)是 Notify 方法用來監(jiān)聽收到的信號(hào)，一個(gè)是 stop 方法用來取消監(jiān)聽。

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

其中，第一個(gè)參數(shù)表示接收信號(hào)的 channel，第二個(gè)及后面的參數(shù)表示設(shè)置要監(jiān)聽的信號(hào)，如果不設(shè)置表示監(jiān)聽所有的信號(hào)。

【示例 1】使用 Notify 方法來監(jiān)聽收到的信號(hào)：

package main
import (
    "fmt"
    "os"
    "os/signal"
)
func main() {
    c := make(chan os.Signal, 0)
    signal.Notify(c)
    // Block until a signal is received.
    s := <-c
    fmt.Println("Got signal:", s)
}

運(yùn)行該程序，然后在 CMD 窗口中通過 Ctrl+C 來結(jié)束該程序，便會(huì)得到輸出結(jié)果：

Got signal: interrupt

【示例 2】使用 stop 方法來取消監(jiān)聽：

package main
import (
    "fmt"
    "os"
    "os/signal"
)
func main() {
    c := make(chan os.Signal, 0)
    signal.Notify(c)
    signal.Stop(c) //不允許繼續(xù)往c中存入內(nèi)容
    s := <-c       //c無內(nèi)容，此處阻塞，所以不會(huì)執(zhí)行下面的語句，也就沒有輸出
    fmt.Println("Got signal:", s)
}

因?yàn)槭褂?Stop 方法取消了 Notify 方法的監(jiān)聽，所以運(yùn)行程序沒有輸出結(jié)果。

到此這篇關(guān)于Go語言os包用法詳解的文章就介紹到這了,更多相關(guān)Go語言os包內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: