zhangqian
2023-10-26 e4957d48bb364f7ee34bd672f53b71a66ca31e14
service/task.go
@@ -137,10 +137,18 @@
   upMap := map[string]interface{}{"status": status}
   if status == model.ProcedureStatusProcessing {
      upMap["process_model_number"] = processModelNumber
      upMap["real_start_time"] = time.Now().Unix()
   } else if status == model.ProcedureStatusWaitProcess {
      upMap["process_model_number"] = ""
   } else if status == model.ProcedureStatusFinished {
      upMap["real_end_time"] = time.Now().Unix()
   }
   return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
}
func (slf TaskService) UpdateProcessModelNumber(db *gorm.DB, id uint, processModelNumber string) error {
   upMap := map[string]interface{}{"process_model_number": processModelNumber}
   return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
}
@@ -186,3 +194,18 @@
      SetStartTimeMin(nowTs).
      SetStatus(model.OrderStatusWaitProcess).First()
}
// SaveTaskStatusSync 保存任务同步信息
func (slf TaskService) SaveTaskStatusSync(db *gorm.DB, record *model.TaskStatusSync) (err error) {
   return model.NewTaskStatusSyncSearch(db).Create(record)
}
// GetTaskStatusSync 获取待同步任务信息
func (slf TaskService) GetTaskStatusSync(limit int) ([]*model.TaskStatusSync, error) {
   return model.NewTaskStatusSyncSearch(nil).SetOrder("id desc").SetPage(1, limit).FindNotTotal()
}
// RemoveTaskStatusSync 删除同步完的记录
func (slf TaskService) RemoveTaskStatusSync(ids []uint) (err error) {
   return model.NewTaskStatusSyncSearch(nil).SetIDs(ids).Delete()
}