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

Track deletes within refresh

Browse Source
This commit is contained in:
Ben Sarmiento
2024-05-21 03:01:42 +02:00
parent 2c5e7a1db0
commit 57df90620e
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/torrent/manager.go
View File

@@ -38,8 +38,7 @@ type TorrentManager struct {
RepairKillSwitch chan struct{} RepairKillSwitch chan struct{}
RemountTrigger chan struct{} RemountTrigger chan struct{}
latestState *LibraryState latestState *LibraryState
allAccessKeys mapset.Set[string]
fixers cmap.ConcurrentMap[string, string] // trigger -> [command, id] fixers cmap.ConcurrentMap[string, string] // trigger -> [command, id]
repairTrigger chan *Torrent repairTrigger chan *Torrent
@@ -69,8 +68,7 @@ func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid, w
RepairKillSwitch: make(chan struct{}, 1), RepairKillSwitch: make(chan struct{}, 1),
RemountTrigger: make(chan struct{}, 1), RemountTrigger: make(chan struct{}, 1),
latestState: &LibraryState{}, latestState: &LibraryState{},
allAccessKeys: mapset.NewSet[string](), // this is for tracking what is deleted
} }
t.fixers = t.readFixersFromFile() t.fixers = t.readFixersFromFile()

5
internal/torrent/refresh.go
View File

@@ -80,12 +80,11 @@ func (t *TorrentManager) refreshTorrents(isInitialRun bool) []string {
} }
} }
// removed torrents // removed torrents
t.allAccessKeys.Difference(freshAccessKeys).Each(func(accessKey string) bool { allAccessKeys := mapset.NewSet[string](allTorrents.Keys()...)
allAccessKeys.Difference(freshAccessKeys).Each(func(accessKey string) bool {
t.Delete(accessKey, false) t.Delete(accessKey, false)
t.allAccessKeys.Remove(accessKey)
return false return false
}) })
t.allAccessKeys.Append(freshAccessKeys.ToSlice()...)
t.log.Infof("Compiled into %d torrents, %d were missing info", allTorrents.Count(), noInfoCount) t.log.Infof("Compiled into %d torrents, %d were missing info", allTorrents.Count(), noInfoCount)