hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reunify workers, add commands

Browse Source
This commit is contained in:
Ben Sarmiento
2024-01-30 16:27:12 +01:00
parent 4e389fa79c
commit 7794e641ab
7 changed files with 29 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/app.go
View File

@@ -57,32 +57,14 @@ func MainApp(configPath string) {
}
defer workerPool.Release()
// extra 1 worker for the refresh job
refreshPool, err := ants.NewPool(1)
if err != nil {
zurglog.Errorf("Failed to create refresh worker: %v", err)
os.Exit(1)
}
defer refreshPool.Release()
var repairPool *ants.Pool
if config.EnableRepair() {
repairPool, err = ants.NewPool(1)
if err != nil {
zurglog.Errorf("Failed to create repair worker: %v", err)
os.Exit(1)
}
defer repairPool.Release()
}
utils.EnsureDirExists("data") // Ensure the data directory exists
torrentMgr := torrent.NewTorrentManager(config, rd, workerPool, refreshPool, repairPool, log.Named("manager"))
torrentMgr := torrent.NewTorrentManager(config, rd, workerPool, log.Named("manager"))
downloadClient := http.NewHTTPClient(config.GetToken(), config.GetRetriesUntilFailed(), config.GetDownloadTimeoutSecs(), true, config, log.Named("dlclient"))
downloader := universal.NewDownloader(downloadClient)
router := chi.NewRouter()
handlers.AttachHandlers(router, downloader, torrentMgr, config, rd, workerPool, refreshPool, repairPool, log.Named("router"))
handlers.AttachHandlers(router, downloader, torrentMgr, config, rd, workerPool, log.Named("router"))
// go func() {
// if err := netHttp.ListenAndServe(":6060", nil); err != nil && err != netHttp.ErrServerClosed {