调整仓库方法归属

internal/infra/repository/pig_pen_repository.go

@@ -0,0 +1,120 @@
+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)
+	// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
+	GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error)
+	// UpdatePenFields 更新猪栏的指定字段 (事务性)
+	UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error
+
+	GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error)
+	UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error
+}
+
+// pigPenRepository 是 PigPenRepository 接口的 gorm 实现。
+type pigPenRepository struct {
+	db *gorm.DB
+}
+
+// NewGormPigPenRepository 创建一个新的 PigPenRepository 实例。
+func NewGormPigPenRepository(db *gorm.DB) PigPenRepository {
+	return &pigPenRepository{db: db}
+}
+
+// GetPenByIDTx 在指定的事务中，通过ID获取单个猪栏信息。
+func (r *pigPenRepository) GetPenByIDTx(tx *gorm.DB, penID uint) (*models.Pen, error) {
+	var pen models.Pen
+	if err := tx.First(&pen, penID).Error; err != nil {
+		return nil, err
+	}
+	return &pen, nil
+}
+
+// GetPensByBatchIDTx 在指定的事务中，获取一个猪群当前关联的所有猪栏。
+func (r *pigPenRepository) GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error) {
+	var pens []*models.Pen
+	if err := tx.Where("pig_batch_id = ?", batchID).Find(&pens).Error; err != nil {
+		return nil, err
+	}
+	return pens, nil
+}
+
+// UpdatePenFieldsTx 在指定的事务中，更新一个猪栏的指定字段。
+func (r *pigPenRepository) UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
+	return tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates).Error
+}
+
+// --- Pen Implementation ---
+
+// CreatePen 创建一个新的猪栏
+func (r *pigPenRepository) CreatePen(pen *models.Pen) error {
+	return r.db.Create(pen).Error
+}
+
+// GetPenByID 根据ID获取单个猪栏 (非事务性)
+func (r *pigPenRepository) GetPenByID(id uint) (*models.Pen, error) {
+	var pen models.Pen
+	if err := r.db.First(&pen, id).Error; err != nil {
+		return nil, err
+	}
+	return &pen, nil
+}
+
+// ListPens 列出所有猪栏
+func (r *pigPenRepository) 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 *pigPenRepository) 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 *pigPenRepository) DeletePen(id uint) (int64, error) {
+	result := r.db.Delete(&models.Pen{}, id)
+	if result.Error != nil {
+		return 0, result.Error
+	}
+	return result.RowsAffected, nil
+}
+
+// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
+func (r *pigPenRepository) GetPensByBatchID(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
+}
+
+// UpdatePenFields 更新猪栏的指定字段 (事务性)
+func (r *pigPenRepository) UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
+	result := tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates)
+	return result.Error
+}