From 81cda4f09d4c0fff6edd087d1c49d05cb6eaab38 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期一, 09 十月 2023 20:47:27 +0800 Subject: [PATCH] 当前使用的工艺编号置顶 --- service/task.go | 42 ++++++++++++++++++------------------------ 1 files changed, 18 insertions(+), 24 deletions(-) diff --git a/service/task.go b/service/task.go index dd1c11c..a9b75ca 100644 --- a/service/task.go +++ b/service/task.go @@ -10,7 +10,6 @@ "apsClient/pkg/ecode" "apsClient/pkg/logx" "apsClient/pkg/structx" - "encoding/json" "fmt" "gorm.io/gorm" "time" @@ -35,7 +34,6 @@ Workers: slf.WorkerDistinct(workers), } - nowTs := time.Now().Unix() var ( err error orders []*model.Order @@ -48,14 +46,10 @@ 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 { @@ -106,15 +100,15 @@ 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) { @@ -137,14 +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, - "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 { @@ -176,11 +175,6 @@ 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 } -- Gitblit v1.8.0