From cacf363f222e354cad6db18c4a00fb1097d35c2c Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 09 十一月 2023 14:30:58 +0800 Subject: [PATCH] fix --- nsq/msg_handler.go | 82 ++++++++++++++++++++++++++++------------- 1 files changed, 56 insertions(+), 26 deletions(-) diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go index 1ad713a..82d2118 100644 --- a/nsq/msg_handler.go +++ b/nsq/msg_handler.go @@ -267,12 +267,17 @@ DeviceID: msg.ID, Procedures: strings.Join(procedures, ","), ExtChannelAmount: msg.ExtChannelAmount, + DeviceMac: msg.DeviceMac, + DeviceName: msg.DeviceName, } oldRecord, err := model.NewDeviceSearch().SetDeviceId(msg.ID).First() if err == gorm.ErrRecordNotFound { err = model.NewDeviceSearch().Create(deviceRecord) - } else if oldRecord.ExtChannelAmount != deviceRecord.ExtChannelAmount || oldRecord.Procedures != deviceRecord.Procedures { + } else if oldRecord.ExtChannelAmount != deviceRecord.ExtChannelAmount || + oldRecord.Procedures != deviceRecord.Procedures || + oldRecord.DeviceMac != deviceRecord.DeviceMac || + oldRecord.DeviceName != deviceRecord.DeviceName { err = model.NewDeviceSearch().SetDeviceId(msg.ID).Save(deviceRecord) } @@ -281,7 +286,7 @@ return err } - if msg.PlcConfig.Method == "" { + if msg.PlcConfig == nil || msg.PlcConfig.Method == "" { return nil } @@ -291,14 +296,33 @@ if err != nil { logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err) } - err = mapstructure.Decode(msg.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) + record.DeviceID = msg.ID + oldPlcRecord, err := model.NewDevicePlcSearch().SetDeviceId(msg.ID).First() + if err == gorm.ErrRecordNotFound { + err = model.NewDevicePlcSearch().Create(&record) + } else if record.Port != oldPlcRecord.Port || record.Method != oldPlcRecord.Method || + record.Address != oldPlcRecord.Address || + record.Detail != oldPlcRecord.Detail || + record.BaudRate != oldPlcRecord.BaudRate || + record.SerialName != oldPlcRecord.SerialName || + record.DataBit != oldPlcRecord.DataBit || + record.StopBit != oldPlcRecord.StopBit || + record.Parity != oldPlcRecord.Parity { + + oldPlcRecord.Method = record.Method + oldPlcRecord.Address = record.Address + oldPlcRecord.Port = record.Port + oldPlcRecord.Detail = record.Detail + oldPlcRecord.BaudRate = record.BaudRate + oldPlcRecord.SerialName = record.SerialName + oldPlcRecord.DataBit = record.DataBit + oldPlcRecord.StopBit = record.StopBit + oldPlcRecord.Parity = record.Parity + + err = model.NewDevicePlcSearch().Save(oldPlcRecord) + } if err != nil { return err } @@ -323,6 +347,8 @@ err = slf.DealProcessModelData(pullDataResponse.Data) case common.PullDataTypeDevice: err = slf.DealDeviceData(pullDataResponse.Data) + case common.PullDataTypeProcessModelPlcAddress: + err = slf.DealProcessModelPlcAddressData(pullDataResponse.Data) } if err != nil { logx.Infof("process pull data err :%s", err) @@ -371,34 +397,38 @@ } func (slf *PullDataResponse) DealDeviceData(data interface{}) error { - var devices []*model.Device - err := mapstructure.Decode(data, &devices) + //宸插湪鍒殑topic澶勭悊 + return nil +} + +func (slf *PullDataResponse) DealProcessModelPlcAddressData(data interface{}) error { + var addressList []*model.ProcessModelPlcAddress + err := mapstructure.Decode(data, &addressList) if err != nil { return err } - numbers := make([]string, 0, len(devices)) - for _, item := range devices { - numbers = append(numbers, item.DeviceID) + deviceIDs := make([]string, 0, len(addressList)) + for _, item := range addressList { + deviceIDs = append(deviceIDs, item.DeviceID) } - existsDevices, err := model.NewDeviceSearch().SetDeviceIds(numbers).FindNotTotal() + existsRecords, err := model.NewProcessModelPlcAddressSearch().SetDeviceIDs(deviceIDs).FindNotTotal() if err != nil { return err } - existsDeviceMap := make(map[string]*model.Device, len(existsDevices)) - for _, device := range existsDevices { - existsDeviceMap[device.DeviceID] = device + existsRecordsMap := make(map[string]*model.ProcessModelPlcAddress, len(existsRecords)) + for _, item := range existsRecords { + existsRecordsMap[item.DeviceID] = item } - for _, device := range devices { - if oldDevice, exists := existsDeviceMap[device.DeviceID]; exists { - if oldDevice.ExtChannelAmount != device.ExtChannelAmount || //todo to be continued - oldDevice.Procedures != device.Procedures { - err = model.NewDeviceSearch().SetDeviceId(device.DeviceID).Save(device) - } - } else { - err = model.NewDeviceSearch().Create(device) + for _, item := range addressList { + if v, exists := existsRecordsMap[item.DeviceID]; !exists { + item.ID = 0 + err = model.NewProcessModelPlcAddressSearch().Create(item) + } else if v.Address != item.Address { + v.Address = item.Address + err = model.NewProcessModelPlcAddressSearch().Save(v) } } - return err + return nil } -- Gitblit v1.8.0