From: Matt Joiner <anacrolix@gmail.com>
Date: Mon, 8 Feb 2016 10:38:30 +0000 (+1100)
Subject: Add some missing calls to torrent.updatePiecePriority
X-Git-Tag: v1.0.0~900
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=c674e0a65f066ff8bcf06a24f0ce6c9257b13ffd;p=btrtrc.git

Add some missing calls to torrent.updatePiecePriority
---

diff --git a/client.go b/client.go
index 2d6a3582..805762f1 100644
--- a/client.go
+++ b/client.go
@@ -28,7 +28,6 @@ import (
 	"github.com/anacrolix/missinggo/pubsub"
 	"github.com/anacrolix/sync"
 	"github.com/anacrolix/utp"
-	"github.com/bradfitz/iter"
 	"github.com/edsrzf/mmap-go"
 
 	"github.com/anacrolix/torrent/bencode"
@@ -2492,13 +2491,14 @@ func (me *Client) onFailedPiece(t *torrent, piece int) {
 
 func (me *Client) pieceChanged(t *torrent, piece int) {
 	correct := t.pieceComplete(piece)
-	defer t.publishPieceChange(piece)
 	defer me.event.Broadcast()
 	if correct {
 		me.onCompletedPiece(t, piece)
 	} else {
 		me.onFailedPiece(t, piece)
 	}
+	t.updatePiecePriority(piece)
+	t.publishPieceChange(piece)
 }
 
 func (cl *Client) verifyPiece(t *torrent, piece int) {
@@ -2510,6 +2510,7 @@ func (cl *Client) verifyPiece(t *torrent, piece int) {
 	}
 	p.QueuedForHash = false
 	if t.isClosed() || t.pieceComplete(piece) {
+		t.updatePiecePriority(piece)
 		t.publishPieceChange(piece)
 		return
 	}