72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package worker
|
||
|
||
import (
|
||
"context"
|
||
"os/exec"
|
||
"syscall"
|
||
|
||
"github.com/RichardKnop/machinery/v1/log"
|
||
"github.com/RichardKnop/machinery/v1/tasks"
|
||
)
|
||
|
||
var asyncTaskMap map[string]interface{}
|
||
|
||
func executeTaskBase(scriptPath string) {
|
||
command := exec.Command("/bin/bash", "-c", scriptPath) //初始化Cmd
|
||
err := command.Start() //运行脚本
|
||
if nil != err {
|
||
log.ERROR.Printf("task exec failed，%v", err.Error())
|
||
return
|
||
}
|
||
|
||
log.INFO.Println("Process PID:", command.Process.Pid)
|
||
|
||
err = command.Wait() //等待执行完成
|
||
if nil != err {
|
||
log.ERROR.Printf("task exec failed，%v", err.Error())
|
||
return
|
||
}
|
||
|
||
log.INFO.Println("ProcessState PID:", command.ProcessState.Pid())
|
||
log.INFO.Println("Exit Code", command.ProcessState.Sys().(syscall.WaitStatus).ExitStatus())
|
||
}
|
||
|
||
// ExecCommand 异步任务
|
||
func ExecCommand(classify string, scriptPath string) error {
|
||
if classify == "shell" {
|
||
log.INFO.Println("start exec shell...", scriptPath)
|
||
executeTaskBase(scriptPath)
|
||
return nil
|
||
} else if classify == "python" {
|
||
log.INFO.Println("start exec python...", scriptPath)
|
||
executeTaskBase(scriptPath)
|
||
return nil
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func SendTask(ctx context.Context, classify string, scriptPath string) {
|
||
args := make([]tasks.Arg, 0)
|
||
args = append(args, tasks.Arg{
|
||
Name: "classify",
|
||
Type: "string",
|
||
Value: classify,
|
||
})
|
||
args = append(args, tasks.Arg{
|
||
Name: "scriptPath",
|
||
Type: "string",
|
||
Value: scriptPath,
|
||
})
|
||
task, _ := tasks.NewSignature("ExecCommandTask", args)
|
||
task.RetryCount = 5
|
||
_, err := AsyncTaskCenter.SendTaskWithContext(ctx, task)
|
||
if err != nil {
|
||
log.ERROR.Println(err.Error())
|
||
}
|
||
}
|
||
|
||
func initAsyncTaskMap() {
|
||
asyncTaskMap = make(map[string]interface{})
|
||
asyncTaskMap["ExecCommandTask"] = ExecCommand
|
||
}
|