package models import ( "time" "github.com/go-xorm/xorm" "errors" "strings" ) 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 Task struct { Id int `xorm:"int pk autoincr"` Name string `xorm:"varchar(32) notnull"` // 任务名称 Level TaskLevel `xorm:"smallint notnull index default 1"` // 任务等级 1: 主任务 2: 依赖任务 DependencyTaskId string `xorm:"varchar(64) notnull default ''"` // 依赖任务ID,多个ID逗号分隔 DependencyStatus TaskDependencyStatus `xorm:"smallint 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命令 Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制 Multi int8 `xorm:"tinyint notnull default 1"` // 是否允许多实例运行 RetryTimes int8 `xorm:"tinyint notnull default 0"` // 重试次数 NotifyStatus int8 `xorm:"smallint notnull default 1"` // 任务执行结束是否通知 0: 不通知 1: 失败通知 2: 执行结束通知 NotifyType int8 `xorm:"smallint notnull default 0"` // 通知类型 1: 邮件 2: slack NotifyReceiverId string `xorm:"varchar(256) notnull default '' "` // 通知接受者ID, setting表主键ID,多个ID逗号分隔 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 * * * * *" // 查询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,remark,notify_status,notify_type,notify_receiver_id, dependency_task_id, dependency_status"). 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) } }