新增共用id结构体

jiangshuai

2023-12-19 f6bf0ee81b080a2bad55aa5bf261c68ab3d555f8

 service/purchase/purchase.go

@@ -2,12 +2,13 @@

import (
   "context"
   "errors"
   "github.com/shopspring/decimal"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "srm/global"
   "srm/model/common/request"
   "srm/model/purchase"
   purchaserequest "srm/model/purchase/request"
   "srm/proto/qualityinspect"
   "srm/service/test"
)
@@ -23,24 +24,44 @@
//@param: params *purchaserequest.AddPurchase
//@return: err error

func (slf *PurchaseService) CreatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) {
func (slf *PurchaseService) CreatePurchase(record *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) {
   err = DealPrice(record, productList)
   if err != nil {
      return err
   }
   err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
      var quantity decimal.Decimal
      for _, product := range productList {
         quantity = quantity.Add(product.Amount)
      }
      params.Quantity = quantity
      err = tx.Create(&params).Error
      err = tx.Create(&record).Error
      if err != nil {
         return err
      }
      for _, product := range productList {
         product.PurchaseId = cast.ToInt(params.ID)
         product.PurchaseId = cast.ToInt(record.ID)
      }
      return tx.Create(productList).Error
   })

   return err
}

func DealPrice(record *purchase.Purchase, productList []*purchase.PurchaseProducts) error {
   var quantity decimal.Decimal
   var totalPrice decimal.Decimal
   var realTotalPrice decimal.Decimal
   for _, product := range productList {
      quantity = quantity.Add(product.Amount)
      totalPrice = totalPrice.Add(product.Price.Mul(product.Amount))
   }
   record.Quantity = quantity
   if !totalPrice.Equal(record.TotalPrice) {
      return errors.New("价税总计计算错误")
   }
   realTotalPrice = record.CalcRealTotalPrice()
   if !realTotalPrice.Equal(record.RealTotalPrice) {
      return errors.New("最终价格计算错误")
   }
   record.UnInvoiceAmount = record.RealTotalPrice
   record.ShouldPayAmount = record.RealTotalPrice
   return nil
}

//@function: DeletePurchase
@@ -65,12 +86,11 @@
//@return: err error

