zhangqian
2023-10-20 d74cd8ed4ce995294c7f65b02742d07d7a420474
nsq/msg_handler.go
@@ -82,19 +82,25 @@
         var err error
         if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" {
            if err = model.NewOrderSearch(db).SetId(oldWorkOrder.ID).Updates(&orderRecord); err != nil {
               logx.Errorf(" save task message error err1: %v", err)
               return err
            }
            for _, procedure := range procedureRecords {
               err = model.NewProceduresSearch(db).SetWorkOrderId(procedure.WorkOrderID).SetProcedureId(procedure.ProcedureID).Upsert(procedure)
               if err != nil {
                  logx.Errorf(" save task message error err2: %v", err)
                  return err
               }
            }
         } else {
            if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
               logx.Errorf(" save task message error err3: %v", err)
               return err
            }
            return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
            if err = model.NewProceduresSearch(db).CreateBatch(procedureRecords); err != nil {
               logx.Errorf(" save task message error err4: %v", err)
               return err
            }
         }
         return nil
      })
@@ -214,11 +220,18 @@
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
   }
   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).
@@ -226,7 +239,6 @@
      if err != nil {
         return err
      }
      processModel.IsNew = true
      return model.NewProcessModelSearch().SetOrm(db).Create(&processModel)
   })