适应性改动

zhangqian

2023-09-13 0457f96716de8125560f332fcf718100b9f754eb

 nsq/msg_handler.go

@@ -49,7 +49,6 @@
      }

      procedureRecords := make([]*model.Procedures, 0, len(task.Procedures))
      procedureIds := make([]string, 0, len(task.Procedures))
      for _, procedure := range task.Procedures {
         procedureRecord := model.Procedures{
            WorkOrderID:    task.WorkOrder.WorkOrderID,
@@ -67,7 +66,6 @@
         }
         procedureRecord.ProcedureData = string(procedureData)
         procedureRecords = append(procedureRecords, &procedureRecord)
         procedureIds = append(procedureIds, procedure.ProcedureID)
      }

      var orderRecord model.Order
@@ -79,16 +77,23 @@
      }

      err = model.WithTransaction(func(db *gorm.DB) error {
         if err = model.NewOrderSearch(db).SetWorkOrderId(task.WorkOrder.WorkOrderID).Delete(); 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)
         }
         if err = model.NewOrderSearch(db).Save(&orderRecord); err != nil {
            return err
         }
         if err = model.NewProceduresSearch(db).SetProcedureIds(procedureIds).Delete(); err != nil {
            return err
         }
         return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
         return nil
      })
      if err != nil {
         logx.Errorf(" save task message error err: %v", err.Error())