Support rebooting workers

internal/torrent/manager.go

@@ -21,24 +21,26 @@ const (
 )
 
 type TorrentManager struct {
-	Config          config.ConfigInterface
-	Api             *realdebrid.RealDebrid
-	DirectoryMap    cmap.ConcurrentMap[string, cmap.ConcurrentMap[string, *Torrent]] // directory -> accessKey -> Torrent
-	DownloadCache   cmap.ConcurrentMap[string, *realdebrid.Download]
-	DownloadMap     cmap.ConcurrentMap[string, *realdebrid.Download]
-	fixers          cmap.ConcurrentMap[string, *Torrent]
-	deleteOnceDone  mapset.Set[string]
-	allAccessKeys   mapset.Set[string]
-	latestState     *LibraryState
-	requiredVersion string
-	workerPool      *ants.Pool
-	refreshPool     *ants.Pool
-	repairPool      *ants.Pool
-	repairTrigger   chan *Torrent
-	repairSet       mapset.Set[*Torrent]
-	repairRunning   bool
-	repairRunningMu sync.Mutex
-	log             *logutil.Logger
+	Config            config.ConfigInterface
+	Api               *realdebrid.RealDebrid
+	DirectoryMap      cmap.ConcurrentMap[string, cmap.ConcurrentMap[string, *Torrent]] // directory -> accessKey -> Torrent
+	DownloadCache     cmap.ConcurrentMap[string, *realdebrid.Download]
+	DownloadMap       cmap.ConcurrentMap[string, *realdebrid.Download]
+	fixers            cmap.ConcurrentMap[string, *Torrent]
+	deleteOnceDone    mapset.Set[string]
+	allAccessKeys     mapset.Set[string]
+	latestState       *LibraryState
+	requiredVersion   string
+	workerPool        *ants.Pool
+	refreshPool       *ants.Pool
+	RefreshKillSwitch chan struct{}
+	RepairKillSwitch  chan struct{}
+	repairPool        *ants.Pool
+	repairTrigger     chan *Torrent
+	repairSet         mapset.Set[*Torrent]
+	repairRunning     bool
+	repairRunningMu   sync.Mutex
+	log               *logutil.Logger
 }
 
 // NewTorrentManager creates a new torrent manager
@@ -46,20 +48,22 @@ type TorrentManager struct {
 // and store them in-memory and cached in files
 func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid, workerPool, refreshPool, repairPool *ants.Pool, log *logutil.Logger) *TorrentManager {
 	t := &TorrentManager{
-		Config:          cfg,
-		Api:             api,
-		DirectoryMap:    cmap.New[cmap.ConcurrentMap[string, *Torrent]](),
-		DownloadCache:   cmap.New[*realdebrid.Download](),
-		DownloadMap:     cmap.New[*realdebrid.Download](),
-		fixers:          cmap.New[*Torrent](),
-		deleteOnceDone:  mapset.NewSet[string](),
-		allAccessKeys:   mapset.NewSet[string](),
-		latestState:     &LibraryState{},
-		requiredVersion: "0.9.3-hotfix.3",
-		workerPool:      workerPool,
-		refreshPool:     refreshPool,
-		repairPool:      repairPool,
-		log:             log,
+		Config:            cfg,
+		Api:               api,
+		DirectoryMap:      cmap.New[cmap.ConcurrentMap[string, *Torrent]](),
+		DownloadCache:     cmap.New[*realdebrid.Download](),
+		DownloadMap:       cmap.New[*realdebrid.Download](),
+		RefreshKillSwitch: make(chan struct{}, 1),
+		RepairKillSwitch:  make(chan struct{}, 1),
+		fixers:            cmap.New[*Torrent](),
+		deleteOnceDone:    mapset.NewSet[string](),
+		allAccessKeys:     mapset.NewSet[string](),
+		latestState:       &LibraryState{},
+		requiredVersion:   "0.9.3-hotfix.3",
+		workerPool:        workerPool,
+		refreshPool:       refreshPool,
+		repairPool:        repairPool,
+		log:               log,
 	}
 
 	t.initializeDirectories()