Final repair fixes

This commit is contained in:
Ben Sarmiento
2024-01-18 21:19:36 +01:00
parent 0a451cccde
commit 9472fae9fe
5 changed files with 28 additions and 20 deletions

View File

@@ -33,13 +33,14 @@ type TorrentManager struct {
latestState *LibraryState
requiredVersion string
workerPool *ants.Pool
repairPool *ants.Pool
log *logutil.Logger
}
// NewTorrentManager creates a new torrent manager
// it will fetch all torrents and their info in the background
// and store them in-memory and cached in files
func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid, p *ants.Pool, log *logutil.Logger) *TorrentManager {
func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid, workerPool, repairPool *ants.Pool, log *logutil.Logger) *TorrentManager {
t := &TorrentManager{
Config: cfg,
Api: api,
@@ -51,7 +52,8 @@ func NewTorrentManager(cfg config.ConfigInterface, api *realdebrid.RealDebrid, p
allAccessKeys: mapset.NewSet[string](),
latestState: &LibraryState{},
requiredVersion: "11.01.2024",
workerPool: p,
workerPool: workerPool,
repairPool: repairPool,
log: log,
}
@@ -149,7 +151,7 @@ func (t *TorrentManager) assignedDirectoryCb(tor *Torrent, cb func(string)) {
tor.SelectedFiles.IterCb(func(key string, file *File) {
filenames = append(filenames, filepath.Base(file.Path))
fileSizes = append(fileSizes, file.Bytes)
if !tor.Unfixable && unplayable && utils.IsStreamable(file.Path) {
if !tor.Unrepairable && unplayable && utils.IsStreamable(file.Path) {
unplayable = false
}
})

View File

@@ -224,7 +224,7 @@ func (t *TorrentManager) mergeToMain(existing, toMerge *Torrent) Torrent {
Hash: existing.Hash,
DownloadedIDs: mapset.NewSet[string](),
InProgressIDs: mapset.NewSet[string](),
Unfixable: existing.Unfixable || toMerge.Unfixable,
Unrepairable: existing.Unrepairable || toMerge.Unrepairable,
UnassignedLinks: existing.UnassignedLinks.Union(toMerge.UnassignedLinks),
BrokenLinks: existing.BrokenLinks.Union(toMerge.BrokenLinks),
}

View File

@@ -17,10 +17,10 @@ const (
)
func (t *TorrentManager) RepairAll() {
_ = t.workerPool.Submit(func() {
t.log.Info("Repairing all broken torrents")
_ = t.repairPool.Submit(func() {
t.log.Info("Checking for broken torrents")
t.repairAll()
t.log.Info("Finished repairing all torrents")
t.log.Info("Finished checking for broken torrents")
})
}
@@ -34,7 +34,7 @@ func (t *TorrentManager) repairAll() {
hashGroups = append(hashGroups, currentGroup)
allTorrents.IterCb(func(_ string, torrent *Torrent) {
if torrent.AnyInProgress() || torrent.Unfixable {
if torrent.AnyInProgress() || torrent.Unrepairable {
return
}
if currentGroup.Cardinality() >= maxGroupSize {
@@ -69,7 +69,7 @@ func (t *TorrentManager) repairAll() {
var toRepair []*Torrent
allTorrents.IterCb(func(_ string, torrent *Torrent) {
if torrent.AnyInProgress() || torrent.Unfixable {
if torrent.AnyInProgress() || torrent.Unrepairable {
return
}
@@ -100,16 +100,15 @@ func (t *TorrentManager) repairAll() {
}
func (t *TorrentManager) Repair(torrent *Torrent) {
if torrent.Unrepairable {
t.log.Warnf("Torrent %s is unfixable, skipping repair", t.GetKey(torrent))
return
}
if repairing, ok := t.Repairs.Get(t.GetKey(torrent)); ok && repairing {
t.log.Warnf("Torrent %s is already being repaired, skipping repair", t.GetKey(torrent))
return
}
t.Repairs.Set(t.GetKey(torrent), true)
if torrent.Unfixable {
t.log.Warnf("Torrent %s is unfixable, skipping repair", t.GetKey(torrent))
return
}
// save the broken files to the file cache
infoCache, _ := t.DirectoryMap.Get(INT_INFO_CACHE)
torrent.DownloadedIDs.Each(func(id string) bool {
@@ -391,11 +390,11 @@ func (t *TorrentManager) markAsUnplayable(torrent *Torrent) {
func (t *TorrentManager) markAsUnfixable(torrent *Torrent) {
t.log.Warnf("Marking torrent %s as unfixable", t.GetKey(torrent))
torrent.Unfixable = true
torrent.Unrepairable = true
infoCache, _ := t.DirectoryMap.Get(INT_INFO_CACHE)
torrent.DownloadedIDs.Each(func(id string) bool {
info, _ := infoCache.Get(id)
info.Unfixable = true
info.Unrepairable = true
t.writeTorrentToFile(id, info)
return false
})

View File

@@ -23,7 +23,7 @@ type Torrent struct {
OriginalName string `json:"OriginalName"` // immutable
Rename string `json:"Rename"` // modified over time
SelectedFiles cmap.ConcurrentMap[string, *File] `json:"-"` // modified over time
Unfixable bool `json:"Unfixable"` // modified over time
Unrepairable bool `json:"Unfixable"` // modified over time
BrokenLinks mapset.Set[string] `json:"BrokenLinks"` // only relevant on repair
Version string `json:"Version"` // only used for files