修改infra.repository包

internal/infra/repository/sensor_data_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"
 )
 
@@ -18,39 +21,43 @@ type SensorDataListOptions struct {
 
 // SensorDataRepository 定义了与传感器数据相关的数据库操作接口。
 type SensorDataRepository interface {
-	Create(sensorData *models.SensorData) error
-	GetLatestSensorDataByDeviceIDAndSensorType(deviceID uint, sensorType models.SensorType) (*models.SensorData, error)
+	Create(ctx context.Context, sensorData *models.SensorData) error
+	GetLatestSensorDataByDeviceIDAndSensorType(ctx context.Context, deviceID uint, sensorType models.SensorType) (*models.SensorData, error)
 	// List 支持分页和过滤的列表查询
-	List(opts SensorDataListOptions, page, pageSize int) ([]models.SensorData, int64, error)
+	List(ctx context.Context, opts SensorDataListOptions, page, pageSize int) ([]models.SensorData, int64, error)
 }
 
 // gormSensorDataRepository 是 SensorDataRepository 的 GORM 实现。
 type gormSensorDataRepository struct {
-	db *gorm.DB
+	ctx context.Context
+	db  *gorm.DB
 }
 
 // NewGormSensorDataRepository 创建一个新的 SensorDataRepository GORM 实现实例。
-func NewGormSensorDataRepository(db *gorm.DB) SensorDataRepository {
-	return &gormSensorDataRepository{db: db}
+func NewGormSensorDataRepository(ctx context.Context, db *gorm.DB) SensorDataRepository {
+	return &gormSensorDataRepository{ctx: ctx, db: db}
 }
 
 // Create 将一条新的传感器数据记录插入数据库。
-func (r *gormSensorDataRepository) Create(sensorData *models.SensorData) error {
-	return r.db.Create(sensorData).Error
+func (r *gormSensorDataRepository) Create(ctx context.Context, sensorData *models.SensorData) error {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "Create")
+	return r.db.WithContext(repoCtx).Create(sensorData).Error
 }
 
 // GetLatestSensorDataByDeviceIDAndSensorType 根据设备ID和传感器类型查询最新的传感器数据。
-func (r *gormSensorDataRepository) GetLatestSensorDataByDeviceIDAndSensorType(deviceID uint, sensorType models.SensorType) (*models.SensorData, error) {
+func (r *gormSensorDataRepository) GetLatestSensorDataByDeviceIDAndSensorType(ctx context.Context, deviceID uint, sensorType models.SensorType) (*models.SensorData, error) {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "GetLatestSensorDataByDeviceIDAndSensorType")
 	var sensorData models.SensorData
 	// 增加一个时间范围来缩小查询范围, 从而加快查找速度, 当使用时序数据库时时间范围可以让数据库忽略时间靠前的分片
-	err := r.db.Where("device_id = ? AND sensor_type = ? AND time >=?", deviceID, sensorType, time.Now().Add(-24*time.Hour)).
+	err := r.db.WithContext(repoCtx).Where("device_id = ? AND sensor_type = ? AND time >=?", deviceID, sensorType, time.Now().Add(-24*time.Hour)).
 		Order("time DESC").
 		First(&sensorData).Error
 	return &sensorData, err
 }
 
 // List 实现了分页和过滤查询传感器数据的功能
-func (r *gormSensorDataRepository) List(opts SensorDataListOptions, page, pageSize int) ([]models.SensorData, int64, error) {
+func (r *gormSensorDataRepository) List(ctx context.Context, opts SensorDataListOptions, page, pageSize int) ([]models.SensorData, int64, error) {
+	repoCtx := logs.AddFuncName(ctx, r.ctx, "List")
 	// --- 校验分页参数 ---
 	if page <= 0 || pageSize <= 0 {
 		return nil, 0, ErrInvalidPagination
@@ -59,7 +66,7 @@ func (r *gormSensorDataRepository) List(opts SensorDataListOptions, page, pageSi
 	var results []models.SensorData
 	var total int64
 
-	query := r.db.Model(&models.SensorData{})
+	query := r.db.WithContext(repoCtx).Model(&models.SensorData{})
 
 	// --- 应用过滤条件 ---
 	if opts.DeviceID != nil {