pg_dividends_order_repository.go 17.7 KB

原文件审查历史永久链接

package repository

import (
	"fmt"
	"github.com/go-pg/pg/v10"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/utils"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log"
	"time"

	"github.com/linmadan/egglib-go/persistent/pg/sqlbuilder"
	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
	"github.com/linmadan/egglib-go/utils/snowflake"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/pg/models"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/pg/transform"
)

type DividendsOrderRepository struct {
	transactionContext *pgTransaction.TransactionContext
}

func (repository *DividendsOrderRepository) nextIdentify() (int64, error) {
	IdWorker, err := snowflake.NewIdWorker(1)
	if err != nil {
		return 0, err
	}
	id, err := IdWorker.NextId()
	return id, err
}

func (repository *DividendsOrderRepository) Save(dividendsOrder *domain.DividendsOrder) (*domain.DividendsOrder, error) {
	sqlBuildFields := []string{
		"dividends_order_id",
		"dividends_order_number",
		"dividends_original_order_num",
		"dividends_order_amount",
		"order_time",
		"dividend_time",
		"dividend_status",
		"region",
		"customer_name",
		"org",
		"company",
		"created_at",
		"deleted_at",
		"updated_at",
		"operate_time",
		"operator",
		"remarks",
	}
	insertFieldsSnippet := sqlbuilder.SqlFieldsSnippet(sqlBuildFields)
	insertPlaceHoldersSnippet := sqlbuilder.SqlPlaceHoldersSnippet(sqlBuildFields)
	returningFieldsSnippet := sqlbuilder.SqlFieldsSnippet(sqlBuildFields)
	updateFields := sqlbuilder.RemoveSqlFields(sqlBuildFields, "dividendsOrder_id")
	updateFieldsSnippet := sqlbuilder.SqlUpdateFieldsSnippet(updateFields)
	tx := repository.transactionContext.PgTx
	if dividendsOrder.Identify() == nil {
		dividendsOrderId, err := repository.nextIdentify()
		if err != nil {
			return dividendsOrder, err
		} else {
			dividendsOrder.DividendsOrderId = dividendsOrderId
		}
		if _, err := tx.QueryOne(
			pg.Scan(
				&dividendsOrder.DividendsOrderId,
				&dividendsOrder.DividendsOrderNumber,
				&dividendsOrder.DividendsOriginalOrderNum,
				&dividendsOrder.DividendsOrderAmount,
				&dividendsOrder.OrderTime,
				&dividendsOrder.DividendTime,
				&dividendsOrder.DividendStatus,
				&dividendsOrder.Region,
				&dividendsOrder.CustomerName,
				&dividendsOrder.Org,
				&dividendsOrder.Company,
				&dividendsOrder.CreatedAt,
				&dividendsOrder.DeletedAt,
				&dividendsOrder.UpdatedAt,
				&dividendsOrder.OperateTime,
				&dividendsOrder.Operator,
				&dividendsOrder.Remarks,
			),
			fmt.Sprintf("INSERT INTO dividends_orders (%s) VALUES (%s) RETURNING %s", insertFieldsSnippet, insertPlaceHoldersSnippet, returningFieldsSnippet),
			dividendsOrder.DividendsOrderId,
			dividendsOrder.DividendsOrderNumber,
			dividendsOrder.DividendsOriginalOrderNum,
			dividendsOrder.DividendsOrderAmount,
			dividendsOrder.OrderTime,
			dividendsOrder.DividendTime,
			dividendsOrder.DividendStatus,
			dividendsOrder.Region,
			dividendsOrder.CustomerName,
			dividendsOrder.Org,
			dividendsOrder.Company,
			dividendsOrder.CreatedAt,
			nil,
			dividendsOrder.UpdatedAt,
			dividendsOrder.OperateTime,
			dividendsOrder.Operator,
			dividendsOrder.Remarks,
		); err != nil {
			return dividendsOrder, err
		}
		// 新增分红订单产品
		var orderGoodsModel []*models.OrderGood
		for _, good := range dividendsOrder.Goods {
			orderGoodsModel = append(orderGoodsModel, &models.OrderGood{
				OrderGoodAmount:              good.OrderGoodAmount,
				OrderGoodName:                good.OrderGoodName,
				OrderGoodPrice:               good.OrderGoodPrice,
				OrderGoodQuantity:            good.OrderGoodQuantity,
				DividendsOrderNumber:         good.DividendsOrderNumber,
				DividendsReturnedOrderNumber: good.DividendsReturnedOrderNumber,
				CooperationContractNumber:    good.CooperationContractNumber,
				CompanyId:                    good.CompanyId,
				OrgId:                        good.OrgId,
				OrderGoodExpense:             good.OrderGoodExpense,
				OrderGoodDividendsStatus:     good.OrderGoodDividendsStatus,
				CreatedAt:                    time.Now(),
				DeletedAt:                    time.Time{},
				UpdatedAt:                    time.Time{},
			})
		}
		if _, err := tx.Model(&orderGoodsModel).Insert(); err != nil {
			return nil, err
		}
	} else {
		if _, err := tx.QueryOne(
			pg.Scan(
				&dividendsOrder.DividendsOrderId,
				&dividendsOrder.DividendsOrderNumber,
				&dividendsOrder.DividendsOriginalOrderNum,
				&dividendsOrder.DividendsOrderAmount,
				&dividendsOrder.OrderTime,
				&dividendsOrder.DividendTime,
				&dividendsOrder.DividendStatus,
				&dividendsOrder.Region,
				&dividendsOrder.CustomerName,
				&dividendsOrder.Org,
				&dividendsOrder.Company,
				&dividendsOrder.CreatedAt,
				&dividendsOrder.DeletedAt,
				&dividendsOrder.UpdatedAt,
				&dividendsOrder.OperateTime,
				&dividendsOrder.Operator,
				&dividendsOrder.Remarks,
			),
			fmt.Sprintf("UPDATE dividends_orders SET %s WHERE dividends_order_id=? RETURNING %s", updateFieldsSnippet, returningFieldsSnippet),
			dividendsOrder.DividendsOrderId,
			dividendsOrder.DividendsOrderNumber,
			dividendsOrder.DividendsOriginalOrderNum,
			dividendsOrder.DividendsOrderAmount,
			dividendsOrder.OrderTime,
			dividendsOrder.DividendTime,
			dividendsOrder.DividendStatus,
			dividendsOrder.Region,
			dividendsOrder.CustomerName,
			dividendsOrder.Org,
			dividendsOrder.Company,
			dividendsOrder.CreatedAt,
			nil,
			dividendsOrder.UpdatedAt,
			dividendsOrder.OperateTime,
			dividendsOrder.Operator,
			dividendsOrder.Remarks,
			dividendsOrder.Identify(),
		); err != nil {
			return dividendsOrder, err
		}

		// 更新分红订单产品
		var orderGoodsFetched []*models.OrderGood
		orderGoodsQuery := tx.Model(&orderGoodsFetched)
		if err := orderGoodsQuery.Where("dividends_order_number = ?", dividendsOrder.DividendsOrderNumber).Select(); err != nil {
			return nil, err
		}

		// 提取分红订单产品id
		var orderGoodIdsFetched []int64
		for _, orderGood := range orderGoodsFetched {
			orderGoodIdsFetched = append(orderGoodIdsFetched, orderGood.OrderGoodId)
		}

		// 待更新分红产品
		var orderGoodsToUpdate []*domain.OrderGood
		// 待添加分红产品
		var orderGoodsToAdd []*domain.OrderGood
		for _, good := range dividendsOrder.Goods {
			if good.OrderGoodId != 0 {
				orderGoodsToUpdate = append(orderGoodsToUpdate, good)
			} else {
				orderGoodsToAdd = append(orderGoodsToAdd, good)
			}
		}

		// 将待添加的分红订单产品领域模型转换为数据模型
		var orderGoodsToAddModels []*models.OrderGood
		for _, goodDomain := range orderGoodsToAdd {
			orderGoodsToAddModels = append(orderGoodsToAddModels, &models.OrderGood{
				OrderGoodAmount:              goodDomain.OrderGoodAmount,
				OrderGoodName:                goodDomain.OrderGoodName,
				OrderGoodPrice:               goodDomain.OrderGoodPrice,
				OrderGoodQuantity:            goodDomain.OrderGoodQuantity,
				DividendsOrderNumber:         goodDomain.DividendsOrderNumber,
				DividendsReturnedOrderNumber: goodDomain.DividendsReturnedOrderNumber,
				CooperationContractNumber:    goodDomain.CooperationContractNumber,
				OrderGoodExpense:             goodDomain.OrderGoodExpense,
				OrgId:                        goodDomain.OrgId,
				CompanyId:                    goodDomain.CompanyId,
				OrderGoodDividendsStatus:     goodDomain.OrderGoodDividendsStatus,
				CreatedAt:                    time.Now(),
				DeletedAt:                    time.Time{},
				UpdatedAt:                    time.Time{},
			})
		}
		// 添加分红订单产品
		log.Logger.Info("待添加的分红订单产品", map[string]interface{}{
			"orderGoodsToAddModels": orderGoodsToAddModels,
		})
		if len(orderGoodsToAddModels) > 0 {
			if _, err := tx.Model(&orderGoodsToAddModels).Insert(); err != nil {
				return nil, err
			}
		}

		// 待更新或者删除的ids
		var orderGoodIdsToUpdateOrDelete []int64
		for _, orderGoodToUpdate := range orderGoodsToUpdate {
			orderGoodIdsToUpdateOrDelete = append(orderGoodIdsToUpdateOrDelete, orderGoodToUpdate.OrderGoodId)
		}

		// 待更新的分红订单产品id
		orderGoodIdsToUpdate := utils.Intersect(orderGoodIdsFetched, orderGoodIdsToUpdateOrDelete)
		var orderGoodModelsToUpdate []*models.OrderGood
		for _, id := range orderGoodIdsToUpdate {
			for _, orderGoodModel := range orderGoodsFetched {
				if orderGoodModel.OrderGoodId == id {
					orderGoodModelsToUpdate = append(orderGoodModelsToUpdate, orderGoodModel)
				}
			}
		}
		log.Logger.Info("待更新的分红订单产品", map[string]interface{}{
			"orderGoodModelsToUpdate": orderGoodModelsToUpdate,
		})
		for i, orderGoodModelToUpdate := range orderGoodModelsToUpdate {
			for j, orderGood := range dividendsOrder.Goods {
				if orderGood.OrderGoodId == orderGoodModelToUpdate.OrderGoodId {
					orderGoodModelsToUpdate[i] = &models.OrderGood{
						OrderGoodId:                  dividendsOrder.Goods[j].OrderGoodId,
						OrderGoodAmount:              dividendsOrder.Goods[j].OrderGoodAmount,
						OrderGoodName:                dividendsOrder.Goods[j].OrderGoodName,
						OrderGoodPrice:               dividendsOrder.Goods[j].OrderGoodPrice,
						OrderGoodQuantity:            dividendsOrder.Goods[j].OrderGoodQuantity,
						DividendsOrderNumber:         dividendsOrder.Goods[j].DividendsOrderNumber,
						DividendsReturnedOrderNumber: dividendsOrder.Goods[j].DividendsReturnedOrderNumber,
						CooperationContractNumber:    dividendsOrder.Goods[j].CooperationContractNumber,
						OrgId:                        dividendsOrder.Goods[j].OrgId,
						CompanyId:                    dividendsOrder.Goods[j].CompanyId,
						OrderGoodExpense:             dividendsOrder.Goods[j].OrderGoodExpense,
						OrderGoodDividendsStatus:     dividendsOrder.Goods[j].OrderGoodDividendsStatus,
						CreatedAt:                    orderGoodModelsToUpdate[i].CreatedAt,
						DeletedAt:                    orderGoodModelsToUpdate[i].DeletedAt,
						UpdatedAt:                    time.Now(),
					}
				}
			}
		}
		if len(orderGoodModelsToUpdate) > 0 {
			if _, err := tx.Model(&orderGoodModelsToUpdate).WherePK().Update(); err != nil {
				return nil, err
			}
		}

		// 待删除的分红订单产品id
		orderGoodIdsToDelete := utils.Difference(orderGoodIdsFetched, orderGoodIdsToUpdateOrDelete)
		var orderGoodModelsToDelete []*models.OrderGood
		for _, id := range orderGoodIdsToDelete {
			for _, orderGoodModel := range orderGoodsFetched {
				if orderGoodModel.OrderGoodId == id {
					orderGoodModelsToDelete = append(orderGoodModelsToDelete, orderGoodModel)
				}
			}
		}
		log.Logger.Info("待删除的订单产品", map[string]interface{}{
			"orderGoodModelsToDelete": orderGoodModelsToDelete,
		})
		if len(orderGoodModelsToDelete) > 0 {
			if _, err := tx.Model(&orderGoodModelsToDelete).Delete(); err != nil {
				return nil, err
			}
		}
	}
	return dividendsOrder, nil
}

