mirror of https://github.com/ouqiang/gocron
283 lines
8.0 KiB
Go
283 lines
8.0 KiB
Go
package models
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/go-xorm/xorm"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type TaskProtocol int8
|
||
|
||
const (
|
||
TaskHTTP TaskProtocol = iota + 1 // HTTP协议
|
||
TaskRPC // RPC方式执行命令
|
||
)
|
||
|
||
type TaskLevel int8
|
||
|
||
const (
|
||
TaskLevelParent TaskLevel = 1 // 父任务
|
||
TaskLevelChild TaskLevel = 2 // 子任务(依赖任务)
|
||
)
|
||
|
||
type TaskDependencyStatus int8
|
||
|
||
const (
|
||
TaskDependencyStatusStrong TaskDependencyStatus = 1 // 强依赖
|
||
TaskDependencyStatusWeak TaskDependencyStatus = 2 // 弱依赖
|
||
)
|
||
|
||
type TaskHTTPMethod int8
|
||
|
||
const (
|
||
TaskHTTPMethodGet TaskHTTPMethod = 1;
|
||
TaskHttpMethodPost TaskHTTPMethod = 2;
|
||
)
|
||
|
||
// 任务
|
||
type Task struct {
|
||
Id int `xorm:"int pk autoincr"`
|
||
Name string `xorm:"varchar(32) notnull"` // 任务名称
|
||
Level TaskLevel `xorm:"tinyint notnull index default 1"` // 任务等级 1: 主任务 2: 依赖任务
|
||
DependencyTaskId string `xorm:"varchar(64) notnull default ''"` // 依赖任务ID,多个ID逗号分隔
|
||
DependencyStatus TaskDependencyStatus `xorm:"tinyint notnull default 1"` // 依赖关系 1:强依赖 主任务执行成功, 依赖任务才会被执行 2:弱依赖
|
||
Spec string `xorm:"varchar(64) notnull"` // crontab
|
||
Protocol TaskProtocol `xorm:"tinyint notnull index"` // 协议 1:http 2:系统命令
|
||
Command string `xorm:"varchar(256) notnull"` // URL地址或shell命令
|
||
HttpMethod TaskHTTPMethod `xorm:"tinyint notnull default 1"` // http请求方法
|
||
Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制
|
||
Multi int8 `xorm:"tinyint notnull default 1"` // 是否允许多实例运行
|
||
RetryTimes int8 `xorm:"tinyint notnull default 0"` // 重试次数
|
||
RetryInterval int16 `xorm:"smallint notnull default 0"` // 重试间隔时间
|
||
NotifyStatus int8 `xorm:"tinyint notnull default 1"` // 任务执行结束是否通知 0: 不通知 1: 失败通知 2: 执行结束通知
|
||
NotifyType int8 `xorm:"tinyint notnull default 0"` // 通知类型 1: 邮件 2: slack
|
||
NotifyReceiverId string `xorm:"varchar(256) notnull default '' "` // 通知接受者ID, setting表主键ID，多个ID逗号分隔
|
||
Tag string `xorm:"varchar(32) notnull default ''"`
|
||
Remark string `xorm:"varchar(100) notnull default ''"` // 备注
|
||
Status Status `xorm:"tinyint notnull index default 0"` // 状态 1:正常 0:停止
|
||
Created time.Time `xorm:"datetime notnull created"` // 创建时间
|
||
Deleted time.Time `xorm:"datetime deleted"` // 删除时间
|
||
BaseModel `xorm:"-"`
|
||
Hosts []TaskHostDetail `xorm:"-"`
|
||
}
|
||
|
||
func taskHostTableName() []string {
|
||
return []string{TablePrefix + "task_host", "th"}
|
||
}
|
||
|
||
// 新增
|
||
func (task *Task) Create() (insertId int, err error) {
|
||
_, err = Db.Insert(task)
|
||
if err == nil {
|
||
insertId = task.Id
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// 新增测试任务
|
||
func (task *Task) CreateTestTask() {
|
||
// HTTP任务
|
||
task.Name = "测试HTTP任务"
|
||
task.Level = TaskLevelParent
|
||
task.Protocol = TaskHTTP
|
||
task.Spec = "*/30 * * * * *"
|
||
task.Tag = "test-task"
|
||
// 查询IP地址区域信息
|
||
task.Command = "http://ip.taobao.com/service/getIpInfo.php?ip=117.27.140.253"
|
||
task.Status = Enabled
|
||
task.Create()
|
||
}
|
||
|
||
func (task *Task) UpdateBean(id int) (int64, error) {
|
||
return Db.ID(id).
|
||
Cols(`name,spec,protocol,command,timeout,multi,
|
||
retry_times,retry_interval,remark,notify_status,
|
||
notify_type,notify_receiver_id, dependency_task_id, dependency_status, tag,http_method`).
|
||
Update(task)
|
||
}
|
||
|
||
// 更新
|
||
func (task *Task) Update(id int, data CommonMap) (int64, error) {
|
||
return Db.Table(task).ID(id).Update(data)
|
||
}
|
||
|
||
// 删除
|
||
func (task *Task) Delete(id int) (int64, error) {
|
||
return Db.Id(id).Delete(task)
|
||
}
|
||
|
||
// 禁用
|
||
func (task *Task) Disable(id int) (int64, error) {
|
||
return task.Update(id, CommonMap{"status": Disabled})
|
||
}
|
||
|
||
// 激活
|
||
func (task *Task) Enable(id int) (int64, error) {
|
||
return task.Update(id, CommonMap{"status": Enabled})
|
||
}
|
||
|
||
// 获取所有激活任务
|
||
func (task *Task) ActiveList() ([]Task, error) {
|
||
list := make([]Task, 0)
|
||
err := Db.Where("status = ? AND level = ?", Enabled, TaskLevelParent).
|
||
Find(&list)
|
||
|
||
if err != nil {
|
||
return list, err
|
||
}
|
||
|
||
return task.setHostsForTasks(list)
|
||
}
|
||
|
||
// 获取某个主机下的所有激活任务
|
||
func (task *Task) ActiveListByHostId(hostId int16) ([]Task, error) {
|
||
taskHostModel := new(TaskHost)
|
||
taskIds, err := taskHostModel.GetTaskIdsByHostId(hostId)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
list := make([]Task, 0)
|
||
err = Db.Where("status = ? AND level = ?", Enabled, TaskLevelParent).
|
||
In("id", taskIds...).
|
||
Find(&list)
|
||
if err != nil {
|
||
return list, err
|
||
}
|
||
|
||
return task.setHostsForTasks(list)
|
||
}
|
||
|
||
func (task *Task) setHostsForTasks(tasks []Task) ([]Task, error) {
|
||
taskHostModel := new(TaskHost)
|
||
var err error
|
||
for i, value := range tasks {
|
||
taskHostDetails, err := taskHostModel.GetHostIdsByTaskId(value.Id)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
tasks[i].Hosts = taskHostDetails
|
||
}
|
||
|
||
return tasks, err
|
||
}
|
||
|
||
// 判断任务名称是否存在
|
||
func (task *Task) NameExist(name string, id int) (bool, error) {
|
||
if id > 0 {
|
||
count, err := Db.Where("name = ? AND status = ? AND id != ?", name, Enabled, id).Count(task)
|
||
return count > 0, err
|
||
}
|
||
count, err := Db.Where("name = ? AND status = ?", name, Enabled).Count(task)
|
||
|
||
return count > 0, err
|
||
}
|
||
|
||
func (task *Task) GetStatus(id int) (Status, error) {
|
||
exist, err := Db.Id(id).Get(task)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
if !exist {
|
||
return 0, errors.New("not exist")
|
||
}
|
||
|
||
return task.Status, nil
|
||
}
|
||
|
||
func (task *Task) Detail(id int) (Task, error) {
|
||
t := Task{}
|
||
_, err := Db.Where("id=?", id).Get(&t)
|
||
|
||
if err != nil {
|
||
return t, err
|
||
}
|
||
|
||
taskHostModel := new(TaskHost)
|
||
t.Hosts, err = taskHostModel.GetHostIdsByTaskId(id)
|
||
|
||
return t, err
|
||
}
|
||
|
||
func (task *Task) List(params CommonMap) ([]Task, error) {
|
||
task.parsePageAndPageSize(params)
|
||
list := make([]Task, 0)
|
||
session := Db.Alias("t").Join("LEFT", taskHostTableName(), "t.id = th.task_id")
|
||
task.parseWhere(session, params)
|
||
err := session.GroupBy("t.id").Desc("t.id").Cols("t.*").Limit(task.PageSize, task.pageLimitOffset()).Find(&list)
|
||
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return task.setHostsForTasks(list)
|
||
}
|
||
|
||
// 获取依赖任务列表
|
||
func (task *Task) GetDependencyTaskList(ids string) ([]Task, error) {
|
||
list := make([]Task, 0)
|
||
if ids == "" {
|
||
return list, nil
|
||
}
|
||
idList := strings.Split(ids, ",")
|
||
taskIds := make([]interface{}, len(idList))
|
||
for i, v := range idList {
|
||
taskIds[i] = v
|
||
}
|
||
fields := "t.*"
|
||
err := Db.Alias("t").
|
||
Where("t.level = ?", TaskLevelChild).
|
||
In("t.id", taskIds).
|
||
Cols(fields).
|
||
Find(&list)
|
||
|
||
if err != nil {
|
||
return list, err
|
||
}
|
||
|
||
return task.setHostsForTasks(list)
|
||
}
|
||
|
||
func (task *Task) Total(params CommonMap) (int64, error) {
|
||
session := Db.Alias("t").Join("LEFT", taskHostTableName(), "t.id = th.task_id")
|
||
task.parseWhere(session, params)
|
||
list := make([]Task, 0)
|
||
|
||
err := session.GroupBy("t.id").Find(&list)
|
||
|
||
return int64(len(list)), err
|
||
}
|
||
|
||
// 解析where
|
||
func (task *Task) parseWhere(session *xorm.Session, params CommonMap) {
|
||
if len(params) == 0 {
|
||
return
|
||
}
|
||
id, ok := params["Id"]
|
||
if ok && id.(int) > 0 {
|
||
session.And("t.id = ?", id)
|
||
}
|
||
hostId, ok := params["HostId"]
|
||
if ok && hostId.(int) > 0 {
|
||
session.And("th.host_id = ?", hostId)
|
||
}
|
||
name, ok := params["Name"]
|
||
if ok && name.(string) != "" {
|
||
session.And("t.name LIKE ?", "%"+name.(string)+"%")
|
||
}
|
||
protocol, ok := params["Protocol"]
|
||
if ok && protocol.(int) > 0 {
|
||
session.And("protocol = ?", protocol)
|
||
}
|
||
status, ok := params["Status"]
|
||
if ok && status.(int) > -1 {
|
||
session.And("status = ?", status)
|
||
}
|
||
|
||
tag, ok := params["Tag"]
|
||
if ok && tag.(string) != "" {
|
||
session.And("tag = ? ", tag)
|
||
}
|
||
}
|