From 066a30751bdc20f9e83b34539de71ae392783e1b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期六, 19 八月 2023 15:58:09 +0800 Subject: [PATCH] 网络配置获取和删除,品牌增删改查 --- nsq/msg_handler.go | 66 ++++++++++++++++++++------------ 1 files changed, 41 insertions(+), 25 deletions(-) diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go index 495b6ef..9485653 100644 --- a/nsq/msg_handler.go +++ b/nsq/msg_handler.go @@ -1,10 +1,13 @@ package nsq import ( + "apsClient/conf" "apsClient/model" + "apsClient/model/common" "apsClient/pkg/logx" + "apsClient/pkg/structx" "encoding/json" - "fmt" + "gorm.io/gorm" ) type MsgHandler interface { @@ -15,36 +18,49 @@ } func (slf *ScheduleTask) HandleMessage(data []byte) (err error) { - fmt.Println(string(data)) - - var tasks = make([]*DeliverScheduleTask, 0) - + logx.Infof("get an message :%s", data) + var tasks = make([]*common.DeliverScheduleTask, 0) err = json.Unmarshal(data, &tasks) if err != nil { logx.Errorf("ScheduleTask HandleMessage Unmarshal json err: %v", err.Error()) return err } for _, task := range tasks { - taskRecord := model.ScheduleTask{ - Id: 0, - OrderId: task.Order.OrderID, - ProductId: task.Order.ProductID, - ProductName: task.Order.ProductName, - Amount: task.Order.Amount, - Unit: task.Order.Unit, - StartTime: task.Order.StartTime, - EndTime: task.Order.EndTime, - } - jsonStr, err := json.Marshal(task) - if err != nil { - logx.Errorf("ScheduleTask HandleMessage Marshal err: %v, old: %#v", err.Error(), task) - return err - } - taskRecord.Data = string(jsonStr) - err = model.NewScheduleTaskSearch(nil).Create(&taskRecord) - if err != nil { - logx.Errorf("ScheduleTask HandleMessage Create taskRecord err: %v, record: %#v", err.Error(), taskRecord) - return err + for _, procedure := range task.Procedures { + if procedure.DeviceID == conf.Conf.System.DeviceId { //鍙瓨鍌ㄦ湰璁惧鐨勫伐搴� + err = model.WithTransaction(func(db *gorm.DB) error { + _, err := model.NewOrderSearch(db).SetWorkOrderId(task.WorkOrder.WorkOrderID).First() + if err == gorm.ErrRecordNotFound { + var orderRecord model.Order + err = structx.AssignTo(task.WorkOrder, &orderRecord) + orderRecord.Status = model.OrderStatusUnFinished + if err != nil { + logx.Errorf(" structx.Assign task.Order to orderRecord err: %v", err.Error()) + return err + } + if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil { + return err + } + } + procedureRecord := model.Procedures{ + StartTime: procedure.StartTime, + EndTime: procedure.EndTime, + WorkOrderID: task.WorkOrder.WorkOrderID, + OrderID: task.WorkOrder.OrderID, + Status: model.ProcedureStatusUnFinished, + } + procedureData, err := json.Marshal(procedure) + procedureRecord.ProcedureData = string(procedureData) + if err != nil { + logx.Errorf(" json.Marshal(procedure) err: %v", err.Error()) + return err + } + return model.NewProceduresSearch(db).Create(&procedureRecord) + }) + if err != nil { + logx.Errorf(" save procedure err: %v", err.Error()) + } + } } } return nil -- Gitblit v1.8.0