修改infra.repository包

internal/infra/repository/pig_transfer_log_repository.go

@@ -1,9 +1,12 @@
 package repository
 
 import (
+	"context"
 	"time"
 
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+
 	"gorm.io/gorm"
 )
 
@@ -22,39 +25,43 @@ type PigTransferLogListOptions struct {
 // PigTransferLogRepository 定义了猪只迁移日志数据持久化的接口。
 type PigTransferLogRepository interface {
 	// CreatePigTransferLog 在数据库中创建一条猪只迁移日志记录。
-	CreatePigTransferLog(tx *gorm.DB, log *models.PigTransferLog) error
+	CreatePigTransferLog(ctx context.Context, tx *gorm.DB, log *models.PigTransferLog) error
 
 	// GetLogsForPenSince 获取指定猪栏自特定时间点以来的所有迁移日志，按时间倒序排列。
-	GetLogsForPenSince(tx *gorm.DB, penID uint, since time.Time) ([]*models.PigTransferLog, error)
+	GetLogsForPenSince(ctx context.Context, tx *gorm.DB, penID uint, since time.Time) ([]*models.PigTransferLog, error)
 
 	// ListPigTransferLogs 支持分页和过滤的猪只迁移日志列表查询
-	ListPigTransferLogs(opts PigTransferLogListOptions, page, pageSize int) ([]models.PigTransferLog, int64, error)
+	ListPigTransferLogs(ctx context.Context, opts PigTransferLogListOptions, page, pageSize int) ([]models.PigTransferLog, int64, error)
 }
 
 // gormPigTransferLogRepository 是 PigTransferLogRepository 接口的 GORM 实现。
 type gormPigTransferLogRepository struct {
-	db *gorm.DB
+	ctx context.Context
+	db  *gorm.DB
 }
 
 // NewGormPigTransferLogRepository 创建一个新的 PigTransferLogRepository GORM 实现实例。
-func NewGormPigTransferLogRepository(db *gorm.DB) PigTransferLogRepository {
-	return &gormPigTransferLogRepository{db: db}
+func NewGormPigTransferLogRepository(ctx context.Context, db *gorm.DB) PigTransferLogRepository {
+	return &gormPigTransferLogRepository{ctx: ctx, db: db}
 }
 
 // CreatePigTransferLog 实现了在数据库中创建猪只迁移日志记录的逻辑。
-func (r *gormPigTransferLogRepository) CreatePigTransferLog(tx *gorm.DB, log *models.PigTransferLog) error {
-	return tx.Create(log).Error
+func (r *gormPigTransferLogRepository) CreatePigTransferLog(ctx context.Context, tx *gorm.DB, log *models.PigTransferLog) error {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigTransferLog")
+	return tx.WithContext(repoCtx).Create(log).Error
 }
 
 // GetLogsForPenSince 实现了获取猪栏自特定时间点以来所有迁移日志的逻辑。
-func (r *gormPigTransferLogRepository) GetLogsForPenSince(tx *gorm.DB, penID uint, since time.Time) ([]*models.PigTransferLog, error) {
+func (r *gormPigTransferLogRepository) GetLogsForPenSince(ctx context.Context, tx *gorm.DB, penID uint, since time.Time) ([]*models.PigTransferLog, error) {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "GetLogsForPenSince")
 	var logs []*models.PigTransferLog
-	err := tx.Where("pen_id = ? AND transfer_time >= ?", penID, since).Order("transfer_time DESC").Find(&logs).Error
+	err := tx.WithContext(repoCtx).Where("pen_id = ? AND transfer_time >= ?", penID, since).Order("transfer_time DESC").Find(&logs).Error
 	return logs, err
 }
 
 // ListPigTransferLogs 实现了分页和过滤查询猪只迁移日志的功能
-func (r *gormPigTransferLogRepository) ListPigTransferLogs(opts PigTransferLogListOptions, page, pageSize int) ([]models.PigTransferLog, int64, error) {
+func (r *gormPigTransferLogRepository) ListPigTransferLogs(ctx context.Context, opts PigTransferLogListOptions, page, pageSize int) ([]models.PigTransferLog, int64, error) {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigTransferLogs")
 	if page <= 0 || pageSize <= 0 {
 		return nil, 0, ErrInvalidPagination
 	}
@@ -62,7 +69,7 @@ func (r *gormPigTransferLogRepository) ListPigTransferLogs(opts PigTransferLogLi
 	var results []models.PigTransferLog
 	var total int64
 
-	query := r.db.Model(&models.PigTransferLog{})
+	query := r.db.WithContext(repoCtx).Model(&models.PigTransferLog{})
 
 	if opts.PigBatchID != nil {
 		query = query.Where("pig_batch_id = ?", *opts.PigBatchID)