2017-03-10 09:24:06 +00:00
|
|
|
|
package models
|
|
|
|
|
|
|
|
|
|
import (
|
2017-04-02 02:38:49 +00:00
|
|
|
|
"time"
|
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协议
|
|
|
|
|
TaskSSH // SSH命令
|
2017-04-16 18:01:41 +00:00
|
|
|
|
TaskLocalCommand // 本地命令
|
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"`
|
|
|
|
|
Name string `xorm:"varchar(64) notnull"` // 任务名称
|
2017-04-07 01:13:36 +00:00
|
|
|
|
Spec string `xorm:"varchar(64) notnull"` // crontab
|
2017-04-16 18:01:41 +00:00
|
|
|
|
Protocol TaskProtocol `xorm:"tinyint notnull"` // 协议 1:http 2:ssh-command 3: 本地命令
|
2017-04-02 02:38:49 +00:00
|
|
|
|
Command string `xorm:"varchar(512) notnull"` // URL地址或shell命令
|
|
|
|
|
Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制
|
2017-04-13 09:35:59 +00:00
|
|
|
|
HostId int16 `xorm:"smallint notnull default 0"` // SSH host id,
|
2017-04-02 02:38:49 +00:00
|
|
|
|
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:"-"`
|
2017-03-10 09:24:06 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-04-10 09:37:16 +00:00
|
|
|
|
type TaskHost struct {
|
|
|
|
|
Task `xorm:"extends"`
|
|
|
|
|
Name string
|
|
|
|
|
Port int
|
|
|
|
|
Username string
|
|
|
|
|
Password string
|
2017-04-13 09:35:59 +00:00
|
|
|
|
Alias string
|
2017-04-10 09:37:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (TaskHost) TableName() string {
|
|
|
|
|
return "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
|
|
|
|
}
|
|
|
|
|
|
2017-04-17 10:04:30 +00:00
|
|
|
|
func (task *Task) UpdateBean(id int) (int64, error) {
|
|
|
|
|
return Db.UseBool("status").Update(task)
|
|
|
|
|
}
|
|
|
|
|
|
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-10 09:37:16 +00:00
|
|
|
|
func (task *Task) ActiveList() ([]TaskHost, error) {
|
2017-04-02 02:38:49 +00:00
|
|
|
|
task.parsePageAndPageSize()
|
2017-04-10 09:37:16 +00:00
|
|
|
|
list := make([]TaskHost, 0)
|
|
|
|
|
fields := "t.*, host.name,host.username,host.password,host.port"
|
2017-04-13 09:35:59 +00:00
|
|
|
|
err := Db.Alias("t").Join("LEFT", "host", "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-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-04-10 09:37:16 +00:00
|
|
|
|
fields := "t.*, host.name,host.username,host.password,host.port"
|
2017-04-13 09:35:59 +00:00
|
|
|
|
_, err := Db.Alias("t").Join("LEFT", "host", "t.host_id=host.id").Where("t.id=?", id).Cols(fields).Get(&taskHost)
|
2017-04-10 09:37:16 +00:00
|
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
|
return taskHost, err
|
2017-04-10 09:37:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (task *Task) List() ([]TaskHost, error) {
|
2017-04-02 02:38:49 +00:00
|
|
|
|
task.parsePageAndPageSize()
|
2017-04-10 09:37:16 +00:00
|
|
|
|
list := make([]TaskHost, 0)
|
2017-04-13 09:35:59 +00:00
|
|
|
|
fields := "t.*, host.alias"
|
2017-04-10 09:37:16 +00:00
|
|
|
|
err := Db.Alias("t").Join("LEFT", "host", "t.host_id=host.id").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-13 09:35:59 +00:00
|
|
|
|
func (task *Task) Total() (int64, error) {
|
|
|
|
|
return Db.Count(task)
|
2017-03-10 09:24:06 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
|
func (task *Task) parsePageAndPageSize() {
|
|
|
|
|
if task.Page <= 0 {
|
|
|
|
|
task.Page = Page
|
2017-04-02 02:38:49 +00:00
|
|
|
|
}
|
2017-04-13 09:35:59 +00:00
|
|
|
|
if task.PageSize >= 0 || task.PageSize > MaxPageSize {
|
|
|
|
|
task.PageSize = PageSize
|
2017-04-02 02:38:49 +00:00
|
|
|
|
}
|
2017-03-10 09:24:06 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
|
func (task *Task) pageLimitOffset() int {
|
|
|
|
|
return (task.Page - 1) * task.PageSize
|
|
|
|
|
}
|