51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
|
/*
|
|||
|
|
* @Author: haodaquan
|
|||
|
|
* @Date: 2017-06-21 12:55:19
|
|||
|
|
* @Last Modified by: haodaquan
|
|||
|
|
* @Last Modified time: 2017-06-21 13:03:06
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
package jobs
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"PPGo_Job/models"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/astaxie/beego"
|
|||
|
|
"os/exec"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func InitJobs() {
|
|||
|
|
list, _ := models.TaskGetList(1, 1000000, "status", 1)
|
|||
|
|
for _, task := range list {
|
|||
|
|
job, err := NewJobFromTask(task)
|
|||
|
|
if err != nil {
|
|||
|
|
beego.Error("InitJobs:", err.Error())
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
AddJob(task.CronSpec, job)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func runCmdWithTimeout(cmd *exec.Cmd, timeout time.Duration) (error, bool) {
|
|||
|
|
done := make(chan error)
|
|||
|
|
go func() {
|
|||
|
|
done <- cmd.Wait()
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
var err error
|
|||
|
|
select {
|
|||
|
|
case <-time.After(timeout):
|
|||
|
|
beego.Warn(fmt.Sprintf("任务执行时间超过%d秒，进程将被强制杀掉: %d", int(timeout/time.Second), cmd.Process.Pid))
|
|||
|
|
go func() {
|
|||
|
|
<-done // 读出上面的goroutine数据，避免阻塞导致无法退出
|
|||
|
|
}()
|
|||
|
|
if err = cmd.Process.Kill(); err != nil {
|
|||
|
|
beego.Error(fmt.Sprintf("进程无法杀掉: %d, 错误信息: %s", cmd.Process.Pid, err))
|
|||
|
|
}
|
|||
|
|
return err, true
|
|||
|
|
case err = <-done:
|
|||
|
|
return err, false
|
|||
|
|
}
|
|||
|
|
}
|