GOLang判断进程是否存在实现方式

GOLang判断进程是否存在

最近在琢磨用Go写一个进程检测工具，发现事情没那么简单。Go的标准库里，并没有提供一个现成的、直接获取系统进程列表的“银弹”方法。这可能是出于跨平台复杂性的考虑。怎么办呢？经过一番搜索和尝试，发现一个绕不开的路径：调用系统命令。没错，就是通过执行外部命令来获取我们需要的信息，这虽然绕了点路，但在Windows环境下却相当有效。

具体思路是这样的：我们可以借助Windows自带的WMIC（Windows Management Instrumentation Command-line）工具来获取所有进程的详细信息，然后再用查找命令来筛选我们关心的目标。下图清晰地展示了这个实现方式的逻辑流程：

关键代码

基于上述思路，一个最直接的实现版本如下。这个例子以检测“chrome.exe”是否运行为目标：

    buf := bytes.Buffer{}
    cmd := exec.Command("wmic", "process", "get", "name,executablepath")
    cmd.Stdout = &buf
    cmd.Run()
    cmd2 := exec.Command("findstr", "chrome.exe")
    cmd2.Stdin = &buf
    data,  := cmd2.CombinedOutput()
    if len(data) == 0 {
        fmt.Println("not runing")
    } else {
        fmt.Println("runing")
    }

这段代码逻辑清晰：先通过wmic命令抓取进程列表输出到一个缓冲区，再让findstr命令从这个缓冲区里搜索特定关键词。如果搜索返回结果为空，就认为进程不存在；反之则存在。

不过，仔细推敲一下，这个基础版本有几个可以优化的地方。比如，错误处理不够完善；再比如，搜索关键词“chrome.exe”可能匹配到其他包含该字符串的进程路径，导致误判。这就需要我们进行更精细化的处理。

优化后的代码

为了让工具更健壮、更通用，我们对代码进行了重构。核心思想是将功能封装成一个函数，并引入双重验证机制：

// IsExeRuning : 程序是否运行，strKey:用于结果查找， strExeName:校验查找的是否是要寻找的进程
func IsExeRuning(strKey string, strExeName string) bool {
    buf := bytes.Buffer{}
    cmd := exec.Command("wmic", "process", "get", "name,executablepath")
    cmd.Stdout = &buf
    cmd.Run()
    cmd2 := exec.Command("findstr", strKey)
    cmd2.Stdin = &buf
    data, err := cmd2.CombinedOutput()
    if err != nil && err.Error() != "exit status 1" {
        //XBLog.LogF("ServerMonitor", "IsExeRuning CombinedOutput error, err:%s", err.Error())
        return false
    }
    strData := string(data)
    if strings.Contains(strData, strExeName) {
        return true
    } else {
        return false
    }
}

这个优化版本做了两处关键改进：

首先，增强了错误处理。findstr命令在找不到匹配项时会返回退出状态码1，这属于正常情况，不应被视为错误。因此，代码特别排除了这种状态，只有当发生其他类型的错误时才返回失败。

其次，也是更重要的一点，引入了进程名精确校验。函数接受两个参数：strKey用于初步的快速查找（比如“chrome”），strExeName用于精确匹配（必须是“chrome.exe”）。这样设计的好处是，即使strKey匹配到了类似“chromedriver.exe”的进程，在最终的字符串包含检查中也会因为名称不精确而被过滤掉，从而大大提高了判断的准确性。

总结

总的来说，在Go语言中判断进程是否存在，尤其是在Windows平台下，通过组合系统命令是一种行之有效的方案。从基础命令调用到封装优化，整个过程的关键在于处理好命令执行的细节和结果的精确匹配。上面提供的思路和代码，希望能为有类似需求的开发者提供一个可靠的参考起点。

您可能感兴趣的文章:

• golang 设置进程退出时kill所有子进程的三种方法
• Golang仿ps实现获取Linux进程信息
• 源码剖析Golang如何fork一个进程
• Golang守护进程用法示例分析
• golang进程内存控制避免docker内oom
• golang进程在docker中OOM后hang住问题解析
• golang 后台进程的启动和停止操作