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

package repository

import (
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
	"gorm.io/gorm"
)

// PigPenRepository 定义了与猪栏模型相关的数据库操作接口。
type PigPenRepository interface {
	CreatePen(pen *models.Pen) error
	// GetPenByID 根据ID获取单个猪栏 (非事务性)
	GetPenByID(id uint) (*models.Pen, error)
	// GetPenByIDTx 根据ID获取单个猪栏 (事务性)
	GetPenByIDTx(tx *gorm.DB, id uint) (*models.Pen, error)
	ListPens() ([]models.Pen, error)
	// UpdatePen 更新一个猪栏，返回受影响的行数和错误
	UpdatePen(pen *models.Pen) (int64, error)
	// DeletePen 根据ID删除一个猪栏，返回受影响的行数和错误
	DeletePen(id uint) (int64, error)
	// GetPensByBatchIDTx 根据批次ID获取所有关联的猪栏 (事务性)
	GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error)
	// UpdatePenFieldsTx 更新猪栏的指定字段 (事务性)
	UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error
}

// gormPigPenRepository 是 PigPenRepository 接口的 GORM 实现。
type gormPigPenRepository struct {
	db *gorm.DB
}

// NewGormPigPenRepository 创建一个新的 PigPenRepository GORM 实现实例。
func NewGormPigPenRepository(db *gorm.DB) PigPenRepository {
	return &gormPigPenRepository{db: db}
}

// CreatePen 创建一个新的猪栏
func (r *gormPigPenRepository) CreatePen(pen *models.Pen) error {
	return r.db.Create(pen).Error
}

// GetPenByID 根据ID获取单个猪栏 (非事务性)
func (r *gormPigPenRepository) GetPenByID(id uint) (*models.Pen, error) {
	return r.GetPenByIDTx(r.db, id) // 非Tx方法直接调用Tx方法
}

// GetPenByIDTx 在指定的事务中，通过ID获取单个猪栏信息。
func (r *gormPigPenRepository) GetPenByIDTx(tx *gorm.DB, id uint) (*models.Pen, error) {
	var pen models.Pen
	if err := tx.First(&pen, id).Error; err != nil {
		return nil, err
	}
	return &pen, nil
}

// ListPens 列出所有猪栏
func (r *gormPigPenRepository) ListPens() ([]models.Pen, error) {
	var pens []models.Pen
	if err := r.db.Find(&pens).Error; err != nil {
		return nil, err
	}
	return pens, nil
}

// UpdatePen 更新一个猪栏，返回受影响的行数和错误
func (r *gormPigPenRepository) UpdatePen(pen *models.Pen) (int64, error) {
	result := r.db.Model(&models.Pen{}).Where("id = ?", pen.ID).Updates(pen)
	if result.Error != nil {
		return 0, result.Error
	}
	return result.RowsAffected, nil
}

// DeletePen 根据ID删除一个猪栏，返回受影响的行数和错误
func (r *gormPigPenRepository) DeletePen(id uint) (int64, error) {
	result := r.db.Delete(&models.Pen{}, id)
	if result.Error != nil {
		return 0, result.Error
	}
	return result.RowsAffected, nil
}

// GetPensByBatchIDTx 在指定的事务中，获取一个猪群当前关联的所有猪栏。
func (r *gormPigPenRepository) GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error) {
	var pens []*models.Pen
	// 注意：PigBatchID 是指针类型，需要处理 nil 值
	result := tx.Where("pig_batch_id = ?", batchID).Find(&pens)
	if result.Error != nil {
		return nil, result.Error
	}
	return pens, nil
}

// UpdatePenFieldsTx 在指定的事务中，更新一个猪栏的指定字段。
func (r *gormPigPenRepository) UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
	result := tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates)
	return result.Error
}