issue-18 优化代码(只保证编译通过没检查)

internal/infra/repository/pending_collection_repository.go

@@ -0,0 +1,67 @@
+package repository
+
+import (
+	"time"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"gorm.io/gorm"
+)
+
+// PendingCollectionRepository 定义了与待采集请求相关的数据库操作接口。
+type PendingCollectionRepository interface {
+	// Create 创建一个新的待采集请求。
+	Create(req *models.PendingCollection) error
+
+	// FindByCorrelationID 根据关联ID查找一个待采集请求。
+	FindByCorrelationID(correlationID string) (*models.PendingCollection, error)
+
+	// UpdateStatusToFulfilled 将指定关联ID的请求状态更新为“已完成”。
+	UpdateStatusToFulfilled(correlationID string, fulfilledAt time.Time) error
+
+	// MarkAllPendingAsTimedOut 将所有“待处理”请求更新为“已超时”。
+	MarkAllPendingAsTimedOut() (int64, error)
+}
+
+// gormPendingCollectionRepository 是 PendingCollectionRepository 的 GORM 实现。
+type gormPendingCollectionRepository struct {
+	db *gorm.DB
+}
+
+// NewGormPendingCollectionRepository 创建一个新的 PendingCollectionRepository GORM 实现实例。
+func NewGormPendingCollectionRepository(db *gorm.DB) PendingCollectionRepository {
+	return &gormPendingCollectionRepository{db: db}
+}
+
+// Create 创建一个新的待采集请求。
+func (r *gormPendingCollectionRepository) Create(req *models.PendingCollection) error {
+	return r.db.Create(req).Error
+}
+
+// FindByCorrelationID 根据关联ID查找一个待采集请求。
+func (r *gormPendingCollectionRepository) FindByCorrelationID(correlationID string) (*models.PendingCollection, error) {
+	var req models.PendingCollection
+	if err := r.db.First(&req, "correlation_id = ?", correlationID).Error; err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+// UpdateStatusToFulfilled 将指定关联ID的请求状态更新为“已完成”。
+func (r *gormPendingCollectionRepository) UpdateStatusToFulfilled(correlationID string, fulfilledAt time.Time) error {
+	return r.db.Model(&models.PendingCollection{}).
+		Where("correlation_id = ?", correlationID).
+		Updates(map[string]interface{}{
+			"status":       models.PendingStatusFulfilled,
+			"fulfilled_at": &fulfilledAt,
+		}).Error
+}
+
+// MarkAllPendingAsTimedOut 将所有状态为 'pending' 的记录更新为 'timed_out'。
+// 返回被更新的记录数量和错误。
+func (r *gormPendingCollectionRepository) MarkAllPendingAsTimedOut() (int64, error) {
+	result := r.db.Model(&models.PendingCollection{}).
+		Where("status = ?", models.PendingStatusPending).
+		Update("status", models.PendingStatusTimedOut)
+
+	return result.RowsAffected, result.Error
+}