修改infra.repository包

internal/infra/repository/medication_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"
 )
 
@@ -20,27 +23,30 @@ type MedicationLogListOptions struct {
 
 // MedicationLogRepository 定义了与群体用药日志模型相关的数据库操作接口。
 type MedicationLogRepository interface {
-	CreateMedicationLog(log *models.MedicationLog) error
-	ListMedicationLogs(opts MedicationLogListOptions, page, pageSize int) ([]models.MedicationLog, int64, error)
+	CreateMedicationLog(ctx context.Context, log *models.MedicationLog) error
+	ListMedicationLogs(ctx context.Context, opts MedicationLogListOptions, page, pageSize int) ([]models.MedicationLog, int64, error)
 }
 
 // gormMedicationLogRepository 是 MedicationLogRepository 接口的 GORM 实现。
 type gormMedicationLogRepository struct {
-	db *gorm.DB
+	ctx context.Context
+	db  *gorm.DB
 }
 
 // NewGormMedicationLogRepository 创建一个新的 MedicationLogRepository GORM 实现实例。
-func NewGormMedicationLogRepository(db *gorm.DB) MedicationLogRepository {
-	return &gormMedicationLogRepository{db: db}
+func NewGormMedicationLogRepository(ctx context.Context, db *gorm.DB) MedicationLogRepository {
+	return &gormMedicationLogRepository{ctx: ctx, db: db}
 }
 
 // CreateMedicationLog 创建一条新的群体用药日志记录
-func (r *gormMedicationLogRepository) CreateMedicationLog(log *models.MedicationLog) error {
-	return r.db.Create(log).Error
+func (r *gormMedicationLogRepository) CreateMedicationLog(ctx context.Context, log *models.MedicationLog) error {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateMedicationLog")
+	return r.db.WithContext(repoCtx).Create(log).Error
 }
 
 // ListMedicationLogs 实现了分页和过滤查询用药记录的功能
-func (r *gormMedicationLogRepository) ListMedicationLogs(opts MedicationLogListOptions, page, pageSize int) ([]models.MedicationLog, int64, error) {
+func (r *gormMedicationLogRepository) ListMedicationLogs(ctx context.Context, opts MedicationLogListOptions, page, pageSize int) ([]models.MedicationLog, int64, error) {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "ListMedicationLogs")
 	if page <= 0 || pageSize <= 0 {
 		return nil, 0, ErrInvalidPagination
 	}
@@ -48,7 +54,7 @@ func (r *gormMedicationLogRepository) ListMedicationLogs(opts MedicationLogListO
 	var results []models.MedicationLog
 	var total int64
 
-	query := r.db.Model(&models.MedicationLog{})
+	query := r.db.WithContext(repoCtx).Model(&models.MedicationLog{})
 
 	if opts.PigBatchID != nil {
 		query = query.Where("pig_batch_id = ?", *opts.PigBatchID)