94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/service"
|
|||
|
|
"github.com/robfig/cron/v3"
|
|||
|
|
|
|||
|
|
"go-admin/app/jobs"
|
|||
|
|
"go-admin/app/jobs/models"
|
|||
|
|
"go-admin/common/dto"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type SysJob struct {
|
|||
|
|
service.Service
|
|||
|
|
Cron *cron.Cron
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RemoveJob 删除job
|
|||
|
|
func (e *SysJob) RemoveJob(c *dto.GeneralDelDto) error {
|
|||
|
|
var err error
|
|||
|
|
var data models.SysJob
|
|||
|
|
err = e.Orm.Table(data.TableName()).First(&data, c.Id).Error
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("db error: %s", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
cn := jobs.Remove(e.Cron, data.EntryId)
|
|||
|
|
|
|||
|
|
select {
|
|||
|
|
case res := <-cn:
|
|||
|
|
if res {
|
|||
|
|
err = e.Orm.Table(data.TableName()).Where("entry_id = ?", data.EntryId).Update("entry_id", 0).Error
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("db error: %s", err)
|
|||
|
|
}
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
case <-time.After(time.Second * 1):
|
|||
|
|
e.Msg = "操作超时！"
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StartJob 启动任务
|
|||
|
|
func (e *SysJob) StartJob(c *dto.GeneralGetDto) error {
|
|||
|
|
var data models.SysJob
|
|||
|
|
var err error
|
|||
|
|
err = e.Orm.Table(data.TableName()).First(&data, c.Id).Error
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("db error: %s", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if data.Status == 1 {
|
|||
|
|
err = errors.New("当前Job是关闭状态不能被启动，请先启用。")
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if data.JobType == 1 {
|
|||
|
|
var j = &jobs.HttpJob{}
|
|||
|
|
j.InvokeTarget = data.InvokeTarget
|
|||
|
|
j.CronExpression = data.CronExpression
|
|||
|
|
j.JobId = data.JobId
|
|||
|
|
j.Name = data.JobName
|
|||
|
|
data.EntryId, err = jobs.AddJob(e.Cron, j)
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("jobs AddJob[HttpJob] error: %s", err)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
var j = &jobs.ExecJob{}
|
|||
|
|
j.InvokeTarget = data.InvokeTarget
|
|||
|
|
j.CronExpression = data.CronExpression
|
|||
|
|
j.JobId = data.JobId
|
|||
|
|
j.Name = data.JobName
|
|||
|
|
j.Args = data.Args
|
|||
|
|
data.EntryId, err = jobs.AddJob(e.Cron, j)
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("jobs AddJob[ExecJob] error: %s", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = e.Orm.Table(data.TableName()).Where(c.Id).Updates(&data).Error
|
|||
|
|
if err != nil {
|
|||
|
|
e.Log.Errorf("db error: %s", err)
|
|||
|
|
}
|
|||
|
|
return err
|
|||
|
|
}
|