rogee
aeddedf2a3
Restructure the monorepo into clear top-level directories: - backend/: Go module root (cmd, internal, pkg, configs, migrations, docs/swagger, scripts, tests, go.mod, Makefile, .air.toml) - deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd - docs/: project documentation + reports/ (moved from repo root) - AGENTS.md: new AI coding-agent guide at repo root Update all references to the new layout: - Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root) - docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile, env_file ../../.env, volume mounts ../../backend:/app - deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile - CI: working-directory: backend for go commands, file deploy/docker/Dockerfile, coverage path backend/coverage.out, health_check backend/scripts/ - backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../ - README: architecture tree, quickstart, config paths updated Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh, gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to preserve history. Build, vet, SQLite tests, and docker compose config verified.
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
|
|
} |