spill 0.5.2

Modern torrent client
Documentation 
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

//! Ten-second rate ordering with one optimistic slot rotated every thirty seconds.
//! Slot grants are policy only; the transfer owner handles wire backpressure.
use std::time::{Duration, Instant};

pub(in crate::engine) struct Candidate<K> {
    pub(in crate::engine) id: K,
    pub(in crate::engine) eligible: bool,
    pub(in crate::engine) downloaded: u64,
    pub(in crate::engine) uploaded: u64,
}

pub(in crate::engine) struct Choker<K> {
    last: Option<Instant>,
    optimistic_at: Option<Instant>,
    optimistic: Option<K>,
    grants: Vec<K>,
    samples: Vec<(K, u64, u64)>,
    seeding: bool,
}
impl<K: Copy + Eq> Choker<K> {
    pub(in crate::engine) fn new() -> Self {
        Self {
            last: None,
            optimistic_at: None,
            optimistic: None,
            grants: Vec::new(),
            samples: Vec::new(),
            seeding: false,
        }
    }
    pub(in crate::engine) fn deadline(&self) -> Option<Instant> {
        self.last
            .map(|at| at + Duration::from_secs(10))
            .into_iter()
            .chain(
                self.optimistic_at
                    .filter(|_| self.optimistic.is_some())
                    .map(|at| at + Duration::from_secs(30)),
            )
            .min()
    }
    /// Eligibility changes are signalled by the owner; rates need a timer only.
    pub(in crate::engine) fn due(&self, now: Instant, seeding: bool) -> bool {
        self.last.is_none()
            || seeding != self.seeding
            || self.deadline().is_some_and(|at| now >= at)
    }
    pub(in crate::engine) fn granted(&self, id: K) -> bool {
        self.grants.contains(&id)
    }
    pub(in crate::engine) fn update(
        &mut self,
        now: Instant,
        seeding: bool,
        slots: usize,
        peers: &[Candidate<K>],
    ) {
        // Fill newly available slots promptly; otherwise rates change grants only
        // at rechoke boundaries. Lost interest/closed peers cannot retain slots.
        let eligible = peers.iter().filter(|p| p.eligible).count();
        let invalid = self
            .grants
            .iter()
            .any(|id| !peers.iter().any(|p| p.id == *id && p.eligible));
        if self.deadline().is_some_and(|at| now < at)
            && seeding == self.seeding
            && !invalid
            && self.grants.len() == slots.min(eligible)
        {
            return;
        }
        let mut ranked: Vec<_> = peers
            .iter()
            .filter(|p| p.eligible)
            .map(|p| {
                let old = self.samples.iter().find(|s| s.0 == p.id);
                let bytes = if seeding { p.uploaded } else { p.downloaded };
                let previous = old.map_or(0, |s| if seeding { s.2 } else { s.1 });
                (p.id, bytes.saturating_sub(previous))
            })
            .collect();
        // All deltas span the same rechoke interval; ordering is rate ordering.
        ranked.sort_by_key(|p| std::cmp::Reverse(p.1));
        self.grants.clear();
        let regular = if ranked.len() > slots {
            slots.saturating_sub(1)
        } else {
            slots
        };
        self.grants.extend(ranked.iter().take(regular).map(|p| p.0));
        let rotate = self
            .optimistic_at
            .is_none_or(|at| now.duration_since(at) >= Duration::from_secs(30));
        if self.grants.len() < slots && ranked.len() > regular {
            let remaining: Vec<_> = peers
                .iter()
                .filter(|p| p.eligible && !self.grants.contains(&p.id))
                .map(|p| p.id)
                .collect();
            if rotate || self.optimistic.is_none_or(|id| !remaining.contains(&id)) {
                let next = self
                    .optimistic
                    .and_then(|id| remaining.iter().position(|p| *p == id))
                    .map_or(0, |position| (position + 1) % remaining.len());
                self.optimistic = Some(remaining[next]);
                self.optimistic_at = Some(now);
            }
            self.grants
                .push(self.optimistic.expect("optimistic candidate"));
        } else {
            self.optimistic = None;
        }
        self.samples = peers
            .iter()
            .map(|p| (p.id, p.downloaded, p.uploaded))
            .collect();
        self.last = Some(now);
        self.seeding = seeding;
    }
}