-
-
Notifications
You must be signed in to change notification settings - Fork 621
/
torrent-piece-request-order.go
63 lines (56 loc) · 1.5 KB
/
torrent-piece-request-order.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package torrent
import (
request_strategy "github.com/anacrolix/torrent/request-strategy"
)
func (t *Torrent) updatePieceRequestOrder(pieceIndex int) {
if t.storage == nil {
return
}
t.cl.pieceRequestOrder[t.clientPieceRequestOrderKey()].Update(
t.pieceRequestOrderKey(pieceIndex),
t.requestStrategyPieceOrderState(pieceIndex))
}
func (t *Torrent) clientPieceRequestOrderKey() interface{} {
if t.storage.Capacity == nil {
return t
}
return t.storage.Capacity
}
func (t *Torrent) deletePieceRequestOrder() {
if t.storage == nil {
return
}
cpro := t.cl.pieceRequestOrder
key := t.clientPieceRequestOrderKey()
pro := cpro[key]
for i := 0; i < t.numPieces(); i++ {
pro.Delete(t.pieceRequestOrderKey(i))
}
if pro.Len() == 0 {
delete(cpro, key)
}
}
func (t *Torrent) initPieceRequestOrder() {
if t.storage == nil {
return
}
if t.cl.pieceRequestOrder == nil {
t.cl.pieceRequestOrder = make(map[interface{}]*request_strategy.PieceRequestOrder)
}
key := t.clientPieceRequestOrderKey()
cpro := t.cl.pieceRequestOrder
if cpro[key] == nil {
cpro[key] = request_strategy.NewPieceOrder(request_strategy.NewAjwernerBtree(), t.numPieces())
}
}
func (t *Torrent) addRequestOrderPiece(i int) {
if t.storage == nil {
return
}
t.cl.pieceRequestOrder[t.clientPieceRequestOrderKey()].Add(
t.pieceRequestOrderKey(i),
t.requestStrategyPieceOrderState(i))
}
func (t *Torrent) getPieceRequestOrder() *request_strategy.PieceRequestOrder {
return t.cl.pieceRequestOrder[t.clientPieceRequestOrderKey()]
}