| | |
| | | "apsClient/pkg/logx" |
| | | "apsClient/pkg/structx" |
| | | "fmt" |
| | | "gorm.io/gorm" |
| | | "github.com/jinzhu/gorm" |
| | | "time" |
| | | ) |
| | | |
| | |
| | | return NoRepeatedWorkers |
| | | } |
| | | |
| | | func (slf TaskService) GetProcedureById(id int) (*model.Procedures, int) { |
| | | func (slf TaskService) GetProcedureById(id uint) (*model.Procedures, int) { |
| | | procedure, err := model.NewProceduresSearch(nil).SetId(id).First() |
| | | if err != nil { |
| | | return nil, ecode.DBErr |
| | |
| | | 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 uint, 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 { |
| | | func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id uint, status model.ProcedureStatus, channel int32) error { |
| | | if status == model.ProcedureStatusFinished { |
| | | ProgressCacheUnset(channel) |
| | | } |