151 lines
4.5 KiB
Go
151 lines
4.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"time"
|
|
|
|
"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(tx *gorm.DB, sale *models.PigSale) error
|
|
|
|
// CreatePigPurchaseTx 在数据库中创建一条猪只采购记录。
|
|
CreatePigPurchaseTx(tx *gorm.DB, purchase *models.PigPurchase) error
|
|
|
|
// ListPigPurchases 支持分页和过滤的猪只采购记录列表查询
|
|
ListPigPurchases(opts PigPurchaseListOptions, page, pageSize int) ([]models.PigPurchase, int64, error)
|
|
|
|
// ListPigSales 支持分页和过滤的猪只销售记录列表查询
|
|
ListPigSales(opts PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error)
|
|
}
|
|
|
|
// gormPigTradeRepository 是 PigTradeRepository 接口的 GORM 实现。
|
|
type gormPigTradeRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormPigTradeRepository 创建一个新的 PigTradeRepository GORM 实现实例。
|
|
func NewGormPigTradeRepository(db *gorm.DB) PigTradeRepository {
|
|
return &gormPigTradeRepository{db: db}
|
|
}
|
|
|
|
// CreatePigSaleTx 实现了在数据库中创建猪只销售记录的逻辑。
|
|
func (r *gormPigTradeRepository) CreatePigSaleTx(tx *gorm.DB, sale *models.PigSale) error {
|
|
return tx.Create(sale).Error
|
|
}
|
|
|
|
// CreatePigPurchaseTx 实现了在数据库中创建猪只采购记录的逻辑。
|
|
func (r *gormPigTradeRepository) CreatePigPurchaseTx(tx *gorm.DB, purchase *models.PigPurchase) error {
|
|
return tx.Create(purchase).Error
|
|
}
|
|
|
|
// ListPigPurchases 实现了分页和过滤查询猪只采购记录的功能
|
|
func (r *gormPigTradeRepository) ListPigPurchases(opts PigPurchaseListOptions, page, pageSize int) ([]models.PigPurchase, int64, error) {
|
|
if page <= 0 || pageSize <= 0 {
|
|
return nil, 0, ErrInvalidPagination
|
|
}
|
|
|
|
var results []models.PigPurchase
|
|
var total int64
|
|
|
|
query := r.db.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(opts PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error) {
|
|
if page <= 0 || pageSize <= 0 {
|
|
return nil, 0, ErrInvalidPagination
|
|
}
|
|
|
|
var results []models.PigSale
|
|
var total int64
|
|
|
|
query := r.db.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
|
|
}
|