gocron/models/task.go

183 lines
5.9 KiB
Go
Raw Normal View History

2017-03-10 09:24:06 +00:00
package models
import (
2017-04-02 02:38:49 +00:00
"time"
2017-04-21 09:41:59 +00:00
"github.com/go-xorm/xorm"
2017-03-10 09:24:06 +00:00
)
2017-04-13 09:35:59 +00:00
type TaskProtocol int8
2017-03-10 09:24:06 +00:00
2017-03-24 05:06:53 +00:00
const (
2017-04-13 09:35:59 +00:00
TaskHTTP TaskProtocol = iota + 1 // HTTP协议
2017-05-26 10:09:07 +00:00
TaskRPC // RPC方式执行命令
2017-03-24 05:06:53 +00:00
)
2017-03-23 05:31:16 +00:00
// 任务
2017-03-10 09:24:06 +00:00
type Task struct {
2017-04-02 02:38:49 +00:00
Id int `xorm:"int pk autoincr"`
2017-04-25 11:05:24 +00:00
Name string `xorm:"varchar(32) notnull"` // 任务名称
2017-04-07 01:13:36 +00:00
Spec string `xorm:"varchar(64) notnull"` // crontab
2017-05-26 10:09:07 +00:00
Protocol TaskProtocol `xorm:"tinyint notnull"` // 协议 1:http 2:系统命令
2017-04-25 11:05:24 +00:00
Command string `xorm:"varchar(256) notnull"` // URL地址或shell命令
2017-04-02 02:38:49 +00:00
Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制
Multi int8 `xorm:"tinyint notnull default 1"` // 是否允许多实例运行
RetryTimes int8 `xorm:"tinyint notnull default 0"` // 重试次数
2017-05-26 10:09:07 +00:00
HostId int16 `xorm:"smallint notnull default 0"` // RPC host id
2017-04-30 17:12:07 +00:00
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逗号分隔
2017-04-25 11:54:06 +00:00
Remark string `xorm:"varchar(100) notnull default ''"` // 备注
Status Status `xorm:"tinyint notnull default 0"` // 状态 1:正常 0:停止
2017-04-02 02:38:49 +00:00
Created time.Time `xorm:"datetime notnull created"` // 创建时间
Deleted time.Time `xorm:"datetime deleted"` // 删除时间
2017-04-23 06:11:53 +00:00
BaseModel `xorm:"-"`
2017-03-10 09:24:06 +00:00
}
type TaskHost struct {
Task `xorm:"extends"`
Name string
Port int
2017-04-13 09:35:59 +00:00
Alias string
}
func (TaskHost) TableName() string {
return TablePrefix + "task"
}
2017-03-10 09:24:06 +00:00
// 新增
2017-04-02 02:19:52 +00:00
func (task *Task) Create() (insertId int, err error) {
2017-04-02 02:38:49 +00:00
_, err = Db.Insert(task)
if err == nil {
insertId = task.Id
}
2017-03-24 09:55:44 +00:00
2017-04-02 02:38:49 +00:00
return
2017-03-10 09:24:06 +00:00
}
// 新增测试任务
func (task *Task) CreateTestTask() {
// HTTP任务
task.Name = "测试HTTP任务"
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,host_id,remark,notify_status,notify_type,notify_receiver_id").Update(task)
2017-04-17 10:04:30 +00:00
}
2017-03-10 09:24:06 +00:00
// 更新
2017-04-02 02:19:52 +00:00
func (task *Task) Update(id int, data CommonMap) (int64, error) {
2017-04-02 02:38:49 +00:00
return Db.Table(task).ID(id).Update(data)
2017-03-10 09:24:06 +00:00
}
// 删除
2017-04-02 02:19:52 +00:00
func (task *Task) Delete(id int) (int64, error) {
2017-04-02 02:38:49 +00:00
return Db.Id(id).Delete(task)
2017-03-10 09:24:06 +00:00
}
// 禁用
2017-04-02 02:19:52 +00:00
func (task *Task) Disable(id int) (int64, error) {
2017-04-02 02:38:49 +00:00
return task.Update(id, CommonMap{"status": Disabled})
2017-03-10 09:24:06 +00:00
}
// 激活
2017-04-02 02:19:52 +00:00
func (task *Task) Enable(id int) (int64, error) {
2017-04-02 02:38:49 +00:00
return task.Update(id, CommonMap{"status": Enabled})
2017-03-10 09:24:06 +00:00
}
2017-04-20 01:36:42 +00:00
// 获取所有激活任务
func (task *Task) ActiveList() ([]TaskHost, error) {
list := make([]TaskHost, 0)
2017-05-26 10:09:07 +00:00
fields := "t.*, host.alias,host.name,host.port"
err := Db.Alias("t").Join("LEFT", hostTableName(), "t.host_id=host.id").Where("t.status = ?", Enabled).Cols(fields).Find(&list)
2017-03-24 09:55:44 +00:00
2017-04-02 02:38:49 +00:00
return list, err
2017-03-24 09:55:44 +00:00
}
2017-04-20 01:36:42 +00:00
// 获取某个主机下的所有激活任务
2017-04-28 03:54:46 +00:00
func (task *Task) ActiveListByHostId(hostId int16) ([]TaskHost, error) {
2017-04-20 01:36:42 +00:00
list := make([]TaskHost, 0)
2017-05-26 10:09:07 +00:00
fields := "t.*, host.alias,host.name,host.port"
err := Db.Alias("t").Join("LEFT", hostTableName(), "t.host_id=host.id").Where("t.status = ? AND t.host_id = ?", Enabled, hostId).Cols(fields).Find(&list)
2017-04-20 01:36:42 +00:00
return list, err
}
2017-04-13 09:35:59 +00:00
// 判断主机id是否有引用
func (task *Task) HostIdExist(hostId int16) (bool, error) {
count, err := Db.Where("host_id = ?", hostId).Count(task);
return count > 0, err
}
// 判断任务名称是否存在
2017-04-17 10:04:30 +00:00
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
}
2017-04-13 09:35:59 +00:00
count, err := Db.Where("name = ? AND status = ?", name, Enabled).Count(task);
return count > 0, err
}
func(task *Task) Detail(id int) (TaskHost, error) {
taskHost := TaskHost{}
2017-05-26 10:09:07 +00:00
fields := "t.*, host.alias,host.name,host.port"
_, err := Db.Alias("t").Join("LEFT", hostTableName(), "t.host_id=host.id").Where("t.id=?", id).Cols(fields).Get(&taskHost)
2017-04-13 09:35:59 +00:00
return taskHost, err
}
2017-04-21 09:41:59 +00:00
func (task *Task) List(params CommonMap) ([]TaskHost, error) {
task.parsePageAndPageSize(params)
list := make([]TaskHost, 0)
fields := "t.*, host.alias,host.name"
session := Db.Alias("t").Join("LEFT", hostTableName(), "t.host_id=host.id")
task.parseWhere(session, params)
2017-04-21 09:41:59 +00:00
err := session.Cols(fields).Desc("t.id").Limit(task.PageSize, task.pageLimitOffset()).Find(&list)
2017-03-10 09:24:06 +00:00
2017-04-02 02:38:49 +00:00
return list, err
2017-03-10 09:24:06 +00:00
}
2017-04-23 06:11:53 +00:00
func (task *Task) Total(params CommonMap) (int64, error) {
session := Db.Alias("t").Join("LEFT", hostTableName(), "t.host_id=host.id")
2017-04-23 06:11:53 +00:00
task.parseWhere(session, params)
return session.Count(task)
2017-03-10 09:24:06 +00:00
}
2017-04-21 09:41:59 +00:00
// 解析where
func (task *Task) parseWhere(session *xorm.Session, params CommonMap) {
2017-04-21 09:41:59 +00:00
if len(params) == 0 {
return
}
id, ok := params["Id"]
if ok && id.(int) > 0 {
session.And("t.id = ?", id)
}
2017-04-21 09:41:59 +00:00
hostId, ok := params["HostId"]
if ok && hostId.(int) > 0 {
session.And("host_id = ?", hostId)
2017-04-21 09:41:59 +00:00
}
name, ok := params["Name"]
if ok && name.(string) != "" {
session.And("t.name LIKE ?", "%" + name.(string) + "%")
2017-04-21 09:41:59 +00:00
}
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)
}
}
func hostTableName() []string {
return []string{TablePrefix + "host", "host"}
2017-04-13 09:35:59 +00:00
}