zhangqian
2023-10-08 a8622c9cc1b20eac20e384e704aa98194372e72a
service/task.go
@@ -131,13 +131,19 @@
   return procedure, ecode.OK
}
func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id int, status model.ProcedureStatus, channel int32) error {
   if status == model.ProcedureStatusFinished {
func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id int, status model.ProcedureStatus, channel int32, processModelNumber string) error {
   if status == model.ProcedureStatusFinished || status == model.ProcedureStatusWaitProcess {
      ProgressCacheUnset(channel)
   }
   return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
      "status": status,
   })
   upMap := map[string]interface{}{"status": status}
   if status == model.ProcedureStatusProcessing {
      upMap["process_model_number"] = processModelNumber
   } else if status == model.ProcedureStatusWaitProcess {
      upMap["process_model_number"] = ""
   }
   return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
}
func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id int, status model.ProcedureStatus, channel int32) error {