debug

zhangqian

2023-10-20 a699ff979c84d024ca33fb1139224258f278319f

 nsq/msg_handler.go

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

@@ -43,41 +44,63 @@
      return err
   }
   for _, task := range tasks {
      oldWorkOrder, err := model.NewOrderSearch(nil).SetWorkOrderId(task.WorkOrder.WorkOrderID).First()
      if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" && oldWorkOrder.Status != model.OrderStatusWaitProcess { //已开始的工序不可以修改
         continue
      }

      procedureRecords := make([]*model.Procedures, 0, len(task.Procedures))
      for _, procedure := range task.Procedures {
         if procedure.DeviceID == conf.Conf.System.DeviceId { //只存储本设备的工序
            err = model.WithTransaction(func(db *gorm.DB) error {
               _, err := model.NewOrderSearch(db).SetWorkOrderId(task.WorkOrder.WorkOrderID).First()
               if err == gorm.ErrRecordNotFound {
                  var orderRecord model.Order
                  err = structx.AssignTo(task.WorkOrder, &orderRecord)
                  orderRecord.Status = model.OrderStatusUnFinished
                  if err != nil {
                     logx.Errorf(" structx.Assign task.Order to orderRecord err: %v", err.Error())
                     return err
                  }
                  if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
                     return err
                  }
               }
               procedureRecord := model.Procedures{
                  StartTime:   procedure.StartTime,
                  EndTime:     procedure.EndTime,
                  WorkOrderID: task.WorkOrder.WorkOrderID,
                  OrderID:     task.WorkOrder.OrderID,
                  Status:      model.ProcedureStatusUnFinished,
               }
               procedureData, err := json.Marshal(procedure)
               procedureRecord.ProcedureData = string(procedureData)
         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{},
         }
         procedureData, err := json.Marshal(procedure)
         if err != nil {
            return err
         }
         procedureRecord.ProcedureData = string(procedureData)
         procedureRecords = append(procedureRecords, &procedureRecord)
      }

      var orderRecord model.Order
      err = structx.AssignTo(task.WorkOrder, &orderRecord)
      orderRecord.Status = model.OrderStatusWaitProcess
      if err != nil {
         logx.Errorf(" structx.Assign task.Order to orderRecord err: %v", err.Error())
         return err
      }

      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).Upsert(procedure)
               if err != nil {
                  logx.Errorf(" json.Marshal(procedure) err: %v", err.Error())
                  return err
               }
               return model.NewProceduresSearch(db).Create(&procedureRecord)
            })
            if err != nil {
               logx.Errorf(" save procedure err: %v", err.Error())
            }
         } else {
            if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
               return err
            }
            return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
         }
         return nil
      })
      if err != nil {
         logx.Errorf(" save task message error err: %v", err)
         return err
      }
   }
   return nil
@@ -93,36 +116,226 @@
   err = json.Unmarshal(data, &resp)
   if err != nil {
      logx.Errorf("ScheduleTask HandleMessage Unmarshal json err: %v", err.Error())
      return err
   }
   //写入到文件
   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)
   if resp.DeviceId != conf.Conf.System.DeviceId {
      return nil
   }
   //通知回复收到
   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
}

type ProcessParams struct {
   Topic string
}

func (slf *ProcessParams) HandleMessage(data []byte) (err error) {
   logx.Infof("get an process params message :%s", data)
   var processModel model.ProcessModel
   err = json.Unmarshal(data, &processModel)
   if err != nil {
      logx.Infof("unmarshal process params sync err :%s", err)
      return err
   }
   if processModel.DeviceId != conf.Conf.System.DeviceId { //不是发给本设备的消息
      return nil
   }
   if processModel.Params != "" {
      err = model.NewProcessModelSearch().Create(&processModel)
      if err != nil {
         logx.Infof("save process params err :%s", err)
      }
   }
   //通知回复收到
   ReceivedMessageChan <- &ReceivedMessage{
      Topic:   slf.Topic,
      Message: data,
   }
   return nil
}

type ProcessParamsSync struct {
   Topic string
}

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)
   if err != nil {
      logx.Infof("unmarshal process params sync err :%s", err)
      return err
   }
   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
      }
      processModel.IsNew = true
      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 device common.Device
   err = json.Unmarshal(data, &device)
   if err != nil {
      logx.Infof("unmarshal device update msg err :%s", err)
      return err
   }

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

   deviceRecord := &model.Device{
      DeviceID:         device.ID,
      Procedures:       strings.Join(procedures, ","),
      ExtChannelAmount: device.ExtChannelAmount,
   }

   oldRecord, err := model.NewDeviceSearch().SetDeviceId(device.ID).First()
   if err == gorm.ErrRecordNotFound {
      err = model.NewDeviceSearch().Create(deviceRecord)
   } else {
      deviceRecord.ID = oldRecord.ID
      err = model.NewDeviceSearch().Save(deviceRecord)
   }

   if err != nil {
      logx.Infof("save device  record err :%s", err)
      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)

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