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

package repository

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

// PigFarmRepository 定义了与猪场资产（猪舍、猪栏）相关的数据库操作接口
type PigFarmRepository interface {
	// PigHouse methods
	CreatePigHouse(house *models.PigHouse) error
	GetPigHouseByID(id uint) (*models.PigHouse, error)
	ListPigHouses() ([]models.PigHouse, error)
	UpdatePigHouse(house *models.PigHouse) error
	DeletePigHouse(id uint) error
	CountPensInHouse(houseID uint) (int64, error)

	// Pen methods
	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(pen *models.Pen) error
	DeletePen(id uint) error
	// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
	GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error)
	// UpdatePenFields 更新猪栏的指定字段 (事务性)
	UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error

	// PigBatch methods
	// GetPigBatchByID 根据ID获取单个猪批次 (非事务性)
	GetPigBatchByID(id uint) (*models.PigBatch, error)
	// GetPigBatchByIDTx 根据ID获取单个猪批次 (事务性)
	GetPigBatchByIDTx(tx *gorm.DB, id uint) (*models.PigBatch, error)
}

// gormPigFarmRepository 是 PigFarmRepository 的 GORM 实现
type gormPigFarmRepository struct {
	db *gorm.DB
}

// NewGormPigFarmRepository 创建一个新的 PigFarmRepository GORM 实现实例
func NewGormPigFarmRepository(db *gorm.DB) PigFarmRepository {
	return &gormPigFarmRepository{db: db}
}

// --- PigHouse Implementation ---

// CreatePigHouse 创建一个新的猪舍
func (r *gormPigFarmRepository) CreatePigHouse(house *models.PigHouse) error {
	return r.db.Create(house).Error
}

// GetPigHouseByID 根据ID获取单个猪舍
func (r *gormPigFarmRepository) GetPigHouseByID(id uint) (*models.PigHouse, error) {
	var house models.PigHouse
	if err := r.db.First(&house, id).Error; err != nil {
		return nil, err
	}
	return &house, nil
}

// ListPigHouses 列出所有猪舍
func (r *gormPigFarmRepository) ListPigHouses() ([]models.PigHouse, error) {
	var houses []models.PigHouse
	if err := r.db.Find(&houses).Error; err != nil {
		return nil, err
	}
	return houses, nil
}

// UpdatePigHouse 更新一个猪舍
func (r *gormPigFarmRepository) UpdatePigHouse(house *models.PigHouse) error {
	result := r.db.Model(&models.PigHouse{}).Where("id = ?", house.ID).Updates(house)
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return gorm.ErrRecordNotFound // 未找到要更新的猪舍或数据未改变
	}
	return nil
}

// DeletePigHouse 根据ID删除一个猪舍
func (r *gormPigFarmRepository) DeletePigHouse(id uint) error {
	result := r.db.Delete(&models.PigHouse{}, id)
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return gorm.ErrRecordNotFound // 未找到要删除的猪舍
	}
	return nil
}

// CountPensInHouse 统计猪舍中的猪栏数量
func (r *gormPigFarmRepository) CountPensInHouse(houseID uint) (int64, error) {
	var count int64
	err := r.db.Model(&models.Pen{}).Where("house_id = ?", houseID).Count(&count).Error
	return count, err
}

// --- Pen Implementation ---

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

// GetPenByID 根据ID获取单个猪栏 (非事务性)
func (r *gormPigFarmRepository) 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
}

// GetPenByIDTx 根据ID获取单个猪栏 (事务性)
func (r *gormPigFarmRepository) 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 *gormPigFarmRepository) 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 *gormPigFarmRepository) UpdatePen(pen *models.Pen) error {
	result := r.db.Model(&models.Pen{}).Where("id = ?", pen.ID).Updates(pen)
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return gorm.ErrRecordNotFound // 未找到要更新的猪栏或数据未改变
	}
	return nil
}

// DeletePen 根据ID删除一个猪栏
func (r *gormPigFarmRepository) DeletePen(id uint) error {
	result := r.db.Delete(&models.Pen{}, id)
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return gorm.ErrRecordNotFound // 未找到要删除的猪栏
	}
	return nil
}

// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
func (r *gormPigFarmRepository) 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 *gormPigFarmRepository) UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
	result := tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates)
	return result.Error
}

// --- PigBatch Implementation ---

// GetPigBatchByID 根据ID获取单个猪批次 (非事务性)
func (r *gormPigFarmRepository) GetPigBatchByID(id uint) (*models.PigBatch, error) {
	var batch models.PigBatch
	if err := r.db.First(&batch, id).Error; err != nil {
		return nil, err
	}
	return &batch, nil
}

// GetPigBatchByIDTx 根据ID获取单个猪批次 (事务性)
func (r *gormPigFarmRepository) GetPigBatchByIDTx(tx *gorm.DB, id uint) (*models.PigBatch, error) {
	var batch models.PigBatch
	if err := tx.First(&batch, id).Error; err != nil {
		return nil, err
	}
	return &batch, nil
}