fix
wangpengfei
2023-08-28 c5a0eb549cba2cd358a2d0496c44f3a289f15d9c
service/salesReturn.go
@@ -32,11 +32,11 @@
      return ecode.SalesReturnNotExist
   }
   err = model.WithTransaction(func(db *gorm.DB) error {
      err = model.NewSalesReturnSearch().SetId(id).Delete()
      err = model.NewSalesReturnSearch().SetOrm(db).SetId(id).Delete()
      if err != nil {
         return err
      }
      err = model.NewSalesRefundSearch().SetSourceType(constvar.RefundSourceTypeSalesReturn).SetSourceId(id).Delete()
      err = model.NewSalesRefundSearch().SetOrm(db).SetSourceType(constvar.RefundSourceTypeSalesReturn).SetSourceId(id).Delete()
      if err != nil {
         return err
      }
@@ -73,9 +73,51 @@
   }
   salesReturn.AmountShouldRefund = salesReturn.AmountTotal
   err = model.NewSalesReturnSearch().SetId(salesReturn.Id).Update(salesReturn)
   old, err := model.NewSalesReturnSearch().SetId(salesReturn.Id).SetPreload(true).First()
   if err != nil {
      return ecode.SalesReturnSetErr
      return ecode.DBErr
   }
   newProducts, removedProducts := NewProductsService().PickDiffProducts(salesReturn.Products, old.Products)
   err = model.WithTransaction(func(db *gorm.DB) error {
      err := model.NewSalesReturnSearch().SetId(salesReturn.Id).Update(salesReturn)
      if err != nil {
         return err
      }
      if len(removedProducts) > 0 {
         removedProductIds := make([]uint, 0, len(removedProducts))
         for _, product := range removedProducts {
            removedProductIds = append(removedProductIds, product.Id)
         }
         err = model.NewProductSearch(db).SetIds(removedProductIds).Delete()
         if err != nil {
            return err
         }
      }
      if len(newProducts) > 0 {
         for _, p := range newProducts {
            p.Id = 0
         }
         err = model.NewProductSearch(db).CreateBatch(newProducts)
         if err != nil {
            return err
         }
         var rel []*model.SalesReturnProduct
         for _, p := range newProducts {
            rel = append(rel, &model.SalesReturnProduct{
               SalesReturnId: salesReturn.Id,
               ProductId:     p.Id,
            })
         }
         err = model.NewSalesReturnProductSearch().CreateBatch(rel)
         if err != nil {
            return err
         }
      }
      return nil
   })
   if err != nil {
      return ecode.DBErr
   }
   return ecode.OK