fix

zhangqian

2023-12-01 8324f872ef3a4d0c978a9b1d062800c6a1701c12

 nsq/msg_handler.go

@@ -11,9 +11,9 @@
   "apsClient/utils/file"
   "encoding/json"
   "fmt"
   "github.com/jinzhu/gorm"
   "github.com/mitchellh/mapstructure"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "strings"
)

@@ -52,14 +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,
            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 {
@@ -78,12 +80,13 @@
      }

      err = model.WithTransaction(func(db *gorm.DB) error {
         var err error
         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).Updates(procedure)
               err = model.NewProceduresSearch(db).SetWorkOrderId(procedure.WorkOrderID).SetProcedureId(procedure.ProcedureID).Upsert(procedure)
               if err != nil {
                  return err
               }
@@ -92,12 +95,14 @@
            if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
               return err
            }
            return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
            if err = model.NewProceduresSearch(db).CreateBatch(procedureRecords); err != nil {
               return err
            }
         }
         return nil
      })
      if err != nil {
         logx.Errorf(" save task message error err: %v", err.Error())
         logx.Errorf(" save task message error err: %v", err)
         return err
      }
   }
@@ -156,6 +161,8 @@
      return nil
   }

   resp.PlcConfig.Method.ConvertToEnglish()

   var record model.DevicePlc
   err = mapstructure.Decode(resp.PlcConfig, &record)
   if err != nil {
@@ -167,8 +174,8 @@
   }
   detail, err := json.Marshal(record.Details)
   record.Detail = string(detail)
   record.Id = 1
   err = model.NewDevicePlcSearch().SetId(record.Id).Save(&record)
   record.ID = 1
   err = model.NewDevicePlcSearch().SetId(record.ID).Save(&record)
   if err != nil {
      return err
   }
@@ -210,16 +217,219 @@

func (slf *ProcessParamsSync) HandleMessage(data []byte) (err error) {
   logx.Infof("get an process params sync message :%s", data)
   var processModel model.ProcessModel
   err = json.Unmarshal(data, &processModel)
   var processModelMsg common.ResponseProcessParams
   err = json.Unmarshal(data, &processModelMsg)
   if err != nil {
      logx.Infof("unmarshal process params sync err :%s", err)
      return err
   }
   err = model.NewProcessModelSearch().Save(&processModel)
   processModel := model.ProcessModel{
      Number:    processModelMsg.Number,
      Product:   processModelMsg.Product,
      Procedure: processModelMsg.Procedure,
      Params:    processModelMsg.Params,
      IsNew:     true,
   }
   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
      }
      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 msg common.DeviceMsg
   err = json.Unmarshal(data, &msg)
   if err != nil {
      logx.Infof("unmarshal device update msg err :%s", err)
      return err
   }

   procedures := make([]string, 0, len(msg.DeviceProcedureAttr))
   for _, attr := range msg.DeviceProcedureAttr {
      procedures = append(procedures, attr.ProcedureName)
   }

   deviceRecord := &model.Device{
      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 ||
      oldRecord.DeviceMac != deviceRecord.DeviceMac ||
      oldRecord.DeviceName != deviceRecord.DeviceName {
      err = model.NewDeviceSearch().SetDeviceId(msg.ID).Save(deviceRecord)
   }

   if err != nil {
      logx.Infof("save device  record err :%s", err)
      return err
   }

   if msg.PlcConfig == nil || msg.PlcConfig.Method == "" {
      return nil
   }

   msg.PlcConfig.Method.ConvertToEnglish()
   var record model.DevicePlc
   err = mapstructure.Decode(msg.PlcConfig, &record)
   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.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
   }

   return nil
}

type PullDataResponse struct {
   Topic string
}

func (slf *PullDataResponse) HandleMessage(data []byte) (err error) {
   logx.Infof("get a pull data response message :%s", data)
   var pullDataResponse common.MsgPullDataResponse
   err = json.Unmarshal(data, &pullDataResponse)
   if err != nil {
      logx.Infof("unmarshal msg err :%s", err)
      return err
   }
   switch pullDataResponse.DataType {
   case common.PullDataTypeProcessModel:
      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)
      return err
   }
   return nil
}

func (slf *PullDataResponse) DealProcessModelData(data interface{}) error {
   var processModels []*model.ProcessModel
   err := mapstructure.Decode(data, &processModels)
   if err != nil {
      return err
   }
   numbers := make([]string, 0, len(processModels))
   for _, processModel := range processModels {
      numbers = append(numbers, processModel.Number)
   }
   existsProcessModels, err := model.NewProcessModelSearch().SetIsNew(true).SetNumbers(numbers).FindNotTotal()
   if err != nil {
      return err
   }

   existsProcessModelsMap := make(map[string]struct{}, len(existsProcessModels))
   for _, processModel := range existsProcessModels {
      existsProcessModelsMap[processModel.Number] = struct{}{}
   }

   for _, processModel := range processModels {
      if _, exists := existsProcessModelsMap[processModel.Number]; exists {
         continue
      }
      err = model.WithTransaction(func(db *gorm.DB) error {
         err = model.NewProcessModelSearch().SetOrm(db).SetProcedure(processModel.Procedure).SetProduct(processModel.Product).SetIsNew(true).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 {
         return err
      }
   }
   return nil
}

func (slf *PullDataResponse) DealDeviceData(data interface{}) error {
   //已在别的topic处理
   return nil
}

func (slf *PullDataResponse) DealProcessModelPlcAddressData(data interface{}) error {
   var addressList []*model.ProcessModelPlcAddress
   err := mapstructure.Decode(data, &addressList)
   if err != nil {
      return err
   }
   deviceIDs := make([]string, 0, len(addressList))
   for _, item := range addressList {
      deviceIDs = append(deviceIDs, item.DeviceID)
   }
   existsRecords, err := model.NewProcessModelPlcAddressSearch().SetDeviceIDs(deviceIDs).FindNotTotal()
   if err != nil {
      return err
   }

   existsRecordsMap := make(map[string]*model.ProcessModelPlcAddress, len(existsRecords))
   for _, item := range existsRecords {
      existsRecordsMap[item.DeviceID] = item
   }

   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 nil
}