mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
7b6805a834
- Added support for new dependency file types: 'go.mod' and 'pom.xml' in dependency.go. - Refactored command configuration in runner.go to improve logging and error handling. - Introduced a new method to configure Node.js paths, enhancing environment setup for tasks. - Enhanced IPC message handling with detailed logging for better traceability. - Updated service logging to remove unnecessary prefixes for cleaner output. - Improved command execution handling in process.go for better compatibility across platforms.
105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"os/exec"
|
|
"runtime"
|
|
|
|
"github.com/shirou/gopsutil/process"
|
|
)
|
|
|
|
func BuildCmd(cmdStr string) (cmd *exec.Cmd, err error) {
|
|
if cmdStr == "" {
|
|
return nil, errors.New("command string is empty")
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
return exec.Command("cmd", "/C", cmdStr), nil
|
|
}
|
|
return exec.Command("sh", "-c", cmdStr), nil
|
|
}
|
|
|
|
func ProcessIdExists(pid int) (exists bool) {
|
|
if runtime.GOOS == "windows" {
|
|
return processIdExistsWindows(pid)
|
|
} else {
|
|
return processIdExistsLinuxMac(pid)
|
|
}
|
|
}
|
|
|
|
func processIdExistsWindows(pid int) (exists bool) {
|
|
exists, err := process.PidExists(int32(pid))
|
|
if err != nil {
|
|
logger.Errorf("error checking if process exists: %v", err)
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func processIdExistsLinuxMac(pid int) (exists bool) {
|
|
exists, err := process.PidExists(int32(pid))
|
|
if err != nil {
|
|
logger.Errorf("error checking if process exists: %v", err)
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func GetProcesses() (processes []*process.Process, err error) {
|
|
processes, err = process.Processes()
|
|
if err != nil {
|
|
logger.Errorf("error getting processes: %v", err)
|
|
return nil, err
|
|
}
|
|
return processes, nil
|
|
}
|
|
|
|
type KillProcessOptions struct {
|
|
Force bool
|
|
}
|
|
|
|
func KillProcess(cmd *exec.Cmd, force bool) error {
|
|
// process
|
|
p, err := process.NewProcess(int32(cmd.Process.Pid))
|
|
if err != nil {
|
|
logger.Errorf("failed to get process: %v", err)
|
|
return err
|
|
}
|
|
|
|
// kill process
|
|
return killProcessRecursive(p, force)
|
|
}
|
|
|
|
func killProcessRecursive(p *process.Process, force bool) (err error) {
|
|
// children processes
|
|
cps, err := p.Children()
|
|
if err != nil {
|
|
if !errors.Is(err, process.ErrorNoChildren) {
|
|
logger.Errorf("failed to get children processes: %v", err)
|
|
} else if errors.Is(err, process.ErrorProcessNotRunning) {
|
|
return nil
|
|
}
|
|
return killProcess(p, force)
|
|
}
|
|
|
|
// iterate children processes
|
|
for _, cp := range cps {
|
|
if err := killProcessRecursive(cp, force); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return killProcess(p, force)
|
|
}
|
|
|
|
func killProcess(p *process.Process, force bool) (err error) {
|
|
if force {
|
|
err = p.Kill()
|
|
} else {
|
|
err = p.Terminate()
|
|
}
|
|
if err != nil {
|
|
logger.Errorf("failed to kill process (force: %v): %v", force, err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|