func (repository *DividendsOrderRepository) UpdateMany(dividendsOrders []*domain.DividendsOrder) ([]*domain.DividendsOrder, error) {
	tx := repository.transactionContext.PgTx
	var dividendsOrderModels []*models.DividendsOrder
	for _, dividendsOrder := range dividendsOrders {
		dividendsOrderModels = append(dividendsOrderModels, &models.DividendsOrder{
			DividendsOrderId:          dividendsOrder.DividendsOrderId,
			DividendsOrderNumber:      dividendsOrder.DividendsOrderNumber,
			DividendsOriginalOrderNum: dividendsOrder.DividendsOriginalOrderNum,
			DividendsOrderAmount:      dividendsOrder.DividendsOrderAmount,
			OrderTime:                 dividendsOrder.OrderTime,
			DividendTime:              dividendsOrder.DividendTime,
			DividendStatus:            dividendsOrder.DividendStatus,
			Region:                    dividendsOrder.Region,
			CustomerName:              dividendsOrder.CustomerName,
			Org:                       dividendsOrder.Org,
			Company:                   dividendsOrder.Company,
			CreatedAt:                 dividendsOrder.CreatedAt,
			DeletedAt:                 dividendsOrder.DeletedAt,
			UpdatedAt:                 time.Now(),
			OperateTime:               dividendsOrder.OperateTime,
			Operator:                  dividendsOrder.Operator,
			Remarks:                   dividendsOrder.Remarks,
		})
	}
	if _, err := tx.Model(&dividendsOrderModels).WherePK().Update(); err != nil {
		return nil, err
	}
	return dividendsOrders, nil
}

