Golang-系统命令操作(os)

落日映苍穹つ 2024-02-21 09:27 20阅读 0赞

### 当前用户信息 ###

可以通过 os/user 包中的 Current() 函数来获取当前用户信息，该函数会返回一个 User 结构体，结构体中的 Username、Uid、HomeDir、Gid 分别表示当前用户的名称、用户 id、用户主目录和用户所属组 id，函数原型如下：

func Current() (*User, error)

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

代码输出结果用户名： LAPTOP-4O4SO9OE\\huanmin  
用户id： S-1-5-21-2557860341-821103020-2882909303-1003  
用户主目录： C:\\Users\\huanmin  
主组id： S-1-5-21-2557860341-821103020-2882909303-1003  
用户所在的所有组： \[S-1-5-21-2557860341-821103020-2882909303 …\]

### 进程参数 ###

func Exit(code int) // 让当前程序以给出的状态码（code）退出。一般来说，状态码0表示成功，非0表示出错。程序会立刻终止，defer的函数不会被执行。
    
    func Getuid() int // 获取调用者的用户id
    
    func Geteuid() int // 获取调用者的有效用户id
    
    func Getgid() int // 获取调用者的组id
    
    func Getegid() int // 获取调用者的有效组id
    
    func Getgroups() ([]int, error) // 获取调用者所在的所有组的组id
    
    func Getpid() int // 获取调用者所在进程的进程id
    
    func Getppid() int // 获取调用者所在进程的父进程的进程id

package main
    
    import (
    	"fmt"
    	"os"
    	"time"
    )
    
    func main() {
        
    	// 获得当前正在运行的进程id
    	fmt.Printf("os.Getpid(): %v\n", os.Getpid())
    	// 父id
    	fmt.Printf("os.Getppid(): %v\n", os.Getppid())
    
    	// 设置新进程的属性
    	attr := &os.ProcAttr{
        
    		// files指定新进程继承的活动文件对象
    		// 前三个分别为，标准输入、标准输出、标准错误输出
    		Files: []*os.File{
        os.Stdin, os.Stdout, os.Stderr},
    		// 新进程的环境变量
    		Env: os.Environ(),
    	}
    
    	// 开始一个新进程
    	p, err := os.StartProcess("c:\\windows\\system32\\notepad.exe", []string{
        "c:\\windows\\system32\\notepad.exe", "d:\\a.txt"}, attr)
    	if err != nil {
        
    		fmt.Println(err)
    	}
    	fmt.Println(p)
    	fmt.Println("进程ID：", p.Pid)
    
    	// 通过进程ID查找进程
    	p2, _ := os.FindProcess(p.Pid)
    	fmt.Println(p2)
    
    	// 等待10秒，执行函数
    	time.AfterFunc(time.Second*10, func() {
        
    		// 向p进程发出退出信号
    		p.Signal(os.Kill)
    	})
    
    	// 等待进程p的退出，返回进程状态
    	ps, _ := p.Wait()
    	fmt.Println(ps.String())
    }

### 环境相关操作 ###

unc Hostname() (name string, err error) // 获取主机名
    
    func Getenv(key string) string // 获取某个环境变量
    
    func Setenv(key, value string) error // 设置一个环境变量,失败返回错误，经测试当前设置的环境变量只在 当前进程有效（当前进程衍生的所以的go程都可以拿到，子go程与父go程的环境变量可以互相获取）；进程退出消失
    
    func Clearenv() // 删除当前程序已有的所有环境变量。不会影响当前电脑系统的环境变量，这些环境变量都是对当前go程序而言的

package main
    
    import (
    	"fmt"
    	"os"
    )
    
    func main() {
        
    	// 获得所有环境变量
    	s := os.Environ()
    	fmt.Printf("s: %v\n", s)
    	// 获得某个环境变量
    	s2 := os.Getenv("GOPATH")
    	fmt.Printf("s2: %v\n", s2)
    	// 设置环境变量
    	os.Setenv("env1", "env1")
    	s2 = os.Getenv("aaa")
    	fmt.Printf("s2: %v\n", s2)
    	fmt.Println("--------------")
    
    	// 查找
    	s3, b := os.LookupEnv("env")
    	fmt.Printf("b: %v\n", b)
    	fmt.Printf("s3: %v\n", s3)
    
    	// 替换
    	os.Setenv("NAME", "gopher")
    	os.Setenv("BURROW", "/usr/gopher")
    
    	os.ExpandEnv("$NAME lives in ${BURROW}.")
    
    	// 清空环境变量
    	// os.Clearenv()
    }

###  ###

### 信号处理 ###

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

Go语言中对信号的处理主要使用 os/signal 包中的两个方法，一个是 Notify 方法用来监听收到的信号，一个是 stop 方法用来取消监听。

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

其中，第一个参数表示接收信号的 channel，第二个及后面的参数表示设置要监听的信号，如果不设置表示监听所有的信号。

【示例 1】使用 Notify 方法来监听收到的信号：

package main
    
    import (
    	"fmt"
    	"os"
    	"os/signal"
    )
    
    func main() {
        
    	c := make(chan os.Signal, 0)
    	signal.Notify(c) 
    	s := <-c //会阻塞,一般放在程序的最后
    	//Got signal: interrupt
    	fmt.Println("Got signal:", s)
    }

【示例 2】使用 stop 方法来取消监听(一般用于正常关闭程序了)：

package main
    
    import (
    	"fmt"
    	"os"
    	"os/signal"
    )
    
    func main() {
        
    	c := make(chan os.Signal, 0)
    	signal.Notify(c)
    	signal.Stop(c) //不允许继续往c中存入内容
    	s := <-c       //c无内容，此处直接退出程序,不会阻塞
    	fmt.Println("Got signal:", s)
    }

### 执行外部命令 ###

exec 包可以执行外部命令，它包装了 os.StartProcess 函数以便更容易的修正输入和输出，使用管道连接 I/O，以及作其它的一些调整。  
![image.png][]

比如`func LookPath(file string) (string, error)`这个函数  
用于在环境变量 PATH 指定的目录中搜索可执行文件，如果 file 中有斜杠，则只在当前目录搜索。返回完整路径或者相对于当前目录的一个相对路径。

package main
    
    import (
    	"fmt"
    	"os/exec"
    )
    
    func main() {
        
    	f, err := exec.LookPath("java")
    	if err != nil {
        
    		fmt.Println(err)
    	}
    	//E:\jdk\jdk-8u151-windows-x64\bin\java.exe
    	fmt.Println(f)
    }

![在这里插入图片描述][241d60d901ab468c8532f2a2d4413093.gif]

点赞 -收藏-关注-便于以后复习和收到最新内容有其他问题在评论区讨论-或者私信我-收到会在第一时间回复在本博客学习的技术不得以任何方式直接或者间接的从事违反中华人民共和国法律,内容仅供学习、交流与参考 免责声明：本文部分素材来源于网络，版权归原创者所有，如存在文章/图片/音视频等使用不当的情况，请随时私信联系我、以迅速采取适当措施，避免给双方造成不必要的经济损失。感谢,配合,希望我的努力对你有帮助^\_^

[image.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/a73ab3a74b834147a107b5e3260025d2.png
[241d60d901ab468c8532f2a2d4413093.gif]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/e1e727a71353404e8ca6703b2ef53738.png