GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-08-07 21:43:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add configurable thread count for file processing

Browse Source
This commit is contained in:
krau
2025-02-21 13:51:30 +08:00
parent ed21b65c98
commit d7cd2ede01
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
config/viper.go
View File

@@ -9,10 +9,12 @@ import (
) )
type Config struct { type Config struct {
Workers int `toml:"workers" mapstructure:"workers"` Workers int `toml:"workers" mapstructure:"workers"`
Retry int `toml:"retry" mapstructure:"retry"` Retry int `toml:"retry" mapstructure:"retry"`
NoCleanCache bool `toml:"no_clean_cache" mapstructure:"no_clean_cache" json:"no_clean_cache"` NoCleanCache bool `toml:"no_clean_cache" mapstructure:"no_clean_cache" json:"no_clean_cache"`
Users []userConfig `toml:"users" mapstructure:"users" json:"users"` Threads int `toml:"threads" mapstructure:"threads" json:"threads"`
Users []userConfig `toml:"users" mapstructure:"users" json:"users"`
Temp tempConfig `toml:"temp" mapstructure:"temp"` Temp tempConfig `toml:"temp" mapstructure:"temp"`
Log logConfig `toml:"log" mapstructure:"log"` Log logConfig `toml:"log" mapstructure:"log"`
@@ -67,6 +69,7 @@ func Init() error {
viper.SetDefault("workers", 3) viper.SetDefault("workers", 3)
viper.SetDefault("retry", 3) viper.SetDefault("retry", 3)
viper.SetDefault("threads", 4)
viper.SetDefault("telegram.app_id", 1025907) viper.SetDefault("telegram.app_id", 1025907)
viper.SetDefault("telegram.app_hash", "452b0359b988148995f22ff0f4229750") viper.SetDefault("telegram.app_hash", "452b0359b988148995f22ff0f4229750")

4
core/utils.go
View File

@@ -162,9 +162,9 @@ func fixTaskFileExt(task *types.Task, localFilePath string) {
func getTaskThreads(fileSize int64) int { func getTaskThreads(fileSize int64) int {
threads := 1 threads := 1
if fileSize > 1024*1024*100 { if fileSize > 1024*1024*100 {
threads = 4 threads = config.Cfg.Threads
} else if fileSize > 1024*1024*50 { } else if fileSize > 1024*1024*50 {
threads = 2 threads = config.Cfg.Threads / 2
} }
return threads return threads
} }