func (repository *DividendsOrderRepository) Remove(dividendsOrder *domain.DividendsOrder) (*domain.DividendsOrder, error) {
	tx := repository.transactionContext.PgTx
	dividendsOrderModel := new(models.DividendsOrder)
	dividendsOrderModel.DividendsOrderId = dividendsOrder.Identify().(int64)
	if _, err := tx.Model(dividendsOrderModel).WherePK().Delete(); err != nil {
		return dividendsOrder, err
	} else {
		//	删除订单产品
		var orderGoodModels []*models.OrderGood
		if _, err := tx.Model(&orderGoodModels).Where("dividends_order_number = ?", dividendsOrder.DividendsOrderNumber).Delete(); err != nil {
			return nil, err
		}
	}
	return dividendsOrder, nil
}

func (repository *DividendsOrderRepository) BatchRemove(dividendsOrders []*domain.DividendsOrder) ([]*domain.DividendsOrder, error) {
	tx := repository.transactionContext.PgTx
	var dividendsOrderModels []*models.DividendsOrder
	for _, dividendsOrder := range dividendsOrders {
		dividendsOrderModels = append(dividendsOrderModels, &models.DividendsOrder{
			DividendsOrderId: dividendsOrder.Identify().(int64),
		})
	}
	if _, err := tx.Model(&dividendsOrderModels).WherePK().Delete(); err != nil {
		return dividendsOrders, err
	} else {
		for _, dividendsOrder := range dividendsOrders {
			//	删除订单产品
			var orderGoodModels []*models.OrderGood
			if _, err := tx.Model(&orderGoodModels).Where("dividends_order_number = ?", dividendsOrder.DividendsOrderNumber).Delete(); err != nil {
				return nil, err
			}
		}
	}
	return dividendsOrders, nil
}

