当前位置：首页 > news >正文

golang 调用exe 并获取pid

news 2026/3/26 22:07:48

// callSendPublishing 调用SendPublishing程序处理任务（内部方法）
// 返回进程句柄和错误
func callSendPublishing(qName string, headerKey string) (*os.Process, error) {// 先在Redis中创建一个占位符，表示进程即将启动placeholderPID := "starting"// 同步保存占位符到 RedissetProcessIdErr := redis.SetProcessId(redis.GetRedisClient("a"), headerKey, placeholderPID)if setProcessIdErr != nil {return nil, fmt.Errorf("保存进程占位符到Redis失败: %v, 队列: %s", setProcessIdErr, qName)}// 构建 SendPublishing程序路径fileUrlConfig, getFileUrlConfigErr := config.GetFileUrlConfig()if getFileUrlConfigErr != nil {return nil, getFileUrlConfigErr}programPath := fileUrlConfig.BFileName// 修正参数顺序，去掉-NoExit（如果不是PowerShell程序）psScript := fmt.Sprintf(`# 启动程序$psi = New-Object System.Diagnostics.ProcessStartInfo$psi.FileName = "%s"$psi.Arguments = "%s"$psi.UseShellExecute = $true$psi.WindowStyle = 'Normal'$process = [System.Diagnostics.Process]::Start($psi)# 等待窗口句柄可用$timeout = 3000  # 3秒超时$startTime = Get-Datewhile ($process.MainWindowHandle -eq [IntPtr]::Zero -and ((Get-Date) - $startTime).TotalMilliseconds -lt $timeout) {Start-Sleep -Milliseconds 50$process.Refresh()}# 如果获取到窗口句柄，提到前台if ($process.MainWindowHandle -ne [IntPtr]::Zero) {Add-Type @"using System;using System.Runtime.InteropServices;public class WindowHelper {[DllImport("user32.dll")]public static extern bool ShowWindow(IntPtr hWnd, int nCmdShow);[DllImport("user32.dll")]public static extern bool SetForegroundWindow(IntPtr hWnd);[DllImport("user32.dll")]public static extern bool AllowSetForegroundWindow(uint dwProcessId);}
"@
            # 允许设置前台窗口[WindowHelper]::AllowSetForegroundWindow($process.Id)[WindowHelper]::ShowWindow($process.MainWindowHandle, 9)  # SW_RESTORE[WindowHelper]::SetForegroundWindow($process.MainWindowHandle)} else {Write-Warning "未能获取到窗口句柄，但进程已启动 (PID: $($process.Id))"}Write-Output $process.Id`, programPath, qName)cmd := exec.Command("powershell", "-Command", psScript)// 关键配置：使程序在独立控制台运行if runtime.GOOS == "windows" {cmd.SysProcAttr = &syscall.SysProcAttr{CreationFlags: 0x00000010, // 创建新控制台
        }}output, err := cmd.Output()if err != nil {return nil, err}// 解析PIDvar pid uint32lines := strings.Split(string(output), "\n")for _, line := range lines {line = strings.TrimSpace(line)if pids, err := strconv.Atoi(line); err == nil {pid = uint32(pids)}}// 获取进程句柄// 立即将真实 PID保存到RedisprocessID := fmt.Sprintf("%d", pid)setProcessIdErr = redis.SetProcessId(redis.GetRedisClient("a"), headerKey, processID)if setProcessIdErr != nil {return nil, fmt.Errorf("更新进程PID到Redis失败: %v, 队列: %s", setProcessIdErr, qName)}// 启动 goroutine等待进程结束并清理 Redisgo func(pid int, qName string, headerKey string) {// 定期检查进程是否还在运行for {time.Sleep(2 * time.Second) // 每2秒检查一次if !isProcessExistWindows(pid) {// 进程已结束，清理Redisif headerKey != "" {deleteProcessIdErr := redis.DeleteProcessId(redis.GetRedisClient("a"), headerKey)if deleteProcessIdErr != nil {logStr := fmt.Sprintf("清理Redis进程记录失败，PID: %d, 队列: %s, 错误: %v", pid, qName, deleteProcessIdErr)logs.LoggingMiddleware(logs.LOG_LEVEL_ERROR, logStr)}break}}}}(int(pid), qName, headerKey)//将生成的process和redisKey 存储到文件 executingfilereturn &os.Process{Pid: int(pid)}, nil
}