From a699ff979c84d024ca33fb1139224258f278319f Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 20 十月 2023 11:32:23 +0800 Subject: [PATCH] debug --- service/task.go | 64 ++++++++++++++----------------- 1 files changed, 29 insertions(+), 35 deletions(-) diff --git a/service/task.go b/service/task.go index f9413cc..fbc355f 100644 --- a/service/task.go +++ b/service/task.go @@ -6,12 +6,10 @@ "apsClient/model" "apsClient/model/common" "apsClient/model/response" - "apsClient/nsq" "apsClient/pkg/ecode" "apsClient/pkg/logx" - "apsClient/pkg/structx" "fmt" - "gorm.io/gorm" + "github.com/jinzhu/gorm" "time" ) @@ -123,7 +121,7 @@ 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 @@ -131,17 +129,22 @@ 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, - "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 { +func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id uint, status model.ProcedureStatus, channel int32) error { if status == model.ProcedureStatusFinished { ProgressCacheUnset(channel) } @@ -172,30 +175,6 @@ if err == nil { return data, nil } - - if err == gorm.ErrRecordNotFound { //濡傛灉鏁版嵁搴撴病鏈変粠浜戠鑾峰彇 - caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicProcessParamsRequest, conf.Conf.NsqConf.NodeId), fmt.Sprintf(constvar.NsqTopicProcessParamsResponse, conf.Conf.NsqConf.NodeId)) - var result common.ResponseProcessParams - err = caller.Call(common.RequestProcessParams{ - WorkOrder: procedure.WorkOrderID, - OrderId: procedure.OrderID, - Product: order.ProductName, - Procedure: procedure.ProceduresInfo.ProcedureName, - Device: procedure.ProceduresInfo.DeviceName, - DeviceId: conf.Conf.System.DeviceId, - }, &result, time.Second*3) - if err != nil { - logx.Errorf("TaskStart GetProcessModel error:%v", err.Error()) - return - } - if result.ParamsMap == nil { - logx.Errorf("TaskStart GetProcessModel response miss process params:%v", result) - return - } - processModel = new(model.ProcessModel) - err = structx.AssignTo(result, &processModel) - return processModel, err - } return } @@ -207,3 +186,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() +} -- Gitblit v1.8.0