实现 SickPigManager
This commit is contained in:
@@ -32,7 +32,6 @@ type PigSickLog struct {
|
||||
gorm.Model
|
||||
PigBatchID uint `gorm:"primaryKey;comment:关联的猪批次ID"`
|
||||
PenID uint `gorm:"not null;index;comment:所在猪圈ID"`
|
||||
PigIDs string `gorm:"size:500;comment:涉及的猪只ID列表,逗号分隔"`
|
||||
ChangeCount int `gorm:"not null;comment:变化数量, 正数表示新增, 负数表示移除"`
|
||||
Reason PigBatchSickPigReasonType `gorm:"size:20;not null;comment:变化原因 (如: 患病, 康复, 死亡, 转入, 转出, 其他)"`
|
||||
BeforeCount int `gorm:"comment:变化前的数量"`
|
||||
|
||||
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user