Drop support for go 1.20

[btrtrc.git] / torrent-piece-request-order.go

diff --git a/torrent-piece-request-order.go b/torrent-piece-request-order.go
index 86c9839893b313a52e7c201c5e24af6293864bfb..10623da0b5d83594c396a14a11a2599123b44151 100644 (file)
--- a/torrent-piece-request-order.go
+++ b/torrent-piece-request-order.go
@@ -8,9 +8,11 @@ func (t *Torrent) updatePieceRequestOrder(pieceIndex int) {
        if t.storage == nil {
                return
        }
-       t.cl.pieceRequestOrder[t.clientPieceRequestOrderKey()].Update(
-               t.pieceRequestOrderKey(pieceIndex),
-               t.requestStrategyPieceOrderState(pieceIndex))
+       if ro, ok := t.cl.pieceRequestOrder[t.clientPieceRequestOrderKey()]; ok {
+               ro.Update(
+                       t.pieceRequestOrderKey(pieceIndex),
+                       t.requestStrategyPieceOrderState(pieceIndex))
+       }
 }
 
 func (t *Torrent) clientPieceRequestOrderKey() interface{} {
@@ -45,7 +47,7 @@ func (t *Torrent) initPieceRequestOrder() {
        key := t.clientPieceRequestOrderKey()
        cpro := t.cl.pieceRequestOrder
        if cpro[key] == nil {
-               cpro[key] = request_strategy.NewPieceOrder()
+               cpro[key] = request_strategy.NewPieceOrder(request_strategy.NewAjwernerBtree(), t.numPieces())
        }
 }