func (slf *PurchaseService) UpdatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) {
   err = DealPrice(params, productList)
   if err != nil {
      return err
   }
   err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
      var quantity decimal.Decimal
      for _, product := range productList {
         quantity = quantity.Add(product.Amount)
      }
      params.Quantity = quantity
      err = tx.Where("id = ?", params.ID).Updates(params).Error
      if err != nil {
         return err
@@ -103,24 +123,38 @@
//@param: info request.PageInfo
//@return: list interface{}, total int64, err error

func (slf *PurchaseService) GetPurchaseList(info request.PageInfo) (list interface{}, total int64, err error) {
func (slf *PurchaseService) GetPurchaseList(info purchaserequest.PurchaseSearch) (list interface{}, total int64, err error) {
   limit := info.PageSize
   offset := info.PageSize * (info.Page - 1)
   db := global.GVA_DB.Model(&purchase.Purchase{})
   var ids []uint
   var purchaseList = make([]*purchase.Purchase, 0)
   if info.Keyword != "" {
      db.Distinct("purchases.id").Joins("left join purchase_products on purchase_products.purchase_id = purchases.id").
         Joins("left join Product on Product.Id = purchase_products.product_id").
         Joins("left join supplier on supplier.Id = purchases.supplier_id").
         Where("purchases.name like ?", "%"+info.Keyword+"%").
         Or("Product.name like ?", "%"+info.Keyword+"%").
         Or("supplier.name like ?", "%"+info.Keyword+"%")
      db.Distinct("srm_purchase.id").Joins("left join srm_purchase_products on srm_purchase_products.purchase_id = srm_purchase.id").
         Joins("left join srm_supplier_material on srm_supplier_material.supplier_id = srm_purchase.id").
         Joins("left join srm_supplier on srm_supplier.Id = srm_purchase.supplier_id").
         Where("srm_purchase.name like ?", "%"+info.Keyword+"%").
         Or("srm_supplier_material.name like ?", "%"+info.Keyword+"%").
         Or("srm_supplier.name like ?", "%"+info.Keyword+"%")
      err = db.Limit(limit).Offset(offset).Find(&ids).Error
      if err != nil {
         return purchaseList, total, err
      }
   } else if info.SupplierId != 0 {
      db = db.Where("supplier_id = ?", info.SupplierId)
   }
   var purchaseList []purchase.Purchase
   err = db.Count(&total).Error
   if err != nil {
   if err != nil || total == 0 {
      return purchaseList, total, err
   }
   err = db.Limit(limit).Offset(offset).Find(&purchaseList).Error
   if len(ids) != 0 {
      db = global.GVA_DB.Model(&purchase.Purchase{})
      err = db.Where("id in (?)", ids).Preload("Supplier").Order("updated_at desc").Find(&purchaseList).Error
   } else {
      //db = global.GVA_DB.Model(&purchase.Purchase{})
      err = db.Limit(limit).Offset(offset).Preload("Supplier").Order("updated_at desc").Find(&purchaseList).Error
   }

   return purchaseList, total, err
}

@@ -141,33 +175,38 @@
//@param: id uint
//@return: err error

func (slf *PurchaseService) Submit(id uint) (err error) {
func (slf *PurchaseService) Submit(id int, status purchase.OrderStatus, warehouse string) (err error) {

   purchaseData, err := slf.GetPurchase(id)
   if err != nil {
      return err
   }
   var targetStatus purchase.OrderStatus
   switch purchaseData.Status {
   case purchase.OrderStatusConfirmed:
      targetStatus = purchase.OrderStatusReceived
   case purchase.OrderStatusReceived:
      targetStatus = purchase.OrderStatusStored
   case purchase.OrderStatusStored:
      targetStatus = purchase.OrderStatusCompleted
   }
   //purchaseData, err := slf.GetPurchase(id)
   //if err != nil {
   //   return err
   //}
   //var targetStatus purchase.OrderStatus
   //switch purchaseData.Status {
   //case purchase.OrderStatusConfirmed:
   //   targetStatus = purchase.OrderStatusReceived
   //case purchase.OrderStatusReceived:
   //   targetStatus = purchase.OrderStatusStored
   //case purchase.OrderStatusStored:
   //   targetStatus = purchase.OrderStatusCompleted
   //}
   err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
      err = tx.Where("id = ?", id).Model(&purchase.Purchase{}).Updates(map[string]interface{}{"status": targetStatus}).Error
      m := make(map[string]interface{})
      m["status"] = status
      if warehouse != "" {
         m["warehouse"] = warehouse
      }
      err = tx.Where("id = ?", id).Model(&purchase.Purchase{}).Updates(m).Error
      if err != nil {
         return err
      }

      switch targetStatus {
      case purchase.OrderStatusReceived:
         return SendInspect(purchaseData)
      case purchase.OrderStatusStored:
      case purchase.OrderStatusCompleted:
      }
      //switch targetStatus {
      //case purchase.OrderStatusReceived:
      //   return SendInspect(purchaseData)
      //case purchase.OrderStatusStored:
      //case purchase.OrderStatusCompleted:
      //}
      return nil
   })
   return err
@@ -180,7 +219,7 @@
   }
   productIds := make([]uint, 0, len(productList))
   for _, product := range productList {
      productIds = append(productIds, uint(product.ProductId))
      productIds = append(productIds, product.ID)
   }
   productService := &test.ProductService{}
   _, productMap, err := productService.GetProducts(productIds)
@@ -189,24 +228,28 @@
   }
   inspectOrders := make([]*qualityinspect.QualityInspect, 0, len(productList))
   for _, productItem := range productList {
      product := productMap[uint(productItem.ProductId)]
      product := productMap[productItem.ID]
      if product == nil {
         continue
      }
      inspectOrder := &qualityinspect.QualityInspect{
         InspectType:   qualityinspect.InspectType_InspectTypePurchase,
         MaterialType:  qualityinspect.MaterialType_MaterialTypeRaw,
         MaterialName:  product.Name,
         MaterialId:    product.Number,
         MaterialTp:    product.ModelNumber,
         MaterialUnit:  product.Unit,
         Supplier:      record.Supplier.Name,
         WarehouseName: "采购总仓",
         ReportAmount:  productItem.Amount.InexactFloat64(),
         InspectMethod: qualityinspect.InspectMethod_InspectMethodAll,
         InspectAmount: productItem.Amount.InexactFloat64(),
         InspectType:     qualityinspect.InspectType_InspectTypePurchase,
         MaterialType:    qualityinspect.MaterialType_MaterialTypeRaw,
         MaterialName:    product.Name,
         MaterialId:      product.Number,
         MaterialTp:      product.ModelNumber,
         MaterialUnit:    product.Unit,
         Supplier:        record.Supplier.Name,
         WarehouseName:   "采购总仓",
         ReportAmount:    productItem.Amount.InexactFloat64(),
         InspectMethod:   qualityinspect.InspectMethod_InspectMethodAll,
         InspectAmount:   productItem.Amount.InexactFloat64(),
         PurchaseOrderId: record.Number,
      }
      inspectOrders = append(inspectOrders, inspectOrder)
   }
   if len(inspectOrders) == 0 {
      return nil
   }
   inspectRequest := qualityinspect.SendPurchaseInspectRequest{List: inspectOrders}
   _, err = qualityinspect.NewQualityInspectServiceClient(qualityinspect.Conn).SendPurchaseInspect(context.Background(), &inspectRequest)
@@ -237,7 +280,7 @@

      for _, item := range list {
         if item.ID != 0 {
            err = tx.Save(item).Error
            err = tx.Where("id = ?", item.ID).Updates(item).Error
            if err != nil {
               return err
            }
@@ -254,3 +297,9 @@
   err = db.Order("pin desc, sort desc, id asc").Find(&list).Error
   return list, err
}

func (slf *PurchaseService) MaxAutoIncr() (int, error) {
   var total int64
   err := global.GVA_DB.Model(&purchase.Purchase{}).Count(&total).Error
   return int(total), err
}