feat: add retry configuration for file upload and validate config values

This commit is contained in:
krau
2025-01-19 14:25:56 +08:00
parent e1bc80ab7d
commit 79fba918cf
2 changed files with 29 additions and 20 deletions

View File

@@ -82,6 +82,7 @@ func Init() {
viper.SetConfigType("toml")
viper.SetDefault("workers", 3)
viper.SetDefault("retry", 3)
viper.SetDefault("temp.base_path", "cache/")
viper.SetDefault("temp.cache_ttl", 3600)
@@ -92,8 +93,6 @@ func Init() {
viper.SetDefault("db.path", "data/saveany.db")
viper.SetDefault("telegram.api", "https://api.telegram.org")
viper.SetDefault("storage.alist.base_path", "/")
viper.SetDefault("storage.alist.token_exp", 3600)
@@ -107,4 +106,8 @@ func Init() {
fmt.Println("Error unmarshalling config file, ", err)
os.Exit(1)
}
if Cfg.Workers < 1 || Cfg.Retry < 1 {
fmt.Println("Invalid workers or retry value")
os.Exit(1)
}
}