mirror of https://github.com/ouqiang/gocron
81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package models
|
|
|
|
|
|
type TaskHost struct {
|
|
Id int `xorm:"int pk autoincr"`
|
|
TaskId int `xorm:"int not null index"`
|
|
HostId int16 `xorm:"smallint not null index"`
|
|
}
|
|
|
|
type TaskHostDetail struct {
|
|
TaskHost `xorm:"extends"`
|
|
Name string
|
|
Port int
|
|
Alias string
|
|
}
|
|
|
|
func (TaskHostDetail) TableName() string {
|
|
return TablePrefix + "task_host"
|
|
}
|
|
|
|
func hostTableName() []string {
|
|
return []string{TablePrefix + "host", "h"}
|
|
}
|
|
|
|
func (th *TaskHost) Remove(taskId int) error {
|
|
_, err := Db.Where("task_id = ?", taskId).Delete(new(TaskHost))
|
|
|
|
return err
|
|
}
|
|
|
|
func (th *TaskHost) Add(taskId int, hostIds []int) error {
|
|
|
|
err := th.Remove(taskId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
taskHosts := make([]TaskHost, len(hostIds))
|
|
for i, value := range hostIds {
|
|
taskHosts[i].TaskId = taskId
|
|
taskHosts[i].HostId = int16(value)
|
|
}
|
|
|
|
_, err = Db.Insert(&taskHosts)
|
|
|
|
return err
|
|
}
|
|
|
|
func (th *TaskHost) GetHostIdsByTaskId(taskId int) ([]TaskHostDetail, error) {
|
|
list := make([]TaskHostDetail, 0)
|
|
fields := "th.id,th.host_id,h.alias,h.name,h.port"
|
|
err := Db.Alias("th").
|
|
Join("LEFT", hostTableName(), "th.host_id=h.id").
|
|
Where("th.task_id = ?", taskId).
|
|
Cols(fields).
|
|
Find(&list)
|
|
|
|
return list, err
|
|
}
|
|
|
|
func (th *TaskHost) GetTaskIdsByHostId(hostId int16) ([]interface{}, error) {
|
|
list := make([]TaskHost, 0)
|
|
err := Db.Where("host_id = ?", hostId).Cols("task_id").Find(&list)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
taskIds := make([]interface{}, len(list))
|
|
for i, value := range list {
|
|
taskIds[i] = value.TaskId
|
|
}
|
|
|
|
return taskIds, err
|
|
}
|
|
|
|
// 判断主机id是否有引用
|
|
func (th *TaskHost) HostIdExist(hostId int16) (bool, error) {
|
|
count, err := Db.Where("host_id = ?", hostId).Count(th);
|
|
|
|
return count > 0, err
|
|
} |