创建批次时插入一条记录

internal/infra/repository/pig_batch_log_repository.go

@@ -0,0 +1,27 @@
+package repository
+
+import (
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"gorm.io/gorm"
+)
+
+// PigBatchLogRepository 定义了与猪批次日志相关的数据库操作接口。
+type PigBatchLogRepository interface {
+	// Create 在指定的事务中创建一条新的猪批次日志。
+	Create(tx *gorm.DB, log *models.PigBatchLog) error
+}
+
+// gormPigBatchLogRepository 是 PigBatchLogRepository 的 GORM 实现。
+type gormPigBatchLogRepository struct {
+	db *gorm.DB
+}
+
+// NewGormPigBatchLogRepository 创建一个新的 PigBatchLogRepository 实例。
+func NewGormPigBatchLogRepository(db *gorm.DB) PigBatchLogRepository {
+	return &gormPigBatchLogRepository{db: db}
+}
+
+// Create 实现了创建猪批次日志的逻辑。
+func (r *gormPigBatchLogRepository) Create(tx *gorm.DB, log *models.PigBatchLog) error {
+	return tx.Create(log).Error
+}

internal/infra/repository/pig_batch_repository.go

@@ -8,6 +8,7 @@ import (
 // PigBatchRepository 定义了与猪批次相关的数据库操作接口
 type PigBatchRepository interface {
 	CreatePigBatch(batch *models.PigBatch) (*models.PigBatch, error)
+	CreatePigBatchTx(tx *gorm.DB, batch *models.PigBatch) (*models.PigBatch, error)
 	GetPigBatchByID(id uint) (*models.PigBatch, error)
 	GetPigBatchByIDTx(tx *gorm.DB, id uint) (*models.PigBatch, error)
 	// UpdatePigBatch 更新一个猪批次，返回更新后的批次、受影响的行数和错误
@@ -29,7 +30,12 @@ func NewGormPigBatchRepository(db *gorm.DB) PigBatchRepository {
 
 // CreatePigBatch 创建一个新的猪批次
 func (r *gormPigBatchRepository) CreatePigBatch(batch *models.PigBatch) (*models.PigBatch, error) {
-	if err := r.db.Create(batch).Error; err != nil {
+	return r.CreatePigBatchTx(r.db, batch)
+}
+
+// CreatePigBatchTx 在指定的事务中，创建一个新的猪批次
+func (r *gormPigBatchRepository) CreatePigBatchTx(tx *gorm.DB, batch *models.PigBatch) (*models.PigBatch, error) {
+	if err := tx.Create(batch).Error; err != nil {
 		return nil, err
 	}
 	return batch, nil