Increment webseed peer piece availability

[btrtrc.git] / webseed-peer.go

diff --git a/webseed-peer.go b/webseed-peer.go
index 94adabe99350c103502190da8437c21727158fd6..f15ecf6692f82f9da0b8193b1240ed9e97b42acc 100644 (file)
--- a/webseed-peer.go
+++ b/webseed-peer.go
@@ -36,6 +36,12 @@ func (ws *webseedPeer) String() string {
 
 func (ws *webseedPeer) onGotInfo(info *metainfo.Info) {
        ws.client.SetInfo(info)
+       // There should be probably be a callback in Client instead, so it can remove pieces at its whim
+       // too.
+       ws.client.Pieces.Iterate(func(x uint32) bool {
+               ws.peer.t.incPieceAvailability(pieceIndex(x))
+               return true
+       })
 }
 
 func (ws *webseedPeer) writeInterested(interested bool) bool {