feat: add file move and trash retention
This commit is contained in:
@@ -56,7 +56,9 @@ func main() {
|
||||
}
|
||||
|
||||
healthH := handler.NewHealthHandler(pool)
|
||||
nodeH := handler.NewNodeHandler(cfg, repository.NewNodeRepository(pool), store)
|
||||
nodeRepo := repository.NewNodeRepository(pool)
|
||||
nodeH := handler.NewNodeHandler(cfg, nodeRepo, store)
|
||||
go runTrashPurger(ctx, nodeRepo, store)
|
||||
|
||||
r := chi.NewRouter()
|
||||
r.Use(chimw.RequestID)
|
||||
@@ -73,6 +75,7 @@ func main() {
|
||||
r.Route("/nodes/{id}", func(r chi.Router) {
|
||||
r.Get("/", nodeH.Get)
|
||||
r.Patch("/", nodeH.Update)
|
||||
r.Post("/move", nodeH.Move)
|
||||
r.Delete("/", nodeH.Delete)
|
||||
r.Get("/download", nodeH.Download)
|
||||
r.Get("/access", nodeH.ListAccess)
|
||||
@@ -116,3 +119,37 @@ func main() {
|
||||
slog.Error("server shutdown error", "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func runTrashPurger(ctx context.Context, repo *repository.NodeRepository, store *storage.Storage) {
|
||||
ticker := time.NewTicker(time.Hour)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
purgeExpiredTrash(ctx, repo, store)
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func purgeExpiredTrash(ctx context.Context, repo *repository.NodeRepository, store *storage.Storage) {
|
||||
keys, err := repo.ListPurgeableStorageKeys(ctx)
|
||||
if err != nil {
|
||||
slog.Warn("list purgeable files failed", "error", err)
|
||||
return
|
||||
}
|
||||
for _, key := range keys {
|
||||
if err := store.RemoveObject(ctx, key); err != nil {
|
||||
slog.Warn("remove purged file object failed", "key", key, "error", err)
|
||||
}
|
||||
}
|
||||
count, err := repo.PurgeExpired(ctx)
|
||||
if err != nil {
|
||||
slog.Warn("purge expired trash failed", "error", err)
|
||||
return
|
||||
}
|
||||
if count > 0 {
|
||||
slog.Info("purged expired trash", "nodes", count, "objects", len(keys))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user