From c8303f55819e2d8b5a56e6316f79c2250d9b659c Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期日, 08 十月 2023 21:16:09 +0800 Subject: [PATCH] 工序更新时保存使用的工艺参数编号 --- nsq/msg_handler.go | 151 ++++++++++++++++++++++++++++++++++++++++---------- 1 files changed, 120 insertions(+), 31 deletions(-) diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go index 00c63d5..175f3e7 100644 --- a/nsq/msg_handler.go +++ b/nsq/msg_handler.go @@ -11,6 +11,7 @@ "apsClient/utils/file" "encoding/json" "fmt" + "github.com/mitchellh/mapstructure" "github.com/spf13/cast" "gorm.io/gorm" "strings" @@ -54,10 +55,11 @@ WorkOrderID: task.WorkOrder.WorkOrderID, OrderID: task.WorkOrder.OrderID, DeviceID: procedure.DeviceID, + ProcedureID: procedure.ProcedureID, StartTime: procedure.StartTime, EndTime: procedure.EndTime, Status: model.ProcedureStatusWaitProcess, - ProcedureData: "", + Channel: procedure.Channel, ProceduresInfo: common.ProductProcedure{}, } procedureData, err := json.Marshal(procedure) @@ -77,10 +79,23 @@ } err = model.WithTransaction(func(db *gorm.DB) error { - if err = model.NewOrderSearch(db).Save(&orderRecord); err != nil { - return err + if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" { + if err = model.NewOrderSearch(db).SetId(oldWorkOrder.ID).Updates(&orderRecord); err != nil { + return err + } + for _, procedure := range procedureRecords { + err = model.NewProceduresSearch(db).SetWorkOrderId(procedure.WorkOrderID).SetProcedureId(procedure.ProcedureID).Upsert(procedure) + if err != nil { + return err + } + } + } else { + if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil { + return err + } + return model.NewProceduresSearch(db).CreateBatch(procedureRecords) } - return model.NewProceduresSearch(db).CreateBatch(procedureRecords) + return nil }) if err != nil { logx.Errorf(" save task message error err: %v", err.Error()) @@ -102,38 +117,63 @@ logx.Errorf("ScheduleTask HandleMessage Unmarshal json err: %v", err.Error()) return nil } - if len(resp.KeyData) == 0 || len(resp.AddressData) == 0 { + if resp.DeviceId != conf.Conf.System.DeviceId { return nil - } - //鍐欏叆鍒版枃浠� - err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataKeyFileName), resp.KeyData) - if err != nil { - return err - } - err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataValueFileName), resp.AddressData) - if err != nil { - return err - } - //鍐欏叆鍒板唴瀛� - keyString := string(resp.KeyData) - addressString := string(resp.AddressData) - - keys := strings.Split(keyString, "\n") - addresses := strings.Split(addressString, "\n") - if len(keys) != len(addresses) { - logx.Error("plc address message error: key length not equal address length") - return nil - } - for i := 0; i < len(keys); i++ { - key := strings.ReplaceAll(keys[i], "\r", "") - address := cast.ToInt(strings.ReplaceAll(addresses[i], "\r", "")) - plc_address.Set(key, address) - logx.Infof("plc address set ok: key:%v, address:%v", key, address) } //閫氱煡鍥炲鏀跺埌 ReceivedMessageChan <- &ReceivedMessage{ Topic: slf.Topic, Message: data, + } + if len(resp.KeyData) != 0 && len(resp.AddressData) != 0 { + //鍐欏叆鍒版枃浠� + err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataKeyFileName), resp.KeyData) + if err != nil { + return err + } + err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataValueFileName), resp.AddressData) + if err != nil { + return err + } + //鍐欏叆鍒板唴瀛� + keyString := string(resp.KeyData) + addressString := string(resp.AddressData) + + keys := strings.Split(keyString, "\n") + addresses := strings.Split(addressString, "\n") + if len(keys) != len(addresses) { + logx.Error("plc address message error: key length not equal address length") + return nil + } + for i := 0; i < len(keys); i++ { + key := strings.ReplaceAll(keys[i], "\r", "") + address := cast.ToInt(strings.ReplaceAll(addresses[i], "\r", "")) + plc_address.Set(key, address) + logx.Infof("plc address set ok: key:%v, address:%v", key, address) + } + } + + if resp.PlcConfig.Method == "" { + return nil + } + + resp.PlcConfig.Method.ConvertToEnglish() + + var record model.DevicePlc + err = mapstructure.Decode(resp.PlcConfig, &record) + if err != nil { + logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err) + } + err = mapstructure.Decode(resp.PlcAddressList, &record.Details) + if err != nil { + logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err) + } + detail, err := json.Marshal(record.Details) + record.Detail = string(detail) + record.Id = 1 + err = model.NewDevicePlcSearch().SetId(record.Id).Save(&record) + if err != nil { + return err } return nil } @@ -179,10 +219,59 @@ logx.Infof("unmarshal process params sync err :%s", err) return err } - err = model.NewProcessModelSearch().Create(&processModel) + err = model.WithTransaction(func(db *gorm.DB) error { + err = model.NewProcessModelSearch().SetOrm(db).SetProduct(processModel.Product).SetProcedure(processModel.Procedure). + UpdateByMap(map[string]interface{}{"is_new": 0}) + if err != nil { + return err + } + processModel.IsNew = true + return model.NewProcessModelSearch().SetOrm(db).Create(&processModel) + }) + if err != nil { logx.Infof("save process params sync err :%s", err) return err } return nil } + +type DeviceUpdate struct { + Topic string +} + +func (slf *DeviceUpdate) HandleMessage(data []byte) (err error) { + logx.Infof("get a device update message :%s", data) + var device common.Device + err = json.Unmarshal(data, &device) + if err != nil { + logx.Infof("unmarshal device update msg err :%s", err) + return err + } + + procedures := make([]string, 0, len(device.DeviceProcedureAttr)) + for _, attr := range device.DeviceProcedureAttr { + procedures = append(procedures, attr.ProcedureName) + } + + deviceRecord := &model.Device{ + DeviceID: device.ID, + Procedures: strings.Join(procedures, ","), + ExtChannelAmount: device.ExtChannelAmount, + } + + oldRecord, err := model.NewDeviceSearch().SetDeviceId(device.ID).First() + if err == gorm.ErrRecordNotFound { + err = model.NewDeviceSearch().Create(deviceRecord) + } else { + deviceRecord.ID = oldRecord.ID + err = model.NewDeviceSearch().Save(deviceRecord) + } + + if err != nil { + logx.Infof("save device record err :%s", err) + return err + } + + return nil +} -- Gitblit v1.8.0