| | |
| | | "apsClient/pkg/ecode" |
| | | "apsClient/pkg/logx" |
| | | "apsClient/pkg/structx" |
| | | "encoding/json" |
| | | "fmt" |
| | | "gorm.io/gorm" |
| | | "time" |
| | |
| | | Workers: slf.WorkerDistinct(workers), |
| | | } |
| | | |
| | | nowTs := time.Now().Unix() |
| | | var ( |
| | | err error |
| | | orders []*model.Order |
| | |
| | | |
| | | if mode == constvar.TaskModeUnStarted { |
| | | search.SetStatus(model.ProcedureStatusWaitProcess). |
| | | SetStartTimeMax(nowTs). |
| | | SetEndTimeMin(nowTs). |
| | | SetOrder("start_time asc") |
| | | } else if mode == constvar.TaskModeCurrent { |
| | | search.SetStatus(model.ProcedureStatusProcessing). |
| | | SetStartTimeMax(nowTs). |
| | | SetEndTimeMin(nowTs). |
| | | SetOrder("status desc, start_time asc") |
| | | SetOrder("start_time asc") |
| | | } else if mode == constvar.TaskModeLastFinished { |
| | | search.SetStatus(model.ProcedureStatusFinished).SetOrder("updated_at desc") |
| | | if len(channels) > 0 { |
| | |
| | | taskList = append(taskList, &taskData) |
| | | workers = append(workers, procedure.ProceduresInfo.Workers...) |
| | | } |
| | | |
| | | count, _ = search.SetStatus(model.ProcedureStatusWaitProcess).SetStartTimeMax(nowTs).SetEndTimeMin(nowTs).Count() |
| | | |
| | | taskResp = &response.TaskResponse{ |
| | | Tasks: taskList, |
| | | TaskCount: count, |
| | | Workers: slf.WorkerDistinct(workers), |
| | | Tasks: taskList, |
| | | Workers: slf.WorkerDistinct(workers), |
| | | } |
| | | return taskResp, ecode.OK |
| | | } |
| | | func (slf TaskService) NewTaskCount() (count int64) { |
| | | count, _ = model.NewProceduresSearch(nil).SetDeviceId(conf.Conf.System.DeviceId).SetStatus(model.ProcedureStatusWaitProcess).Count() |
| | | return count |
| | | } |
| | | |
| | | func (slf TaskService) WorkerDistinct(workers []*common.ProcedureWorker) (NoRepeatedWorkers []*common.ProcedureWorker) { |
| | |
| | | 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, |
| | | "channel": channel, |
| | | }) |
| | | |
| | | 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 { |
| | |
| | | return nil, err |
| | | } |
| | | if err == nil { |
| | | err = json.Unmarshal([]byte(data.Params), &data.ParamsMap) |
| | | if err != nil { |
| | | logx.Errorf("process model json.Unmarshal:%v", err) |
| | | return |
| | | } |
| | | return data, nil |
| | | } |
| | | |