From 84d2953e7b641600779e7aecc80bdd35e96d31d6 Mon Sep 17 00:00:00 2001 From: jiangshuai <291802688@qq.com> Date: 星期二, 06 二月 2024 10:13:12 +0800 Subject: [PATCH] 拦截器 --- service/purchase/purchase.go | 187 +++++++++++++++++++++++++++++++++------------- 1 files changed, 134 insertions(+), 53 deletions(-) diff --git a/service/purchase/purchase.go b/service/purchase/purchase.go index 0df0c5c..fb3db12 100644 --- a/service/purchase/purchase.go +++ b/service/purchase/purchase.go @@ -2,11 +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" ) @@ -22,19 +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 { - err = tx.Create(¶ms).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 @@ -59,6 +86,10 @@ //@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 { err = tx.Where("id = ?", params.ID).Updates(params).Error if err != nil { @@ -92,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 } @@ -121,7 +166,7 @@ func (slf *PurchaseService) GetPurchaseProductList(purchaseId uint) (list []*purchase.PurchaseProducts, err error) { db := global.GVA_DB.Model(&purchase.PurchaseProducts{}) list = make([]*purchase.PurchaseProducts, 0) - err = db.Where("purchase_id = ?", purchaseId).Find(&list).Error + err = db.Where("purchase_id = ?", purchaseId).Preload("Product").Find(&list).Error return list, err } @@ -130,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 @@ -169,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) @@ -178,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) @@ -203,15 +257,36 @@ } func (slf *PurchaseService) SavePurchaseType(list []*purchase.PurchaseType) (err error) { + ids := make([]uint, 0) + newRecords := make([]*purchase.PurchaseType, 0) + for _, item := range list { + if item.ID != 0 { + ids = append(ids, item.ID) + } else { + newRecords = append(newRecords, item) + } + } err = global.GVA_DB.Transaction(func(tx *gorm.DB) error { - err = tx.Where("1 = ?", 1).Delete(&purchase.PurchaseType{}).Error + err = tx.Where("id not in ?", ids).Delete(&purchase.PurchaseType{}).Error if err != nil { return err } - for _, item := range list { - item.ID = 0 + if len(newRecords) > 0 { + err = tx.Create(newRecords).Error + if err != nil { + return err + } } - return tx.Create(list).Error + + for _, item := range list { + if item.ID != 0 { + err = tx.Where("id = ?", item.ID).Updates(item).Error + if err != nil { + return err + } + } + } + return nil }) return err } @@ -222,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 +} -- Gitblit v1.8.0