Go 语言中通过进程名检查进程运行状态的实用指南

package main

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

// IsProcessRunningByNamePgrep 检查指定名称的进程是否正在运行，使用 pgrep 命令
func IsProcessRunningByNamePgrep(processName string) (bool, error) {
    // 构建 pgrep 命令，-x 选项表示精确匹配进程名
    cmd := exec.Command("pgrep", "-x", processName)
    var stdout, stderr bytes.Buffer
    cmd.Stdout = &stdout
    cmd.Stderr = &stderr

    err := cmd.Run()
    if err != nil {
        // pgrep 在没有找到匹配进程时会返回非零退出码（通常是1）
        // 但如果 stderr 有内容，说明可能是其他执行错误
        if stderr.Len() > 0 {
            return false, fmt.Errorf("执行 pgrep 命令失败: %s, 错误信息: %s", err, stderr.String())
        }
        // 如果没有错误输出，仅是 pgrep 未找到进程的退出码，则认为进程未运行
        return false, nil
    }

    // 如果 pgrep 成功执行且有输出，则说明找到了进程
    // 输出通常是 PID 列表，每行一个
    output := strings.TrimSpace(stdout.String())
    return len(output) > 0, nil
}

func main() {
    // 示例：检查 "bash" 进程
    isRunning, err := IsProcessRunningByNamePgrep("bash")
    if err != nil {
        fmt.Printf("检查 bash 进程出错: %v\n", err)
    } else if isRunning {
        fmt.Println("bash 进程正在运行。")
    } else {
        fmt.Println("bash 进程未运行。")
    }

    // 示例：检查一个不存在的进程
    isRunning, err = IsProcessRunningByNamePgrep("nonexistent_process_12345")
    if err != nil {
        fmt.Printf("检查 nonexistent_process_12345 进程出错: %v\n", err)
    } else if isRunning {
        fmt.Println("nonexistent_process_12345 进程正在运行。")
    } else {
        fmt.Println("nonexistent_process_12345 进程未运行。")
    }
}

package main

import (
    "fmt"
    "io/ioutil"
    "os"
    "path/filepath"
    "strconv"
    "strings"
)

// IsProcessRunningByNameProcfs 检查指定名称的进程是否正在运行，通过读取 procfs
// 此方法仅适用于 Linux 系统
func IsProcessRunningByNameProcfs(processName string) (bool, error) {
    // 检查当前操作系统是否为 Linux
    if os.Getenv("GOOS") != "linux" && os.Getenv("GOOS") == "" { // os.Getenv("GOOS")为空表示未交叉编译，直接运行
        // 运行时检查 os.GOOS
        if runtime.GOOS != "linux" {
            return false, fmt.Errorf("此方法仅支持 Linux 系统，当前系统为: %s", runtime.GOOS)
        }
    } else if os.Getenv("GOOS") != "linux" { // 交叉编译时检查 GOOS 环境变量
        return false, fmt.Errorf("此方法仅支持 Linux 系统，交叉编译目标系统为: %s", os.Getenv("GOOS"))
    }


    entries, err := ioutil.ReadDir("/proc")
    if err != nil {
        return false, fmt.Errorf("无法读取 /proc 目录: %w", err)
    }

    for _, entry := range entries {
        // 检查目录名是否为数字，表示一个进程 PID
        if !entry.IsDir() {
            continue
        }
        pidStr := entry.Name()
        if _, err := strconv.Atoi(pidStr); err != nil {
            continue // 不是数字，跳过
        }

        // 读取 /proc/<pid>/comm 文件
        commPath := filepath.Join("/proc", pidStr, "comm")
        commContent, err := ioutil.ReadFile(commPath)
        if err != nil {
            // 进程可能已经退出，或者没有读取权限
            continue
        }
        // comm 文件内容末尾通常有换行符
        actualProcessName := strings.TrimSpace(string(commContent))

        if actualProcessName == processName {
            return true, nil
        }
    }

    return false, nil
}

func main() {
    // 示例：检查 "systemd" 进程
    isRunning, err := IsProcessRunningByNameProcfs("systemd")
    if err != nil {
        fmt.Printf("检查 systemd 进程出错: %v\n", err)
    } else if isRunning {
        fmt.Println("systemd 进程正在运行。")
    } else {
        fmt.Println("systemd 进程未运行。")
    }

    // 示例：检查一个不存在的进程
    isRunning, err = IsProcessRunningByNameProcfs("nonexistent_process_abc")
    if err != nil {
        fmt.Printf("检查 nonexistent_process_abc 进程出错: %v\n", err)
    } else if isRunning {
        fmt.Println("nonexistent_process_abc 进程未运行。")
    } else {
        fmt.Println("nonexistent_process_abc 进程未运行。")
    }
}