From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 01 十二月 2023 09:58:17 +0800
Subject: [PATCH] fix

---
 nsq/msg_handler.go |  101 +++++++++++++++++++++++++++++++++-----------------
 1 files changed, 66 insertions(+), 35 deletions(-)

diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go
index 1ad713a..a45aed8 100644
--- a/nsq/msg_handler.go
+++ b/nsq/msg_handler.go
@@ -52,15 +52,16 @@
 		procedureRecords := make([]*model.Procedures, 0, len(task.Procedures))
 		for _, procedure := range task.Procedures {
 			procedureRecord := model.Procedures{
-				WorkOrderID:    task.WorkOrder.WorkOrderID,
-				OrderID:        task.WorkOrder.OrderID,
-				DeviceID:       procedure.DeviceID,
-				ProcedureID:    procedure.ProcedureID,
-				StartTime:      procedure.StartTime,
-				EndTime:        procedure.EndTime,
-				Status:         model.ProcedureStatusWaitProcess,
-				Channel:        procedure.Channel,
-				ProceduresInfo: common.ProductProcedure{},
+				ProductProcedureID: procedure.ProductProcedureID,
+				WorkOrderID:        task.WorkOrder.WorkOrderID,
+				OrderID:            task.WorkOrder.OrderID,
+				DeviceID:           procedure.DeviceID,
+				ProcedureID:        procedure.ProcedureID,
+				StartTime:          procedure.StartTime,
+				EndTime:            procedure.EndTime,
+				Status:             model.ProcedureStatusWaitProcess,
+				Channel:            procedure.Channel,
+				ProceduresInfo:     common.ProductProcedure{},
 			}
 			procedureData, err := json.Marshal(procedure)
 			if err != nil {
@@ -267,12 +268,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 +287,7 @@
 		return err
 	}
 
-	if msg.PlcConfig.Method == "" {
+	if msg.PlcConfig == nil || msg.PlcConfig.Method == "" {
 		return nil
 	}
 
@@ -291,14 +297,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 +348,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 +398,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