mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-02 06:56:36 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
59 lines
1.8 KiB
Go
59 lines
1.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package ingest
|
|
|
|
import (
|
|
"context"
|
|
|
|
database "github.com/Rain-kl/Wavelet/plugins/infra/database"
|
|
uploadcache "github.com/Rain-kl/Wavelet/plugins/domain/upload/cache"
|
|
"github.com/Rain-kl/Wavelet/plugins/domain/upload/models"
|
|
"github.com/Rain-kl/Wavelet/plugins/domain/upload/repository"
|
|
uploadstats "github.com/Rain-kl/Wavelet/plugins/domain/upload/stats"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Remove soft-deletes an upload and decrements incremental stats.
|
|
func Remove(ctx context.Context, uploadID uint64) (models.Upload, error) {
|
|
upload, err := repository.GetActiveUploadByID(ctx, uploadID)
|
|
if err != nil {
|
|
return models.Upload{}, err
|
|
}
|
|
if err := softDeleteUploadWithStats(ctx, &upload); err != nil {
|
|
return models.Upload{}, err
|
|
}
|
|
upload.Status = models.UploadStatusDeleted
|
|
return upload, nil
|
|
}
|
|
|
|
// RemoveOwned soft-deletes an upload owned by userID and decrements incremental stats.
|
|
func RemoveOwned(ctx context.Context, userID, uploadID uint64) (models.Upload, error) {
|
|
upload, err := repository.GetActiveUploadByID(ctx, uploadID)
|
|
if err != nil {
|
|
return models.Upload{}, err
|
|
}
|
|
if upload.UserID != userID {
|
|
return models.Upload{}, ErrForbidden
|
|
}
|
|
if err := softDeleteUploadWithStats(ctx, &upload); err != nil {
|
|
return models.Upload{}, err
|
|
}
|
|
upload.Status = models.UploadStatusDeleted
|
|
return upload, nil
|
|
}
|
|
|
|
func softDeleteUploadWithStats(ctx context.Context, upload *models.Upload) error {
|
|
statsSnapshot := *upload
|
|
if err := database.DB(ctx).Transaction(func(tx *gorm.DB) error {
|
|
if err := repository.SoftDeleteUploadTx(tx, upload); err != nil {
|
|
return err
|
|
}
|
|
return uploadstats.ApplyUploadStatsDeltaTx(tx, &statsSnapshot, -1)
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
uploadcache.InvalidateUploadMetaCache(ctx, upload.ID)
|
|
return nil
|
|
}
|