Fix stalls for responsive transfer tests

[btrtrc.git] / torrent.go

diff --git a/torrent.go b/torrent.go
index 26f49b82a6b289a08e9526d015b5d974db90f902..66c834545cfaa5870f4d6bde6a33f545003d19d1 100644 (file)
--- a/torrent.go
+++ b/torrent.go
@@ -1760,13 +1760,14 @@ func (t *Torrent) pieceHashed(piece pieceIndex, passed bool, hashIoErr error) {
                        c._stats.incrementPiecesDirtiedGood()
                }
                t.clearPieceTouchers(piece)
+               p.incrementPendingWrites()
                t.cl.unlock()
                err := p.Storage().MarkComplete()
                if err != nil {
                        t.logger.Printf("%T: error marking piece complete %d: %s", t.storage, piece, err)
                }
                t.cl.lock()
-
+               p.decrementPendingWrites()
                if t.closed.IsSet() {
                        return
                }