add unique index

zhangqian

2023-10-21 d9fffbd8032ef2083ae4d5d55ba62d66a460d824

 nsq/msg_handler.go

@@ -281,7 +281,7 @@
      return err
   }

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

@@ -297,8 +297,21 @@
   }
   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 {

      oldPlcRecord.Port = record.Port
      oldPlcRecord.Method = record.Method
      oldPlcRecord.Address = record.Address
      oldPlcRecord.Detail = record.Detail
      err = model.NewDevicePlcSearch().Save(oldPlcRecord)
   }
   if err != nil {
      return err
   }
@@ -371,34 +384,5 @@
}

func (slf *PullDataResponse) DealDeviceData(data interface{}) error {
   var devices []*model.Device
   err := mapstructure.Decode(data, &devices)
   if err != nil {
      return err
   }
   numbers := make([]string, 0, len(devices))
   for _, item := range devices {
      numbers = append(numbers, item.DeviceID)
   }
   existsDevices, err := model.NewDeviceSearch().SetDeviceIds(numbers).FindNotTotal()
   if err != nil {
      return err
   }

   existsDeviceMap := make(map[string]*model.Device, len(existsDevices))
   for _, device := range existsDevices {
      existsDeviceMap[device.DeviceID] = device
   }

   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)
      }
   }
   return err
   return nil
}