golang 检查进程存在

检查进程存在的方法

在使用Golang开发中，有时我们需要检查某个进程是否存在。本文将介绍一些可以用于检查进程存在的方法。

通过进程ID检查进程存在

最简单的方法就是通过进程ID（PID）来检查进程是否存在。首先，我们需要获取当前运行程序的PID：

package main

import (
	"fmt"
	"os"
)

func main() {
	fmt.Println("当前运行程序的PID: ", os.Getpid())
}

上述代码会输出当前运行程序的PID。

接下来，我们可以使用`os.FindProcess()`函数来生成一个用于表示指定进程ID的`*os.Process`对象，并通过调用该对象的`Signal(os.Signal)`方法来判断进程是否存在：

package main

import (
	"fmt"
	"os"
	"syscall"
)

func IsProcessExist(pid int) bool {
	process, err := os.FindProcess(pid)
	if err != nil {
		return false
	}

	err = process.Signal(syscall.Signal(0))
	if err != nil {
		return false
	}

	return true
}

func main() {
	pid := 12345
	exist := IsProcessExist(pid)
	fmt.Printf("进程 %d 是否存在: %v\n", pid, exist)
}

上述代码中的`Signal(0)`相当于发送一个空信号给指定的进程ID，成功则表示进程存在，失败则表示进程不存在。

通过进程名称检查进程存在

除了通过进程ID检查进程存在外，我们还可以通过进程名称来检查进程是否存在。这个方法比较依赖于平台，因为不同的操作系统有不同的实现方式。

在Linux上检查进程存在

在Linux上，可以通过读取`/proc`目录下的相关信息来检查进程是否存在。

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"strings"
)

func IsProcessExistByName(name string) bool {
	pids, err := ioutil.ReadDir("/proc")
	if err != nil {
		return false
	}

	for _, pid := range pids {
		if pid.IsDir() {
			pidNameFile := fmt.Sprintf("/proc/%s/comm", pid.Name())

			pidNameData, err := ioutil.ReadFile(pidNameFile)
			if err == nil && strings.TrimSpace(string(pidNameData)) == name {
				return true
			}
		}
	}

	return false
}

func main() {
	processName := "myprocess"
	exist := IsProcessExistByName(processName)
	fmt.Printf("进程 %s 是否存在: %v\n", processName, exist)
}

上述代码中，我们遍历`/proc`目录下的所有进程目录，读取每个进程目录下的`comm`文件（该文件记录了进程的名称），然后判断是否与指定的进程名称匹配。

在Windows上检查进程存在

在Windows上，可以通过使用`tasklist`命令来检查进程是否存在。我们可以在Golang中使用`os/exec`包来执行命令。

package main

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

func IsProcessExistByName(name string) bool {
	cmd := exec.Command("tasklist", "/FO", "CSV")

	output, err := cmd.Output()
	if err != nil {
		return false
	}

	processes := strings.Split(string(output), "\n")
	for _, process := range processes[1 : len(processes)-1] {
		processName := strings.Trim(strings.Split(process, "\",\"")[0], "\"")
		if processName == name {
			return true
		}
	}

	return false
}

func main() {
	processName := "myprocess.exe"
	exist := IsProcessExistByName(processName)
	fmt.Printf("进程 %s 是否存在: %v\n", processName, exist)
}

上述代码中，我们通过执行`tasklist`命令获取当前系统上所有进程的列表，并逐行判断每个进程名称是否与指定的进程名称匹配。

小结

通过进程ID和进程名称检查进程存在是常见的方法。根据具体的需求和操作系统，我们可以选择合适的方法来进行进程的检查。