Merge branch 'dev' of http://gitlab.fjmaimaimai.com/allied-creation/allied-creat…

…ion-cooperation into dev

Merge branch 'dev' of http://gitlab.fjmaimaimai.com/allied-creation/allied-creat…
…ion-cooperation into dev
yangfu
pkg/application/cooperationContract/service/cooperation_contract.go
pkg/application/dividendsEstimate/service/dividends_estimate.go
pkg/domain/dividends_order.go
pkg/infrastructure/domain_service/pg_cancel_dividends_estimate_service.go
pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
pkg/infrastructure/pg/models/cooperation_contract_change_log.go
pkg/infrastructure/repository/pg_cooperation_contract_repository.go
pkg/infrastructure/repository/pg_dividends_estimate_repository.go
--- a/pkg/application/cooperationContract/service/cooperation_contract.go
查看文件 @e3f8571
+++ b/pkg/application/cooperationContract/service/cooperation_contract.go
查看文件 @e3f8571
@@ -1095,6 +1095,7 @@ func (cooperationContractService *CooperationContractService) UpdateCooperationC
 			UndertakerId:              undertakerId,
 			UserId:                    undertakerDomain.UserId,
 			CooperationContractNumber: cooperationContract.CooperationContractNumber,
+			CooperationContractId:     cooperationContract.CooperationContractId,
 			UserBaseId:                undertakerDomain.UserBaseId,
 			Org:                       undertakerDomain.Org,
 			Orgs:                      undertakerDomain.Orgs,
--- a/pkg/application/dividendsEstimate/service/dividends_estimate.go
查看文件 @e3f8571
+++ b/pkg/application/dividendsEstimate/service/dividends_estimate.go
查看文件 @e3f8571
@@ -5,6 +5,7 @@ import (
 	"github.com/linmadan/egglib-go/core/application"
 	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
 	"github.com/linmadan/egglib-go/utils/tool_funs"
+	"github.com/shopspring/decimal"
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/application/dividendsEstimate/command"
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/application/dividendsEstimate/dto"
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/application/dividendsEstimate/query"
@@ -354,12 +355,44 @@ func (dividendsEstimateService *DividendsEstimateService) BatchCancelDividendsEs
 				}
 				// 校验是否能够取消
