实现 SickPigManager

internal/infra/repository/pig_sick_repository.go

@@ -1,13 +1,20 @@
 package repository
 
 import (
+	"errors"
+
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"gorm.io/gorm"
 )
 
 // PigSickLogRepository 定义了与病猪日志模型相关的数据库操作接口。
 type PigSickLogRepository interface {
+	// CreatePigSickLog 创建一条新的病猪日志记录
 	CreatePigSickLog(log *models.PigSickLog) error
+	CreatePigSickLogTx(tx *gorm.DB, log *models.PigSickLog) error
+
+	// GetLastLogByBatchTx 在事务中获取指定批次和猪栏的最新一条 PigSickLog 记录
+	GetLastLogByBatchTx(tx *gorm.DB, batchID uint) (*models.PigSickLog, error)
 }
 
 // gormPigSickLogRepository 是 PigSickLogRepository 接口的 GORM 实现。
@@ -22,5 +29,25 @@ func NewGormPigSickLogRepository(db *gorm.DB) PigSickLogRepository {
 
 // CreatePigSickLog 创建一条新的病猪日志记录
 func (r *gormPigSickLogRepository) CreatePigSickLog(log *models.PigSickLog) error {
-	return r.db.Create(log).Error
+	return r.CreatePigSickLogTx(r.db, log)
+}
+func (r *gormPigSickLogRepository) CreatePigSickLogTx(tx *gorm.DB, log *models.PigSickLog) error {
+	return tx.Create(log).Error
+}
+
+// GetLastLogByBatchTx 在事务中获取指定批次和猪栏的最新一条 PigSickLog 记录
+func (r *gormPigSickLogRepository) GetLastLogByBatchTx(tx *gorm.DB, batchID uint) (*models.PigSickLog, error) {
+	var lastLog models.PigSickLog
+	err := tx.
+		Where("pig_batch_id = ?", batchID).
+		Order("happened_at DESC"). // 按时间降序排列
+		First(&lastLog).Error      // 获取第一条记录 (即最新一条)
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, gorm.ErrRecordNotFound // 明确返回记录未找到错误
+		}
+		return nil, err
+	}
+	return &lastLog, nil
 }