42 lines
1.5 KiB
Go
42 lines
1.5 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"go-common/app/admin/ep/merlin/model"
|
|||
|
|
|
|||
|
|
pkgerr "github.com/pkg/errors"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// FindDeleteMachineTasks find delete machine tasks.
|
|||
|
|
func (d *Dao) FindDeleteMachineTasks() (tasks []*model.Task, err error) {
|
|||
|
|
tasks = []*model.Task{}
|
|||
|
|
err = pkgerr.WithStack(d.db.Where("status=? AND type=? AND DATEDIFF(NOW(),execute_time)=0 AND execute_time < NOW()", model.TaskInit, model.DeleteMachine).Find(&tasks).Error)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateTaskStatusByMachines update task status by machines.
|
|||
|
|
func (d *Dao) UpdateTaskStatusByMachines(machineIDs []int64, status int) (err error) {
|
|||
|
|
return pkgerr.WithStack(d.db.Model(&model.Task{}).Where("machine_id IN (?)", machineIDs).Update("status", status).Error)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateTaskStatusByTaskID update task status by taskId.
|
|||
|
|
func (d *Dao) UpdateTaskStatusByTaskID(taskID int64, status int) (err error) {
|
|||
|
|
return pkgerr.WithStack(d.db.Model(&model.Task{}).Where("id IN (?)", taskID).Update("status", status).Error)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InsertDeleteMachinesTasks insert delete machines tasks.
|
|||
|
|
// TODO:这块逻辑不是很好，如果一个写入失败希望继续写入失败的话最好返回失败list，后续联调时优化
|
|||
|
|
func (d *Dao) InsertDeleteMachinesTasks(ms []*model.Machine) (err error) {
|
|||
|
|
for _, machine := range ms {
|
|||
|
|
mrTask := &model.Task{
|
|||
|
|
TYPE: model.DeleteMachine,
|
|||
|
|
ExecuteTime: machine.EndTime,
|
|||
|
|
MachineID: machine.ID,
|
|||
|
|
Status: model.TaskInit,
|
|||
|
|
}
|
|||
|
|
if tempErr := d.db.Create(mrTask).Error; tempErr != nil {
|
|||
|
|
err = pkgerr.WithStack(tempErr)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|