92 lines
3.5 KiB
Go
92 lines
3.5 KiB
Go
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"
|
|
)
|
|
|
|
// OtaRepository 定义了与 OTA 升级任务相关的数据库操作接口。
|
|
type OtaRepository interface {
|
|
// Create 创建一个新的 OTA 任务。
|
|
Create(ctx context.Context, task *models.OTATask) error
|
|
// CreateTx 在指定的事务中创建一个新的 OTA 任务。
|
|
CreateTx(ctx context.Context, tx *gorm.DB, task *models.OTATask) error
|
|
// FindByID 根据任务 ID 查找任务。
|
|
FindByID(ctx context.Context, id uint32) (*models.OTATask, error)
|
|
// FindTasksByStatusesAndCreationTime 根据状态列表和创建时间查找任务。
|
|
FindTasksByStatusesAndCreationTime(ctx context.Context, statuses []models.OTATaskStatus, createdBefore time.Time) ([]*models.OTATask, error)
|
|
// Update 更新单个 OTA 任务。
|
|
Update(ctx context.Context, task *models.OTATask) error
|
|
// UpdateTx 在指定的事务中更新单个 OTA 任务。
|
|
UpdateTx(ctx context.Context, tx *gorm.DB, task *models.OTATask) error
|
|
}
|
|
|
|
// gormOtaRepository 是 OtaRepository 的 GORM 实现
|
|
type gormOtaRepository struct {
|
|
ctx context.Context
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormOtaRepository 创建一个新的 OtaRepository GORM 实现实例
|
|
func NewGormOtaRepository(ctx context.Context, db *gorm.DB) OtaRepository {
|
|
return &gormOtaRepository{
|
|
ctx: ctx,
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// Create 实现了创建新 OTA 任务的逻辑，内部调用 CreateTx 以复用代码。
|
|
func (r *gormOtaRepository) Create(ctx context.Context, task *models.OTATask) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "Create")
|
|
// 使用 r.db 作为事务对象，调用通用的事务方法
|
|
return r.CreateTx(repoCtx, r.db, task)
|
|
}
|
|
|
|
// CreateTx 实现了在事务中创建新 OTA 任务的核心逻辑。
|
|
func (r *gormOtaRepository) CreateTx(ctx context.Context, tx *gorm.DB, task *models.OTATask) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateTx")
|
|
return tx.WithContext(repoCtx).Create(task).Error
|
|
}
|
|
|
|
// FindByID 实现了根据 ID 查找任务的逻辑。
|
|
func (r *gormOtaRepository) FindByID(ctx context.Context, id uint32) (*models.OTATask, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "FindByID")
|
|
var task models.OTATask
|
|
err := r.db.WithContext(repoCtx).First(&task, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &task, nil
|
|
}
|
|
|
|
// FindTasksByStatusesAndCreationTime 实现了根据状态和创建时间查找任务的逻辑。
|
|
func (r *gormOtaRepository) FindTasksByStatusesAndCreationTime(ctx context.Context,
|
|
statuses []models.OTATaskStatus,
|
|
createdBefore time.Time,
|
|
) ([]*models.OTATask, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "FindTasksByStatusesAndCreationTime")
|
|
var tasks []*models.OTATask
|
|
err := r.db.WithContext(repoCtx).
|
|
Where("status IN ? AND created_at < ?", statuses, createdBefore).
|
|
Find(&tasks).Error
|
|
return tasks, err
|
|
}
|
|
|
|
// Update 实现了更新单个 OTA 任务的逻辑，内部调用 UpdateTx 以复用代码。
|
|
func (r *gormOtaRepository) Update(ctx context.Context, task *models.OTATask) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "Update")
|
|
// 使用 r.db 作为事务对象，调用通用的事务方法
|
|
return r.UpdateTx(repoCtx, r.db, task)
|
|
}
|
|
|
|
// UpdateTx 实现了在事务中更新单个 OTA 任务的核心逻辑。
|
|
func (r *gormOtaRepository) UpdateTx(ctx context.Context, tx *gorm.DB, task *models.OTATask) error {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "UpdateTx")
|
|
return tx.WithContext(repoCtx).Save(task).Error
|
|
}
|