gocron/models/task_host.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
}