zhangqian
2023-09-15 e67802cd15e273da46bfb734e339a947b7d1eceb
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,10 @@
            WorkOrderID:    task.WorkOrder.WorkOrderID,
            OrderID:        task.WorkOrder.OrderID,
            DeviceID:       procedure.DeviceID,
            ProcedureID:    procedure.ProcedureID,
            StartTime:      procedure.StartTime,
            EndTime:        procedure.EndTime,
            Status:         model.ProcedureStatusWaitProcess,
            ProcedureData:  "",
            ProceduresInfo: common.ProductProcedure{},
         }
         procedureData, err := json.Marshal(procedure)
@@ -77,10 +78,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).Updates(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 +116,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,7 +218,7 @@
      logx.Infof("unmarshal process params sync err :%s", err)
      return err
   }
   err = model.NewProcessModelSearch().Create(&processModel)
   err = model.NewProcessModelSearch().Save(&processModel)
   if err != nil {
      logx.Infof("save process params sync err :%s", err)
      return err