fix:修复分红预算单生成

陈志颖
pkg/application/dividendsEstimate/service/dividends_estimate.go
pkg/infrastructure/dao/pg_dividends_estimate_dao.go
pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
pkg/infrastructure/repository/pg_cooperation_contract_repository.go
--- a/pkg/application/dividendsEstimate/service/dividends_estimate.go
查看文件 @3866ab9
+++ b/pkg/application/dividendsEstimate/service/dividends_estimate.go
查看文件 @3866ab9
@@ -619,6 +619,15 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmDividendsIncent
 		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 	} else {
 		var dividendsEstimates []*domain.DividendsEstimate
+		// 统计当前分红预算单数
+		count, err := dividendsEstimateDao.CountDividendsEstimate(map[string]interface{}{
+			"companyId": confirmDividendsIncentivesEstimateCommand.CompanyId,
+			"orgId":     confirmDividendsIncentivesEstimateCommand.OrgId,
+		})
+		if err != nil {
+			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
+		}
+		var countDividendsEstimate int64
 		for _, orderGood := range orderGoods {
 			dividendsEstimate := &domain.DividendsEstimate{}
 			if orderGood.DividendsOrderNumber != "" { // 查询分红订单
@@ -637,18 +646,13 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmDividendsIncent
 				if dividendsEstimateDetails, err := confirmDividendsIncentivesEstimateService.Confirm(orderGoods); err != nil {
 					return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 				} else {
-					// 统计当前分红预算单数
-					count, err := dividendsEstimateDao.CountDividendsEstimate(map[string]interface{}{})
-					if err != nil {
-						return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
-					}
-
-					for i, dividendsEstimateDetail := range dividendsEstimateDetails {
+					for _, dividendsEstimateDetail := range dividendsEstimateDetails {
 						//  生成分红预算单号
-						dividendsEstimateOrderNumber, err := dividendsEstimateDetail.GenerateSpecificDividendsEstimateNumber(int64(count), int64(i))
+						dividendsEstimateOrderNumber, err := dividendsEstimateDetail.GenerateSpecificDividendsEstimateNumber(int64(count), countDividendsEstimate)
 						if err != nil {
 							return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
 						}
+						countDividendsEstimate = countDividendsEstimate + 1
 						dividendsEstimate = &domain.DividendsEstimate{
 							DividendsEstimateId:          0,
 							DividendsAccountStatus:       domain.TO_BE_ACCOUNT,
@@ -735,7 +739,7 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmDividendsIncent
 			}
 			// TODO 分析成功和失败原因
 			successfullyCount := len(dividendsEstimatesSaved)
-			return fmt.Sprintf("已完成%d单订单分红预算，生成%d单分红预算，%d订单分红预算失败，失败原因：%s", countDividendsOrders, successfullyCount, 0, ""), nil
+			return fmt.Sprintf("已完成%d笔单订单分红预算，生成%d笔单分红预算，%d笔订单分红预算失败，失败原因：%s", countDividendsOrders, successfullyCount, 0, ""), nil
 		}
 	}
 }
--- a/pkg/infrastructure/dao/pg_dividends_estimate_dao.go
查看文件 @3866ab9
+++ b/pkg/infrastructure/dao/pg_dividends_estimate_dao.go
查看文件 @3866ab9
@@ -80,6 +80,12 @@ func (dao *DividendsEstimateDao) CountDividendsEstimate(queryOptions map[string]
 	tx := dao.transactionContext.PgTx
 	var dividendsEstimateModels []*models.DividendsEstimate
 	query := tx.Model(&dividendsEstimateModels)
+	if companyId, ok := queryOptions["companyId"]; ok && companyId.(int64) != 0 {
+		query = query.Where(`dividends_estimate.company @> '{"companyId":"?"}'`, companyId)
+	}
+	if orgId, ok := queryOptions["orgId"]; ok && orgId.(int64) != 0 {
+		query = query.Where(`dividends_estimate.org @> '{"orgId":"?"}'`, orgId)
+	}
 	if count, err := query.AllWithDeleted().SelectAndCount(); err != nil {
 		return 0, err
 	} else {
--- a/pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
查看文件 @3866ab9
+++ b/pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
查看文件 @3866ab9
@@ -44,7 +44,7 @@ func (domainService *ConfirmDividendsIncentivesEstimateService) Confirm(orderGoo
 		// 获取合约
 		cooperationContract, err := cooperationContractRepository.FindOne(map[string]interface{}{"cooperationContractNumber": orderGood.CooperationContractNumber})
 		if err != nil {
-			return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, "共创合约不存在")
+			return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 		}
 		if cooperationContract == nil {
 			return nil, application.ThrowError(application.RES_NO_FIND_ERROR, fmt.Sprintf("共创合约%s不存在", orderGood.CooperationContractNumber))
@@ -112,7 +112,7 @@ func (domainService *ConfirmDividendsIncentivesEstimateService) Confirm(orderGoo
 								Roles:      undertaker.Referrer.Roles,
 								UserInfo:   undertaker.Referrer.UserInfo,
 								UserType:   undertaker.Referrer.UserType,
-								UserName:   undertaker.Referrer.UserInfo.UserName,
+								UserName:   undertaker.Referrer.UserName,
 								UserPhone:  undertaker.Referrer.UserPhone,
 								Company:    undertaker.Referrer.Company,
 							},
--- a/pkg/infrastructure/repository/pg_cooperation_contract_repository.go
查看文件 @3866ab9
+++ b/pkg/infrastructure/repository/pg_cooperation_contract_repository.go
查看文件 @3866ab9
@@ -1041,7 +1041,15 @@ func (repository *CooperationContractRepository) FindOne(queryOptions map[string
 	cooperationContractModel := new(models.CooperationContract)
 	query := sqlbuilder.BuildQuery(tx.Model(cooperationContractModel), queryOptions)
 	query.SetWhereByQueryOption("cooperation_contract.cooperation_contract_id = ?", "cooperationContractId")
-	query.SetWhereByQueryOption("cooperation_contract.cooperation_contract_number = ?", "cooperationContractNumber")
+	if cooperationContractNumber, ok := queryOptions["cooperationContractNumber"]; ok && cooperationContractNumber != "" {
+		query.Where("cooperation_contract.cooperation_contract_number = ?", cooperationContractNumber)
+	}
+	if companyId, ok := queryOptions["companyId"]; ok && companyId.(int64) != 0 {
+		query.Where("company->>'companyId' = '?'", companyId)
+	}
+	if orgId, ok := queryOptions["orgId"]; ok && orgId.(int64) != 0 {
+		query.Where("org->>'orgId' = '?'", orgId)
+	}
 	if err := query.First(); err != nil {
 		if err.Error() == "pg: no rows in result set" {
 			return nil, fmt.Errorf("共创合约不存在")
@@ -1053,15 +1061,28 @@ func (repository *CooperationContractRepository) FindOne(queryOptions map[string
 		return nil, nil
 	} else {
 		// 获取共创模式
-		cooperationModeModels := new(models.CooperationMode)
-		cooperationModeQuery := tx.Model(cooperationModeModels)
-		if err := cooperationModeQuery.
+		//cooperationModeModels := new(models.CooperationMode)
+		//cooperationModeQuery := tx.Model(cooperationModeModels)
+		//if err := cooperationModeQuery.
+		//	Where("company->>'companyId' = '?'", cooperationContractModel.Company.CompanyId).
+		//	Where("org->>'orgId' = '?'", cooperationContractModel.Org.OrgId).
+		//	Where("cooperation_mode_number = ?", cooperationContractModel.CooperationModeNumber).
+		//	First(); err != nil {
+		//	return nil, fmt.Errorf("共创合约关联的共创模式不存在")
+		//}
+		var cooperationModeModels []*models.CooperationMode
+		cooperationModeQuery := tx.Model(&cooperationModeModels)
+		if countMode, err := cooperationModeQuery.
 			Where("company->>'companyId' = '?'", cooperationContractModel.Company.CompanyId).
 			Where("org->>'orgId' = '?'", cooperationContractModel.Org.OrgId).
 			Where("cooperation_mode_number = ?", cooperationContractModel.CooperationModeNumber).
-			First(); err != nil {
-			return nil, fmt.Errorf("共创合约关联的共创模式不存在")
-		}
+			Limit(1).
+			SelectAndCount(); err != nil {
+			log.Logger.Error("合约关联的共创模式不存在", map[string]interface{}{
+				"cooperationContractModel": cooperationContractModel,
+			})
+		} else {
+			if countMode > 0 {
 				// 获取分红激励规则列表
 				var dividendsIncentivesRuleModels []*models.DividendsIncentivesRule
 				dividendsIncentivesRuleQuery := tx.Model(&dividendsIncentivesRuleModels)
@@ -1104,12 +1125,15 @@ func (repository *CooperationContractRepository) FindOne(queryOptions map[string
 				}
 				return transform.TransformToCooperationContractDomainModelFromPgModels(
 					cooperationContractModel,
-			cooperationModeModels,
+					cooperationModeModels[0],
 					dividendsIncentivesRuleModels,
 					moneyIncentivesRuleModels,
 					cooperationContractRelevantModels,
 					cooperationContractUndertakerModels)
 			}
+		}
+		return nil, fmt.Errorf("共创合约不存在")
+	}
 }
 func (repository *CooperationContractRepository) Find(queryOptions map[string]interface{}) (int64, []*domain.CooperationContract, error) {