1
0

Snake case to camel case. Fixed argument parsing. Added a statistics log line to tell how much files gets deleted.

This commit is contained in:
Alexander Andreev 2022-02-06 03:03:01 +04:00
parent 10a64306a3
commit 9b5ae547cc
Signed by: Arav
GPG Key ID: 1327FE8A374CC86F

View File

@ -2,6 +2,7 @@ package main
import ( import (
"dwelling-upload/internal/configuration" "dwelling-upload/internal/configuration"
"dwelling-upload/pkg/utils"
"flag" "flag"
"io/ioutil" "io/ioutil"
"log" "log"
@ -10,32 +11,36 @@ import (
"time" "time"
) )
var configPath string = *flag.String("conf", "config.yaml", "path to configuration file") var configPath *string = flag.String("conf", "config.yaml", "path to configuration file")
func main() { func main() {
config, err := configuration.LoadConfiguration(configPath) flag.Parse()
config, err := configuration.LoadConfiguration(*configPath)
if err != nil { if err != nil {
log.Fatalln(err) log.Fatalln(err)
} }
_ = config _ = config
uploads_dir, err := ioutil.ReadDir(config.Uploads.Directory) uploadsDir, err := ioutil.ReadDir(config.Uploads.Directory)
if err != nil { if err != nil {
log.Fatalf("failed to open directory %s: %s\n", config.Uploads.Directory, err) log.Fatalf("failed to open directory %s: %s\n", config.Uploads.Directory, err)
} }
var deleted_count int64 = 0 var deletedCount int64 = 0
var deteted_size int64 = 0 var deletedSize int64 = 0
for _, entry := range uploads_dir { for _, entry := range uploadsDir {
if time.Duration(entry.ModTime().UTC().Sub(time.Now().UTC()).Hours()) >= time.Duration(config.Uploads.Limits.KeepForHours)*time.Hour { if time.Duration(entry.ModTime().UTC().Sub(time.Now().UTC()).Hours()) >= time.Duration(config.Uploads.Limits.KeepForHours)*time.Hour {
if err := os.Remove(path.Join(config.Uploads.Directory, entry.Name())); err != nil { if err := os.Remove(path.Join(config.Uploads.Directory, entry.Name())); err != nil {
log.Fatalln("failed to remove file ", entry.Name(), ": ", err) log.Fatalln("failed to remove file ", entry.Name(), ": ", err)
} }
deteted_size += entry.Size() deletedSize += entry.Size()
deleted_count++ deletedCount++
} }
} }
log.Println(deleted_count, " file(s) in total of ", deteted_size/1024/1024, " MiB was removed during this run.") _, _, cFSz := utils.ConvertFileSize(deletedSize)
log.Println(deletedCount, "file(s) in total of", cFSz, "was removed during this run.")
} }