From 090bc59310ec2daddfa83780d1853d52e5952c1d Mon Sep 17 00:00:00 2001
From: YouROK <8yourok8@mail.ru>
Date: Mon, 19 Jul 2021 11:22:40 +0300
Subject: [PATCH] fix removed pieces redownload

---
 server/torr/storage/torrstor/diskpiece.go | 4 ----
 server/torr/storage/torrstor/mempiece.go  | 4 ----
 server/torr/storage/torrstor/piece.go     | 5 +++++
 3 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/server/torr/storage/torrstor/diskpiece.go b/server/torr/storage/torrstor/diskpiece.go
index f716fe7..b20de74 100644
--- a/server/torr/storage/torrstor/diskpiece.go
+++ b/server/torr/storage/torrstor/diskpiece.go
@@ -8,8 +8,6 @@ import (
 	"sync"
 	"time"
 
-	"github.com/anacrolix/torrent"
-
 	"server/log"
 	"server/settings"
 )
@@ -84,6 +82,4 @@ func (p *DiskPiece) Release() {
 	p.piece.Complete = false
 
 	os.Remove(p.name)
-
-	p.piece.cache.torrent.Piece(p.piece.Id).SetPriority(torrent.PiecePriorityNone)
 }
diff --git a/server/torr/storage/torrstor/mempiece.go b/server/torr/storage/torrstor/mempiece.go
index 6b69724..acb3074 100644
--- a/server/torr/storage/torrstor/mempiece.go
+++ b/server/torr/storage/torrstor/mempiece.go
@@ -4,8 +4,6 @@ import (
 	"io"
 	"sync"
 	"time"
-
-	"github.com/anacrolix/torrent"
 )
 
 type MemPiece struct {
@@ -69,6 +67,4 @@ func (p *MemPiece) Release() {
 	}
 	p.piece.Size = 0
 	p.piece.Complete = false
-
-	p.piece.cache.torrent.Piece(p.piece.Id).SetPriority(torrent.PiecePriorityNone)
 }
diff --git a/server/torr/storage/torrstor/piece.go b/server/torr/storage/torrstor/piece.go
index ac2fd58..8080c08 100644
--- a/server/torr/storage/torrstor/piece.go
+++ b/server/torr/storage/torrstor/piece.go
@@ -1,7 +1,9 @@
 package torrstor
 
 import (
+	"github.com/anacrolix/torrent"
 	"github.com/anacrolix/torrent/storage"
+
 	"server/settings"
 )
 
@@ -73,4 +75,7 @@ func (p *Piece) Release() {
 	} else {
 		p.dPiece.Release()
 	}
+
+	p.cache.torrent.Piece(p.Id).SetPriority(torrent.PiecePriorityNone)
+	p.cache.torrent.Piece(p.Id).UpdateCompletion()
 }