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

Repairs

Browse Source
This commit is contained in:
Ben Sarmiento
2024-01-16 20:10:46 +01:00
parent 2cb29284f6
commit 727c694c02
5 changed files with 41 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/torrent/repair.go
View File

@@ -100,6 +100,12 @@ func (t *TorrentManager) repairAll() {
}
func (t *TorrentManager) Repair(torrent *Torrent) {
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
@@ -121,17 +127,13 @@ func (t *TorrentManager) Repair(torrent *Torrent) {
})
_ = t.workerPool.Submit(func() {
t.log.Infof("Repairing torrent %s", t.GetKey(torrent))
torrent.Lock.Lock()
defer torrent.Lock.Unlock()
torrent.Repairing = true
t.repair(torrent)
torrent.Repairing = false
t.log.Infof("Finished repairing torrent %s", t.GetKey(torrent))
})
}
func (t *TorrentManager) repair(torrent *Torrent) {
if torrent.AnyInProgress() || torrent.Repairing {
if torrent.AnyInProgress() {
t.log.Infof("Torrent %s is in progress, skipping repair until download is done", t.GetKey(torrent))
return
}
@@ -387,14 +389,10 @@ func (t *TorrentManager) markAsUnplayable(torrent *Torrent) {
func (t *TorrentManager) markAsUnfixable(torrent *Torrent) {
t.log.Warnf("Marking torrent %s as unfixable", t.GetKey(torrent))
torrent.Lock.Lock()
defer torrent.Lock.Unlock()
torrent.Unfixable = true
infoCache, _ := t.DirectoryMap.Get(INT_INFO_CACHE)
torrent.DownloadedIDs.Each(func(id string) bool {
info, _ := infoCache.Get(id)
info.Lock.Lock()
defer info.Lock.Unlock()
info.Unfixable = true
t.writeTorrentToFile(id, info)
return false