func (repository *DividendsOrderRepository) FindOne(queryOptions map[string]interface{}) (*domain.DividendsOrder, error) {
	tx := repository.transactionContext.PgTx
	dividendsOrderModel := new(models.DividendsOrder)
	query := sqlbuilder.BuildQuery(tx.Model(dividendsOrderModel), queryOptions)
	query.SetWhereByQueryOption("dividends_order.dividends_order_id = ?", "dividendsOrderId")
	//query.SetWhereByQueryOption("dividends_order.dividends_order_number = ?", "dividendsOrderNumber")
	if dividendsOrderNumber, ok := queryOptions["dividendsOrderNumber"]; ok && dividendsOrderNumber != "" {
		query.Where("dividends_order.dividends_order_number = ?", dividendsOrderNumber)
	}
	if err := query.First(); err != nil {
		if err.Error() == "pg: no rows in result set" {
			return nil, fmt.Errorf("没有此资源")
		} else {
			return nil, err
		}
	}
	if dividendsOrderModel.DividendsOrderId == 0 {
		return nil, nil
	} else {
		// 获取订单产品
		var orderGoodModels []*models.OrderGood
		orderGoodModelQuery := tx.Model(&orderGoodModels)
		if err := orderGoodModelQuery.Where("dividends_order_number = ?", dividendsOrderModel.DividendsOrderNumber).Select(); err != nil {
			return nil, err
		}
		// 聚合分红订单
		return transform.TransformToDividendsOrderDomainModelFromPgModels(dividendsOrderModel, orderGoodModels)
	}
}

