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
}