pig-farm-controller/internal/infra/repository/pig_trade_repository.go

package repository

import (
	"context"
	"time"

	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"

	"gorm.io/gorm"
)

// PigPurchaseListOptions 定义了查询猪只采购记录时的可选参数
type PigPurchaseListOptions struct {
	PigBatchID *uint
	Supplier   *string
	OperatorID *uint
	StartTime  *time.Time // 基于 purchase_date 字段
	EndTime    *time.Time // 基于 purchase_date 字段
	OrderBy    string     // 例如 "purchase_date desc"
}

// PigSaleListOptions 定义了查询猪只销售记录时的可选参数
type PigSaleListOptions struct {
	PigBatchID *uint
	Buyer      *string
	OperatorID *uint
	StartTime  *time.Time // 基于 sale_date 字段
	EndTime    *time.Time // 基于 sale_date 字段
	OrderBy    string     // 例如 "sale_date desc"
}

// PigTradeRepository 定义了猪只交易数据持久化的接口。
// 领域服务通过此接口与数据层交互，实现解耦。
type PigTradeRepository interface {
	// CreatePigSaleTx 在数据库中创建一条猪只销售记录。
	CreatePigSaleTx(ctx context.Context, tx *gorm.DB, sale *models.PigSale) error

	// CreatePigPurchaseTx 在数据库中创建一条猪只采购记录。
	CreatePigPurchaseTx(ctx context.Context, tx *gorm.DB, purchase *models.PigPurchase) error

	// ListPigPurchases 支持分页和过滤的猪只采购记录列表查询
	ListPigPurchases(ctx context.Context, opts PigPurchaseListOptions, page, pageSize int) ([]models.PigPurchase, int64, error)

	// ListPigSales 支持分页和过滤的猪只销售记录列表查询
	ListPigSales(ctx context.Context, opts PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error)
}

// gormPigTradeRepository 是 PigTradeRepository 接口的 GORM 实现。
type gormPigTradeRepository struct {
	ctx context.Context
	db  *gorm.DB
}

// NewGormPigTradeRepository 创建一个新的 PigTradeRepository GORM 实现实例。
func NewGormPigTradeRepository(ctx context.Context, db *gorm.DB) PigTradeRepository {
	return &gormPigTradeRepository{ctx: ctx, db: db}
}

// CreatePigSaleTx 实现了在数据库中创建猪只销售记录的逻辑。
func (r *gormPigTradeRepository) CreatePigSaleTx(ctx context.Context, tx *gorm.DB, sale *models.PigSale) error {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigSaleTx")
	return tx.WithContext(repoCtx).Create(sale).Error
}

// CreatePigPurchaseTx 实现了在数据库中创建猪只采购记录的逻辑。
func (r *gormPigTradeRepository) CreatePigPurchaseTx(ctx context.Context, tx *gorm.DB, purchase *models.PigPurchase) error {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigPurchaseTx")
	return tx.WithContext(repoCtx).Create(purchase).Error
}

// ListPigPurchases 实现了分页和过滤查询猪只采购记录的功能
func (r *gormPigTradeRepository) ListPigPurchases(ctx context.Context, opts PigPurchaseListOptions, page, pageSize int) ([]models.PigPurchase, int64, error) {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigPurchases")
	if page <= 0 || pageSize <= 0 {
		return nil, 0, ErrInvalidPagination
	}

	var results []models.PigPurchase
	var total int64

	query := r.db.WithContext(repoCtx).Model(&models.PigPurchase{})

	if opts.PigBatchID != nil {
		query = query.Where("pig_batch_id = ?", *opts.PigBatchID)
	}
	if opts.Supplier != nil {
		query = query.Where("supplier LIKE ?", "%"+*opts.Supplier+"%")
	}
	if opts.OperatorID != nil {
		query = query.Where("operator_id = ?", *opts.OperatorID)
	}
	if opts.StartTime != nil {
		query = query.Where("purchase_date >= ?", *opts.StartTime)
	}
	if opts.EndTime != nil {
		query = query.Where("purchase_date <= ?", *opts.EndTime)
	}

	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	orderBy := "purchase_date DESC"
	if opts.OrderBy != "" {
		orderBy = opts.OrderBy
	}
	query = query.Order(orderBy)

	offset := (page - 1) * pageSize
	err := query.Limit(pageSize).Offset(offset).Find(&results).Error

	return results, total, err
}

// ListPigSales 实现了分页和过滤查询猪只销售记录的功能
func (r *gormPigTradeRepository) ListPigSales(ctx context.Context, opts PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error) {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigSales")
	if page <= 0 || pageSize <= 0 {
		return nil, 0, ErrInvalidPagination
	}

	var results []models.PigSale
	var total int64

	query := r.db.WithContext(repoCtx).Model(&models.PigSale{})

	if opts.PigBatchID != nil {
		query = query.Where("pig_batch_id = ?", *opts.PigBatchID)
	}
	if opts.Buyer != nil {
		query = query.Where("buyer LIKE ?", "%"+*opts.Buyer+"%")
	}
	if opts.OperatorID != nil {
		query = query.Where("operator_id = ?", *opts.OperatorID)
	}
	if opts.StartTime != nil {
		query = query.Where("sale_date >= ?", *opts.StartTime)
	}
	if opts.EndTime != nil {
		query = query.Where("sale_date <= ?", *opts.EndTime)
	}

	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	orderBy := "sale_date DESC"
	if opts.OrderBy != "" {
		orderBy = opts.OrderBy
	}
	query = query.Order(orderBy)

	offset := (page - 1) * pageSize
	err := query.Limit(pageSize).Offset(offset).Find(&results).Error

	return results, total, err
}