pg_cancel_dividends_estimate_service.go 12.7 KB
package domain_service

import (
	"fmt"
	"github.com/linmadan/egglib-go/core/application"
	coreDomain "github.com/linmadan/egglib-go/core/domain"
	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/repository"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/utils"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log"
)

type CancelDividendsEstimateService struct {
	coreDomain.BaseEventPublisher
	transactionContext *pgTransaction.TransactionContext
}

func (domainService *CancelDividendsEstimateService) CancelEstimate(dividendsEstimateIds []int64, operator *domain.User) ([]*domain.DividendsEstimate, error) {
	var dividendsEstimateRepository domain.DividendsEstimateRepository           // 分红预算单仓储
	var dividendsOrderRepository domain.DividendsOrderRepository                 // 分红订单仓储
	var dividendsReturnedOrderRepository domain.DividendsReturnedOrderRepository // 分红退货单仓储
	var orderGoodRepository domain.OrderGoodRepository                           // 订单产品仓储

	type orderGoodInfo struct {
		OrderGoodId int64
		CompanyId   int64
		OrgId       int64
	}

	// 分红预算单仓储初始化
	if repo, err := repository.NewDividendsEstimateRepository(domainService.transactionContext); err != nil {
		return nil, err
	} else {
		dividendsEstimateRepository = repo
	}

	// 分红订单仓储初始化
	if repo, err := repository.NewDividendsOrderRepository(domainService.transactionContext); err != nil {
		return nil, err
	} else {
		dividendsOrderRepository = repo
	}

	// 分红退货单仓储初始化
	if repo, err := repository.NewDividendsReturnedOrderRepository(domainService.transactionContext); err != nil {
		return nil, err
	} else {
		dividendsReturnedOrderRepository = repo
	}

	// 订单产品仓储初始化
	if repo, err := repository.NewOrderGoodRepository(domainService.transactionContext); err != nil {
		return nil, err
	} else {
		orderGoodRepository = repo
	}

	// 获取分红预算单
	if count, dividendsEstimates, err := dividendsEstimateRepository.Find(map[string]interface{}{
		"dividendsEstimateIds": dividendsEstimateIds,
		"offsetLimit":          false,
	}); err != nil {
		return nil, err
	} else {
		if count > 0 {
			var dividendsEstimatesCanceled []*domain.DividendsEstimate
			var orderOrReturnOrderNums []string
			var orderGoodIds []int64
			orderGoodMaps := make(map[string][]int64, 0)
			for _, dividendsEstimate := range dividendsEstimates {
				if dividendsEstimate.DividendsType == 1 || dividendsEstimate.DividendsType == 2 { //  1.订单分红 2.退货冲销
					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(map[string]interface{}{
						"companyId":                 dividendsEstimate.Company.CompanyId,
						"orgId":                     dividendsEstimate.Org.OrgId,
						"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
						"dividendsParticipateTypes": []int32{1, 2, 3},
						"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
						"offsetLimit":               false,
						"dividendsStage":            dividendsEstimate.DividendsStage,
						"dividendsTypes":            []int32{1, 2},
						"orderGoodId":               dividendsEstimate.OrderGoodId,
					}); err2 != nil {
						return nil, err2
					} else {
						if countRelative > 0 {
							for i, _ := range dividendsEstimatesRelative {
								dividendsEstimatesRelative[i].IsCanceled = true
								dividendsEstimatesRelative[i].Operator = operator
								//	获取相关订单/退货单号
								orderOrReturnOrderNums = append(orderOrReturnOrderNums, dividendsEstimatesRelative[i].OrderOrReturnedOrderNum)
								orderGoodIds = append(orderGoodIds, dividendsEstimatesRelative[i].OrderGoodId)
								orderGoodMaps[dividendsEstimatesRelative[i].OrderOrReturnedOrderNum] = append(orderGoodMaps[dividendsEstimatesRelative[i].OrderOrReturnedOrderNum], dividendsEstimatesRelative[i].OrderGoodId)
							}
							//dividendsEstimatesRelativeCanceled, err3 := dividendsEstimateRepository.BatchRemove(dividendsEstimatesRelative)
							dividendsEstimatesRelativeCanceled, err3 := dividendsEstimateRepository.UpdateMany(dividendsEstimatesRelative)
							if err3 != nil {
								return nil, err3
							}
							dividendsEstimatesCanceled = append(dividendsEstimatesCanceled, dividendsEstimatesRelativeCanceled...)
						}
					}
				} else if dividendsEstimate.DividendsType == 3 { // 3.金额激励
					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(map[string]interface{}{
						"companyId":                       dividendsEstimate.Company.CompanyId,
						"orgId":                           dividendsEstimate.Org.OrgId,
						"cooperationContractNumber":       dividendsEstimate.CooperationContractNumber,
						"cooperationContractUndertakerId": dividendsEstimate.CooperationContractUndertakerId,
						"dividendsParticipateTypes":       []int32{1, 2, 3},
						"offsetLimit":                     false,
						"dividendsStage":                  dividendsEstimate.DividendsStage,
						"dividendsType":                   int32(3),
					}); err2 != nil {
						return nil, err2
					} else {
						if countRelative > 0 {
							for i, _ := range dividendsEstimatesRelative {
								dividendsEstimatesRelative[i].IsCanceled = true
								dividendsEstimatesRelative[i].Operator = operator
							}
							//dividendsEstimatesRelativeCanceled, err4 := dividendsEstimateRepository.BatchRemove(dividendsEstimatesRelative)
							dividendsEstimatesRelativeCanceled, err4 := dividendsEstimateRepository.UpdateMany(dividendsEstimatesRelative)
							if err4 != nil {
								return nil, err4
							}
							dividendsEstimatesCanceled = append(dividendsEstimatesCanceled, dividendsEstimatesRelativeCanceled...)
						}
					}
				}
			}

			// 分红订单
			var orderNums []string

			// 分红退货单
			var returnedOrderNums []string

			if len(orderOrReturnOrderNums) > 0 { // 业绩分红
				// 订单/退货单号数组去重
				orderOrReturnOrderNumsWithoutDup := utils.RemoveDuplication(orderOrReturnOrderNums)

				// 分离订单和退货单
				for _, orderNullNum := range orderOrReturnOrderNumsWithoutDup {
					if len(orderNullNum) > 0 {
						if orderNullNum[:2] == "RE" { // 分红退后单
							returnedOrderNums = append(returnedOrderNums, orderNullNum)
						} else if orderNullNum[:2] == "SL" { // 分红订单
							orderNums = append(orderNums, orderNullNum)
						}
					}
				}

				// 获取分红订单并更新分红订单产品的状态
				if len(orderNums) > 0 {
					if countDividendsOrder, orders, err4 := dividendsOrderRepository.Find(map[string]interface{}{
						"companyId":             dividendsEstimates[0].Company.CompanyId,
						"orgId":                 dividendsEstimates[0].Org.OrgId,
						"dividendsOrderNumbers": orderNums,
						"offsetLimit":           false,
					}); err4 != nil {
						return nil, err4
					} else {
						if countDividendsOrder > 0 {
							for i, order := range orders {
								// 取消分红的订单产品
								newOrderGoodIds := utils.RemoveDuplicationInt64(orderGoodMaps[order.DividendsOrderNumber])
								var goodIds []int64
								for _, good := range order.Goods {
									if good.OrderGoodDividendsStatus == 2 {
										goodIds = append(goodIds, good.OrderGoodId)
									}
								}
								//  goodIds去重
								newGoodIds := utils.RemoveDuplicationInt64(goodIds)
								log.Logger.Info("")
								if len(newGoodIds) == len(newOrderGoodIds) { // 1.已分红的产品数量等于取消的产品数量,订单分红状态变更为未分红
									orders[i].DividendStatus = 1
								} else if len(newGoodIds) > len(newOrderGoodIds) { // 3.已分红的产品数量大于被取消分红的产品数量,订单分红状态变更为部分分红
									orders[i].DividendStatus = 3
								}
								orders[i].DividendStatus = order.DividendStatus
							}

							//	保存订单状态
							if _, err6 := dividendsOrderRepository.UpdateMany(orders); err6 != nil {
								return nil, err6
							}

							// 获取订单编号
							ordersNumbers := make([]string, 0)
							for _, order := range orders {
								ordersNumbers = append(ordersNumbers, order.DividendsOrderNumber)
							}

							// 获取分红订单产品并更新产品状态
							if len(ordersNumbers) > 0 {
								if countOrderGood, orderGoodsFound, err := orderGoodRepository.Find(map[string]interface{}{
									"orderGoodIds": orderGoodIds,
									"companyId":    dividendsEstimates[0].Company.CompanyId,
									"orgId":        dividendsEstimates[0].Org.OrgId,
									"offsetLimit":  false,
								}); err != nil {
									return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
								} else {
									if countOrderGood > 0 {
										for i, _ := range orderGoodsFound {
											orderGoodsFound[i].OrderGoodDividendsStatus = int32(1) // 待分红
										}
										if _, err6 := orderGoodRepository.UpdateMany(orderGoodsFound); err6 != nil {
											return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err6.Error())
										}
									}
								}
							}
						}
					}
				}

				// 获取分红退货单并更新订单产品的状态
				if len(returnedOrderNums) > 0 {
					if countDividendsReturnedOrder, returnedOrders, err5 := dividendsReturnedOrderRepository.Find(map[string]interface{}{
						"companyId":                     dividendsEstimates[0].Company.CompanyId,
						"orgId":                         dividendsEstimates[0].Org.OrgId,
						"dividendsReturnedOrderNumbers": returnedOrderNums,
						"offsetLimit":                   false,
					}); err5 != nil {
						return nil, err5
					} else {
						if countDividendsReturnedOrder > 0 {
							for i, returnedOrder := range returnedOrders {
								// 已取消的退货单产品
								newReturnedOrderGoodIds := utils.RemoveDuplicationInt64(orderGoodMaps[returnedOrder.DividendsReturnedOrderNumber])
								var goodIds []int64
								for _, good := range returnedOrder.Goods {
									if good.OrderGoodDividendsStatus == 2 {
										goodIds = append(goodIds, good.OrderGoodId)
									}
								}
								// goodIds去重
								newGoodIds := utils.RemoveDuplicationInt64(goodIds)
								if len(newGoodIds) == len(newReturnedOrderGoodIds) { // 1.已分红的产品数量等于取消的产品数量,退货单分红状态变更为未分红
									returnedOrders[i].DividendStatus = 1
								} else if len(newGoodIds) > len(newReturnedOrderGoodIds) { // 3.已分红的产品数量大于被取消分红的产品数量,退货单分红状态变更为部分分红
									returnedOrders[i].DividendStatus = 3
								}
								returnedOrders[i].DividendStatus = returnedOrder.DividendStatus
							}

							//	保存退货单状态
							if _, err7 := dividendsReturnedOrderRepository.UpdateMany(returnedOrders); err7 != nil {
								return nil, err7
							}

							// 获取订单编号
							returnedOrdersNumbers := make([]string, 0)
							for _, returnedOrder := range returnedOrders {
								returnedOrdersNumbers = append(returnedOrdersNumbers, returnedOrder.DividendsReturnedOrderNumber)
							}

							// 获取退货单产品并更新产品状态
							if len(returnedOrdersNumbers) > 0 {
								if countOrderGood, orderGoodsFound, err := orderGoodRepository.Find(map[string]interface{}{
									"orderGoodIds": orderGoodIds,
									"companyId":    dividendsEstimates[0].Company.CompanyId,
									"orgId":        dividendsEstimates[0].Org.OrgId,
									"offsetLimit":  false,
								}); err != nil {
									return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
								} else {
									if countOrderGood > 0 {
										for i, _ := range orderGoodsFound {
											orderGoodsFound[i].OrderGoodDividendsStatus = int32(1) // 待分红
										}
										if _, err6 := orderGoodRepository.UpdateMany(orderGoodsFound); err6 != nil {
											return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err6.Error())
										}
									}
								}
							}
						}
					}
				}
			}
			return dividendsEstimatesCanceled, nil
		}
	}
	return nil, nil
}

func NewCancelDividendsEstimateService(transactionContext *pgTransaction.TransactionContext) (*CancelDividendsEstimateService, error) {
	if transactionContext == nil {
		return nil, fmt.Errorf("transactionContext参数不能为nil")
	} else {
		return &CancelDividendsEstimateService{
			transactionContext: transactionContext,
		}, nil
	}
}