mirror of https://github.com/ouqiang/gocron
126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
package models
|
||
|
||
import (
|
||
"time"
|
||
)
|
||
|
||
type Protocol int8
|
||
|
||
type TaskType int8
|
||
|
||
const (
|
||
HTTP Protocol = iota + 1 // HTTP协议
|
||
SSHCommand // SSHM命令
|
||
)
|
||
|
||
const (
|
||
Timing TaskType = iota + 1 // 定时任务
|
||
Delay // 延时任务
|
||
)
|
||
|
||
// 任务
|
||
type Task struct {
|
||
Id int `xorm:"int pk autoincr"`
|
||
Name string `xorm:"varchar(64) notnull"` // 任务名称
|
||
Spec string `xorm:"varchar(64) notnull"` // crontab
|
||
Protocol Protocol `xorm:"tinyint notnull"` // 协议 1:http 2:ssh-command
|
||
Type TaskType `xorm:"tinyint notnull default 1"` // 任务类型 1: 定时任务 2: 延时任务
|
||
Command string `xorm:"varchar(512) notnull"` // URL地址或shell命令
|
||
Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制
|
||
Delay int `xorm:"int notnull default 0"` // 延时任务，延时时间(单位秒)
|
||
HostId int16 `xorm:"smallint notnull default 0"` // SSH host id，
|
||
Remark string `xorm:"varchar(512) notnull default ''"` // 备注
|
||
Created time.Time `xorm:"datetime notnull created"` // 创建时间
|
||
Deleted time.Time `xorm:"datetime deleted"` // 删除时间
|
||
Status Status `xorm:"tinyint notnull default 1"` // 状态 1:正常 0:停止
|
||
Page int `xorm:"-"`
|
||
PageSize int `xorm:"-"`
|
||
}
|
||
|
||
type TaskHost struct {
|
||
Task `xorm:"extends"`
|
||
Name string
|
||
Port int
|
||
Username string
|
||
Password string
|
||
}
|
||
|
||
func (TaskHost) TableName() string {
|
||
return "task"
|
||
}
|
||
|
||
// 新增
|
||
func (task *Task) Create() (insertId int, err error) {
|
||
task.Status = Enabled
|
||
|
||
_, err = Db.Insert(task)
|
||
if err == nil {
|
||
insertId = task.Id
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// 更新
|
||
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() ([]TaskHost, error) {
|
||
task.parsePageAndPageSize()
|
||
list := make([]TaskHost, 0)
|
||
fields := "t.*, host.name,host.username,host.password,host.port"
|
||
err := Db.Alias("t").Join("LEFT", "host", "t.host_id=host.id").Where("status = ?", Enabled).Cols(fields).Find(&list)
|
||
|
||
return list, err
|
||
}
|
||
|
||
func(task *Task) Detail(id int) error {
|
||
list := make([]TaskHost, 0)
|
||
fields := "t.*, host.name,host.username,host.password,host.port"
|
||
err := Db.Alias("t").Join("LEFT", "host", "t.host_id=host.id").Cols(fields).Find(list)
|
||
|
||
return err
|
||
}
|
||
|
||
func (task *Task) List() ([]TaskHost, error) {
|
||
task.parsePageAndPageSize()
|
||
list := make([]TaskHost, 0)
|
||
fields := "t.*, host.name"
|
||
err := Db.Alias("t").Join("LEFT", "host", "t.host_id=host.id").Cols(fields).Desc("t.id").Limit(task.PageSize, task.pageLimitOffset()).Find(&list)
|
||
|
||
return list, err
|
||
}
|
||
|
||
func (taskLog *TaskLog) Total() (int64, error) {
|
||
return Db.Count(taskLog)
|
||
}
|
||
|
||
func (taskLog *TaskLog) parsePageAndPageSize() {
|
||
if taskLog.Page <= 0 {
|
||
taskLog.Page = Page
|
||
}
|
||
if taskLog.PageSize >= 0 || taskLog.PageSize > MaxPageSize {
|
||
taskLog.PageSize = PageSize
|
||
}
|
||
}
|
||
|
||
func (taskLog *TaskLog) pageLimitOffset() int {
|
||
return (taskLog.Page - 1) * taskLog.PageSize
|
||
}
|