调整仓库方法归属

2025-10-05 17:46:03 +08:00
parent b3933b6d63
commit b6e68e861b
4 changed files with 18 additions and 42 deletions
--- a/internal/infra/repository/pig_batch_repository.go
+++ b/internal/infra/repository/pig_batch_repository.go
@@ -37,11 +37,7 @@ func (r *gormPigBatchRepository) CreatePigBatch(batch *models.PigBatch) (*models

 // GetPigBatchByID 根据ID获取单个猪批次
 func (r *gormPigBatchRepository) 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
+	return r.GetPigBatchByIDTx(r.db, id)
 }

 // UpdatePigBatch 更新一个猪批次