func (repository *DividendsOrderRepository) Find(queryOptions map[string]interface{}) (int64, []*domain.DividendsOrder, error) {
	tx := repository.transactionContext.PgTx
	var dividendsOrderModels []*models.DividendsOrder
	dividendsOrders := make([]*domain.DividendsOrder, 0)
	query := sqlbuilder.BuildQuery(tx.Model(&dividendsOrderModels), queryOptions)
	if dividendsOrderNumber, ok := queryOptions["dividendsOrderNumber"]; ok && dividendsOrderNumber != "" {
		query.Where("dividends_order_number ilike ?", fmt.Sprintf("%%%s%%", dividendsOrderNumber))
	}
	if customerName, ok := queryOptions["customerName"]; ok && customerName != "" {
		query.Where("customer_name like ?", fmt.Sprintf("%%%s%%", customerName))
	}
	if dividendsOrderIds, ok := queryOptions["dividendsOrderIds"]; ok && len(dividendsOrderIds.([]int64)) > 0 {
		query.Where("dividends_order_id IN (?)", pg.In(dividendsOrderIds.([]int64)))
	}
	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 orgIds, ok := queryOptions["orgIds"]; ok && len(orgIds.([]int64)) > 0 {
		query.Where("org->>'orgId' in (?)", pg.In(orgIds))
	}
	offsetLimitFlag := true
	if offsetLimit, ok := queryOptions["offsetLimit"]; ok {
		offsetLimitFlag = offsetLimit.(bool)
	}
	if offsetLimitFlag {
		query.SetOffsetAndLimit(20)
	}
	query.SetOrderDirect("dividends_order_id", "DESC")
	if count, err := query.SelectAndCount(); err != nil {
		return 0, dividendsOrders, err
	} else {
		for _, dividendsOrderModel := range dividendsOrderModels {
			//获取订单产品
			var orderGoodModels []*models.OrderGood
			orderGoodModelQuery := tx.Model(&orderGoodModels)
			if err := orderGoodModelQuery.Where("dividends_order_number = ?", dividendsOrderModel.DividendsOrderNumber).Select(); err != nil {
				return 0, nil, err
			}
			// 聚合分红订单
			if dividendsOrder, err := transform.TransformToDividendsOrderDomainModelFromPgModels(dividendsOrderModel, orderGoodModels); err != nil {
				return 0, dividendsOrders, err
			} else {
				dividendsOrders = append(dividendsOrders, dividendsOrder)
			}
		}
		return int64(count), dividendsOrders, nil
	}
}

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