-				if dividendsEstimate.DividendsType == domain.ORDER_DIVIDENDS || dividendsEstimate.DividendsType == domain.RETURN_WRITE_OFF { // 订单分红、退货冲销
-					if countRelativeDividendsEstimates, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(map[string]interface{}{
+				if dividendsEstimate.DividendsType == domain.ORDER_DIVIDENDS || dividendsEstimate.DividendsType == domain.RETURN_WRITE_OFF { // 订单分红/退货冲销
+					// 自定义查询
+					queryOptions := make(map[string]interface{})
+
+					//参与分红类型，1承接人，2推荐人，3关联业务员
+					if dividendsEstimate.DividendsParticipateType == 1 { // 当前取消的是承接人的分红预算单
+						queryOptions = map[string]interface{}{
+							"companyId":                 dividendsEstimate.Company.CompanyId,
+							"orgId":                     dividendsEstimate.Org.OrgId,
 							"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+							"dividendsParticipateTypes": []int32{2, 3},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsTypes":            []int32{1, 2},
+						}
+					} else if dividendsEstimate.DividendsParticipateType == 2 { // 当前取消的是推荐人的分红预算单
+						queryOptions = map[string]interface{}{
 							"companyId":                 dividendsEstimate.Company.CompanyId,
 							"orgId":                     dividendsEstimate.Org.OrgId,
-					}); err2 != nil {
+							"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+							"dividendsParticipateTypes": []int32{1, 3},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsTypes":            []int32{1, 2},
+						}
+					} else if dividendsEstimate.DividendsParticipateType == 3 { // 当前取消的是关联业务员的分红预算单
+						queryOptions = map[string]interface{}{
+							"companyId":                 dividendsEstimate.Company.CompanyId,
+							"orgId":                     dividendsEstimate.Org.OrgId,
+							"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+							"dividendsParticipateTypes": []int32{1, 2},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsTypes":            []int32{1, 2},
+						}
+					}
+
+					if countRelativeDividendsEstimates, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(queryOptions); err2 != nil {
 						return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err2.Error())
 					} else {
 						if countRelativeDividendsEstimates > 0 {
@@ -371,11 +404,43 @@ func (dividendsEstimateService *DividendsEstimateService) BatchCancelDividendsEs
 						}
 					}
 				} else if dividendsEstimate.DividendsType == domain.MONEY_INCENTIVES { // 金额激励分红
-					if countRelativeDividendsEstimates, dividendsEstimatesRelative, err3 := dividendsEstimateRepository.Find(map[string]interface{}{
+					// 自定义查询
+					queryOptions := make(map[string]interface{})
+
+					//参与分红类型，1承接人，2推荐人，3关联业务员
+					if dividendsEstimate.DividendsParticipateType == 1 { // 当前取消的是承接人的分红预算单
+						queryOptions = map[string]interface{}{
+							"companyId":                 dividendsEstimate.Company.CompanyId,
+							"orgId":                     dividendsEstimate.Org.OrgId,
 							"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+							"dividendsParticipateTypes": []int32{2, 3},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsType":             int32(3),
+						}
+					} else if dividendsEstimate.DividendsParticipateType == 2 { // 当前取消的是推荐人的分红预算单
+						queryOptions = map[string]interface{}{
 							"companyId":                 dividendsEstimate.Company.CompanyId,
 							"orgId":                     dividendsEstimate.Org.OrgId,
-					}); err3 != nil {
+							"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+							"dividendsParticipateTypes": []int32{1, 3},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsType":             int32(3),
+						}
+					} else if dividendsEstimate.DividendsParticipateType == 3 { // 当前取消的是关联业务员的分红预算单
+						queryOptions = map[string]interface{}{
+							"companyId":                 dividendsEstimate.Company.CompanyId,
+							"orgId":                     dividendsEstimate.Org.OrgId,
+							"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+							"dividendsParticipateTypes": []int32{1, 2},
+							"offsetLimit":               false,
+							"dividendsStage":            dividendsEstimate.DividendsStage,
+							"dividendsType":             int32(3),
+						}
+					}
+
+					if countRelativeDividendsEstimates, dividendsEstimatesRelative, err3 := dividendsEstimateRepository.Find(queryOptions); err3 != nil {
 						return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err3.Error())
 					} else {
 						if countRelativeDividendsEstimates > 0 {
@@ -388,12 +453,13 @@ func (dividendsEstimateService *DividendsEstimateService) BatchCancelDividendsEs
 					}
 				}
 			}
+
 			// 批量取消分红预算单
-			if dividendsEstimatesCanceled, err := cancelDividendsEstimateService.CancelEstimate(dividendsEstimateIds, operator); err != nil {
-				return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
+			if dividendsEstimatesCanceled, errCancel := cancelDividendsEstimateService.CancelEstimate(dividendsEstimateIds, operator); errCancel != nil {
+				return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, errCancel.Error())
 			} else {
-				if err := transactionContext.CommitTransaction(); err != nil {
-					return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
+				if errTx := transactionContext.CommitTransaction(); errTx != nil {
+					return nil, application.ThrowError(application.TRANSACTION_ERROR, errTx.Error())
 				}
 				return dividendsEstimatesCanceled, nil
 			}
@@ -651,6 +717,7 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmDividendsIncent
 		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
 	} else {
 		dividendsEstimates := make([]*domain.DividendsEstimate, 0)
+
 		// 统计当前企业分红预算单数
 		count, err := dividendsEstimateDao.CountDividendsEstimate(map[string]interface{}{
 			"companyId": confirmDividendsIncentivesEstimateCommand.CompanyId,
@@ -658,6 +725,7 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmDividendsIncent
 		if err != nil {
 			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
 		}
+
 		var countDividendsEstimate int64
 		for _, orderGood := range orderGoods {
 			dividendsEstimate := &domain.DividendsEstimate{}
@@ -1076,10 +1144,13 @@ func (dividendsEstimateService *DividendsEstimateService) ConfirmMoneyIncentives
 			// 统计生成的预算单数目
 			countDividendsEstimate = countDividendsEstimate + 1
+			// 分红金额精度处理
+			dividendsAmount, _ := decimal.NewFromFloatWithExponent(dividendsEstimateDetail.DividendsAmount, -2).Float64()
+
 			dividendsEstimate := &domain.DividendsEstimate{
 				DividendsEstimateId:          0,
 				DividendsAccountStatus:       domain.TO_BE_ACCOUNT,
-				DividendsAmount:              dividendsEstimateDetail.DividendsAmount,
+				DividendsAmount:              dividendsAmount,
 				DividendsEstimateOrderNumber: dividendsEstimateOrderNumber,
 				DividendsEstimateTime:        time.Now(),
 				DividendsParticipateType:     dividendsEstimateDetail.DividendsParticipateType,
--- a/pkg/domain/dividends_order.go
查看文件 @e3f8571
+++ b/pkg/domain/dividends_order.go
查看文件 @e3f8571
@@ -106,6 +106,9 @@ func (dividendsOrder *DividendsOrder) Update(data map[string]interface{}) error 
 	if customerName, ok := data["customerName"]; ok {
 		dividendsOrder.CustomerName = customerName.(string)
 	}
+	if remarks, ok := data["remarks"]; ok {
+		dividendsOrder.Remarks = remarks.(string)
+	}
 	dividendsOrder.UpdatedAt = time.Now()
 	return nil
 }
--- a/pkg/infrastructure/domain_service/pg_cancel_dividends_estimate_service.go
查看文件 @e3f8571
+++ b/pkg/infrastructure/domain_service/pg_cancel_dividends_estimate_service.go
查看文件 @e3f8571
@@ -40,7 +40,6 @@ func (domainService *CancelDividendsEstimateService) CancelEstimate(dividendsEst
 		return nil, err
 	} else {
 		dividendsReturnedOrderRepository = repo
-
 	}
 	// 订单产品仓储初始化
@@ -60,14 +59,55 @@ func (domainService *CancelDividendsEstimateService) CancelEstimate(dividendsEst
 			var dividendsEstimatesCanceled []*domain.DividendsEstimate
 			var orderOrReturnOrderNums []string
 			for _, dividendsEstimate := range dividendsEstimates {
-				if dividendsEstimate.DividendsType != 3 { // 非金额激励分红预算单处理
-					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
-					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(map[string]interface{}{
+				if dividendsEstimate.DividendsType == 1 || dividendsEstimate.DividendsType == 2 { //  1.订单分红 2.退货冲销
+					// 自定义查询
+					queryOptions := make(map[string]interface{})
+
+					queryOptions = map[string]interface{}{
 						"companyId":                 dividendsEstimate.Company.CompanyId,
 						"orgId":                     dividendsEstimate.Org.OrgId,
 						"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+						"dividendsParticipateTypes": []int32{1, 2, 3},
 						"offsetLimit":               false,
-					}); err2 != nil {
+						"dividendsStage":            dividendsEstimate.DividendsStage,
+						"dividendsTypes":            []int32{1, 2},
+					}
+
+					//参与分红类型，1承接人，2推荐人，3关联业务员
+					//if dividendsEstimate.DividendsParticipateType == 1 { // 当前取消的是承接人的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+					//		"dividendsParticipateTypes": []int32{1, 2, 3},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsTypes":            []int32{1, 2},
+					//	}
+					//} else if dividendsEstimate.DividendsParticipateType == 2 { // 当前取消的是推荐人的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+					//		"dividendsParticipateTypes": []int32{1, 3},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsTypes":            []int32{1, 2},
+					//	}
+					//} else if dividendsEstimate.DividendsParticipateType == 3 { // 当前取消的是关联业务员的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"orderOrReturnedOrderNum":   dividendsEstimate.OrderOrReturnedOrderNum,
+					//		"dividendsParticipateTypes": []int32{1, 2},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsTypes":            []int32{1, 2},
+					//	}
+					//}
+
+					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
+					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(queryOptions); err2 != nil {
 						return nil, err2
 					} else {
 						if countRelative > 0 {
@@ -84,15 +124,55 @@ func (domainService *CancelDividendsEstimateService) CancelEstimate(dividendsEst
 							dividendsEstimatesCanceled = append(dividendsEstimatesCanceled, dividendsEstimatesRelativeCanceled...)
 						}
 					}
-				} else if dividendsEstimate.DividendsType == 3 {
-					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
-					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(map[string]interface{}{
+				} else if dividendsEstimate.DividendsType == 3 { // 3.金额激励
+					// 自定义查询
+					queryOptions := make(map[string]interface{})
+
+					queryOptions = map[string]interface{}{
 						"companyId":                 dividendsEstimate.Company.CompanyId,
 						"orgId":                     dividendsEstimate.Org.OrgId,
 						"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
-						"dividendsType":             int32(3),
+						"dividendsParticipateTypes": []int32{1, 2, 3},
 						"offsetLimit":               false,
-					}); err2 != nil {
+						"dividendsStage":            dividendsEstimate.DividendsStage,
+						"dividendsType":             int32(3),
+					}
+
+					//参与分红类型，1承接人，2推荐人，3关联业务员
+					//if dividendsEstimate.DividendsParticipateType == 1 { // 当前取消的是承接人的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+					//		"dividendsParticipateTypes": []int32{2, 3},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsType":             int32(3),
+					//	}
+					//} else if dividendsEstimate.DividendsParticipateType == 2 { // 当前取消的是推荐人的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+					//		"dividendsParticipateTypes": []int32{1, 3},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsType":             int32(3),
+					//	}
+					//} else if dividendsEstimate.DividendsParticipateType == 3 { // 当前取消的是关联业务员的分红预算单
+					//	queryOptions = map[string]interface{}{
+					//		"companyId":                 dividendsEstimate.Company.CompanyId,
+					//		"orgId":                     dividendsEstimate.Org.OrgId,
+					//		"cooperationContractNumber": dividendsEstimate.CooperationContractNumber,
+					//		"dividendsParticipateTypes": []int32{1, 2},
+					//		"offsetLimit":               false,
+					//		"dividendsStage":            dividendsEstimate.DividendsStage,
+					//		"dividendsType":             int32(3),
+					//	}
+					//}
+
+					// 根据当前分红预算单的分红订单号或退货单号获取相关的分红预算单
+					if countRelative, dividendsEstimatesRelative, err2 := dividendsEstimateRepository.Find(queryOptions); err2 != nil {
 						return nil, err2
 					} else {
 						if countRelative > 0 {
@@ -110,20 +190,26 @@ func (domainService *CancelDividendsEstimateService) CancelEstimate(dividendsEst
 				}
 			}
-			// 分红订单或分红退货单状态变更
+			// 分红订单
 			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 {
--- a/pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
查看文件 @e3f8571
+++ b/pkg/infrastructure/domain_service/pg_confirm_dividends_incentives_estimate_service.go
查看文件 @e3f8571
@@ -8,6 +8,7 @@ import (
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/domain"
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/domain/service"
 	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/repository"
+	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log"
 )
 type ConfirmDividendsIncentivesEstimateService struct {
@@ -53,6 +54,10 @@ func (domainService *ConfirmDividendsIncentivesEstimateService) Confirm(orderGoo
 	// 确认业绩分红预算
 	var dividendsEstimateDetails []*service.DividendsEstimateDetail
 	for _, orderGood := range orderGoods {
+		log.Logger.Info("业绩分红预算产品", map[string]interface{}{
+			"orderGood": orderGood,
+		})
+
 		// 获取合约
 		cooperationContract, err := cooperationContractRepository.FindOne(map[string]interface{}{
 			"cooperationContractNumber": orderGood.CooperationContractNumber,
@@ -95,6 +100,9 @@ func (domainService *ConfirmDividendsIncentivesEstimateService) Confirm(orderGoo
 			})
 			continue
 		} else {
+			log.Logger.Info("产品相关的合约", map[string]interface{}{
+				"cooperationContract": cooperationContract,
+			})
 			if cooperationContract.Status == 2 {
 				var orderNumber string
 				if orderGood.DividendsOrderNumber != "" {
--- a/pkg/infrastructure/pg/models/cooperation_contract_change_log.go
查看文件 @e3f8571
+++ b/pkg/infrastructure/pg/models/cooperation_contract_change_log.go
查看文件 @e3f8571
@@ -30,7 +30,7 @@ type CooperationContractChangeLog struct {
 	// 更新时间
 	UpdatedAt time.Time `comment:"更新时间"`
 	// 删除时间
-	DeletedAt time.Time `comment:"删除时间"`
+	DeletedAt time.Time `comment:"删除时间" pg:",soft_delete"`
 	// 创建时间
 	CreatedAt time.Time `comment:"创建时间"`
 }
--- a/pkg/infrastructure/repository/pg_cooperation_contract_repository.go
查看文件 @e3f8571
+++ b/pkg/infrastructure/repository/pg_cooperation_contract_repository.go
查看文件 @e3f8571
@@ -1146,7 +1146,7 @@ func (repository *CooperationContractRepository) Find(queryOptions map[string]in
 	if cooperationContractNumber, ok := queryOptions["cooperationContractNumber"]; ok && cooperationContractNumber != "" {
 		query.Where("cooperation_contract_number ilike ?", fmt.Sprintf("%%%s%%", cooperationContractNumber))
 	}
-	if cooperationContractName, ok := queryOptions["cooperationContactName"]; ok && cooperationContractName != "" {
+	if cooperationContractName, ok := queryOptions["cooperationContractName"]; ok && cooperationContractName != "" {
 		query.Where("cooperation_contract_name like ?", fmt.Sprintf("%%%s%%", cooperationContractName))
 	}
 	if departmentName, ok := queryOptions["departmentName"]; ok && departmentName != "" {
--- a/pkg/infrastructure/repository/pg_dividends_estimate_repository.go
查看文件 @e3f8571
+++ b/pkg/infrastructure/repository/pg_dividends_estimate_repository.go
查看文件 @e3f8571
@@ -307,6 +307,9 @@ func (repository *DividendsEstimateRepository) Find(queryOptions map[string]inte
 	if dividendsType, ok := queryOptions["dividendsType"]; ok && dividendsType.(int32) != 0 {
 		query.Where("dividends_type = ?", dividendsType)
 	}
+	if dividendsTypes, ok := queryOptions["dividendsTypes"]; ok && len(dividendsTypes.([]int32)) > 0 {
+		query.Where("dividends_type in (?)", pg.In(dividendsTypes.([]int32)))
+	}
 	if dividendsEstimateOrderNumber, ok := queryOptions["dividendsEstimateOrderNumber"]; ok && dividendsEstimateOrderNumber != "" {
 		query.Where("dividends_estimate_order_number ilike ?", fmt.Sprintf("%%%s%%", dividendsEstimateOrderNumber))
 	}
@@ -325,6 +328,12 @@ func (repository *DividendsEstimateRepository) Find(queryOptions map[string]inte
 	if dividendsParticipateType, ok := queryOptions["dividendsParticipateType"]; ok && dividendsParticipateType.(int32) != 0 {
 		query.Where("dividends_participate_type = ?", dividendsParticipateType)
 	}
+	if dividendsParticipateTypes, ok := queryOptions["dividendsParticipateTypes"]; ok && len(dividendsParticipateTypes.([]int32)) > 0 {
+		query.Where("dividends_participate_type in (?)", pg.In(dividendsParticipateTypes.([]int32)))
+	}
+	if dividendsStage, ok := queryOptions["dividendsStage"]; ok && dividendsStage.(int32) > 0 {
+		query.Where("dividends_stage = ?", dividendsStage)
+	}
 	if cooperationContractNumber, ok := queryOptions["cooperationContractNumber"]; ok && cooperationContractNumber != "" {
 		query.Where("cooperation_contract_number = ?", cooperationContractNumber)
 	}