实现修改批次绑定的猪栏

internal/infra/repository/pig_farm_repository.go

@@ -17,13 +17,23 @@ type PigFarmRepository interface {
 
 	// 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 实现
@@ -38,10 +48,12 @@ func NewGormPigFarmRepository(db *gorm.DB) PigFarmRepository {
 
 // --- 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 {
@@ -50,6 +62,7 @@ func (r *gormPigFarmRepository) GetPigHouseByID(id uint) (*models.PigHouse, erro
 	return &house, nil
 }
 
+// ListPigHouses 列出所有猪舍
 func (r *gormPigFarmRepository) ListPigHouses() ([]models.PigHouse, error) {
 	var houses []models.PigHouse
 	if err := r.db.Find(&houses).Error; err != nil {
@@ -58,28 +71,31 @@ func (r *gormPigFarmRepository) ListPigHouses() ([]models.PigHouse, error) {
 	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 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 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
@@ -88,10 +104,12 @@ func (r *gormPigFarmRepository) CountPensInHouse(houseID uint) (int64, error) {
 
 // --- 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 {
@@ -100,6 +118,16 @@ func (r *gormPigFarmRepository) GetPenByID(id uint) (*models.Pen, error) {
 	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 {
@@ -108,30 +136,50 @@ func (r *gormPigFarmRepository) ListPens() ([]models.Pen, error) {
 	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 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 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 {
@@ -139,3 +187,12 @@ func (r *gormPigFarmRepository) GetPigBatchByID(id uint) (*models.PigBatch, erro
 	}
 	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
+}

internal/infra/repository/unit_of_work.go

@@ -0,0 +1,58 @@
+package repository
+
+import (
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+// UnitOfWork 定义了工作单元接口，用于抽象事务管理
+type UnitOfWork interface {
+	// ExecuteInTransaction 在一个数据库事务中执行给定的函数。
+	// 如果函数返回错误，事务将被回滚；否则，事务将被提交。
+	// tx 参数是当前事务的 GORM DB 实例，应传递给所有仓库方法。
+	ExecuteInTransaction(fn func(tx *gorm.DB) error) error
+}
+
+// gormUnitOfWork 是 UnitOfWork 接口的 GORM 实现
+type gormUnitOfWork struct {
+	db *gorm.DB
+}
+
+// NewGormUnitOfWork 创建一个新的 gormUnitOfWork 实例
+func NewGormUnitOfWork(db *gorm.DB) UnitOfWork {
+	return &gormUnitOfWork{db: db}
+}
+
+// ExecuteInTransaction 实现了 UnitOfWork 接口的事务执行逻辑
+func (u *gormUnitOfWork) ExecuteInTransaction(fn func(tx *gorm.DB) error) error {
+	tx := u.db.Begin()
+	if tx.Error != nil {
+		return fmt.Errorf("开启事务失败: %w", tx.Error)
+	}
+
+	defer func() {
+		if r := recover(); r != nil {
+			tx.Rollback()
+			// 可以选择在此处记录 panic 日志
+			// u.logger.Errorf("事务中发生 panic，已回滚: %v", r)
+		} else if tx.Error != nil { // 如果函数执行过程中返回错误，或者事务本身有错误，则回滚
+			tx.Rollback()
+			// 可以选择在此处记录错误日志
+			// u.logger.Errorf("事务执行失败，已回滚: %v", tx.Error)
+		}
+	}()
+
+	// 执行业务逻辑函数
+	if err := fn(tx); err != nil {
+		tx.Rollback()
+		return err // 返回业务逻辑函数中的错误
+	}
+
+	// 提交事务
+	if err := tx.Commit().Error; err != nil {
+		return fmt.Errorf("提交事务失败: %w", err)
+	}
+
+	return nil
+}