Golang仿ps获取Linux进程信息

原理

遍历读取/proc/获取所有进程ID
cat /proc/5181/stat中前四列分别为进程PID，进程名，进程状态，父进程PID

Go代码

1. 获取/proc/下面所有文件名+文件夹名为数字的名字
2. 读取/proc/xxx/stat获取进程信息输出

package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"regexp"
	"sort"
	"strconv"
)

func main() {
	var process []int
	var validId = regexp.MustCompile("^[0-9]+$")

	infoList, err := ioutil.ReadDir("/proc")
	if err != nil {
		log.Println(infoList)
	}

	for _, info := range infoList {
		if info.IsDir() && validId.MatchString(info.Name()) {
			p, _ := strconv.Atoi(info.Name())
			process = append(process, p)
		}
	}

	sort.Ints(process)

	statRe := regexp.MustCompile(`([0-9]+) \((.+?)\) [a-zA-Z]+ ([0-9]+)`)
	fmt.Printf("%6s\t%6s\t%s\n", "PID", "PPID", "NAME")
	for _, p := range process {
		b, err := ioutil.ReadFile(fmt.Sprintf("/proc/%d/stat", p))
		if err != nil {
			continue
		}
		matches := statRe.FindStringSubmatch(string(b))
		fmt.Printf("%6s\t%6s\t%s\n", matches[1], matches[3], matches[2])
	}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

测试验证

运行程序查看