86 lines
2.9 KiB
Go
86 lines
2.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// DirectUploadRepo implements GORM repository for DirectUpload.
|
|
type DirectUploadRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewDirectUploadRepo creates a new DirectUpload repository.
|
|
func NewDirectUploadRepo(db *gorm.DB) *DirectUploadRepo {
|
|
return &DirectUploadRepo{db: db}
|
|
}
|
|
|
|
// FindByID retrieves a direct upload by primary key.
|
|
func (r *DirectUploadRepo) FindByID(ctx context.Context, id uint) (*model.DirectUpload, error) {
|
|
var upload model.DirectUpload
|
|
err := r.db.WithContext(ctx).First(&upload, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &upload, nil
|
|
}
|
|
|
|
// FindByUUID retrieves a direct upload by its upload_uuid.
|
|
func (r *DirectUploadRepo) FindByUUID(ctx context.Context, uploadUUID string) (*model.DirectUpload, error) {
|
|
var upload model.DirectUpload
|
|
err := r.db.WithContext(ctx).Where("upload_uuid = ?", uploadUUID).First(&upload).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &upload, nil
|
|
}
|
|
|
|
// FindByAccountAndID retrieves a direct upload by account ID and primary key.
|
|
func (r *DirectUploadRepo) FindByAccountAndID(ctx context.Context, accountID uint, id uint) (*model.DirectUpload, error) {
|
|
var upload model.DirectUpload
|
|
err := r.db.WithContext(ctx).Where("account_id = ? AND id = ?", accountID, id).First(&upload).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &upload, nil
|
|
}
|
|
|
|
// Create inserts a new direct upload record.
|
|
func (r *DirectUploadRepo) Create(ctx context.Context, upload *model.DirectUpload) error {
|
|
return r.db.WithContext(ctx).Create(upload).Error
|
|
}
|
|
|
|
// Update modifies an existing direct upload record.
|
|
func (r *DirectUploadRepo) Update(ctx context.Context, upload *model.DirectUpload) error {
|
|
return r.db.WithContext(ctx).Save(upload).Error
|
|
}
|
|
|
|
// UpdateStatus changes the status of a direct upload (pending → completed → expired).
|
|
func (r *DirectUploadRepo) UpdateStatus(ctx context.Context, id uint, status model.DirectUploadStatus) error {
|
|
return r.db.WithContext(ctx).Model(&model.DirectUpload{}).Where("id = ?", id).Update("status", status).Error
|
|
}
|
|
|
|
// FindExpired retrieves all direct uploads that have passed their expiry timestamp
|
|
// and are still in pending status.
|
|
func (r *DirectUploadRepo) FindExpired(ctx context.Context, before time.Time) ([]model.DirectUpload, error) {
|
|
var uploads []model.DirectUpload
|
|
err := r.db.WithContext(ctx).
|
|
Where("status = ? AND expires_at < ?", model.DirectUploadStatusPending, before).
|
|
Find(&uploads).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return uploads, nil
|
|
}
|
|
|
|
// BatchDeleteExpired permanently removes expired uploads older than the given cutoff.
|
|
func (r *DirectUploadRepo) BatchDeleteExpired(ctx context.Context, before time.Time) (int64, error) {
|
|
result := r.db.WithContext(ctx).
|
|
Where("status = ? AND expires_at < ?", model.DirectUploadStatusPending, before).
|
|
Delete(&model.DirectUpload{})
|
|
return result.RowsAffected, result.Error
|
|
} |