iroh-bitswap 0.2.0

Implementation of the bitswap protocol
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

use std::collections::VecDeque;

use ahash::AHashSet;
use cid::Cid;

#[derive(Default, Debug)]
pub struct CidQueue {
    elements: VecDeque<Cid>,
    set: AHashSet<Cid>,
}

impl CidQueue {
    pub fn pop(&mut self) -> Option<Cid> {
        while let Some(el) = self.elements.pop_front() {
            if self.set.contains(&el) {
                return Some(el);
            }
        }

        None
    }

    #[allow(dead_code)]
    pub fn cids(&mut self) -> Vec<Cid> {
        // Lazily deletes cids removed from the set.
        self.elements.retain(|el| self.set.contains(el));

        self.elements.iter().copied().collect()
    }

    pub fn push(&mut self, cid: Cid) {
        if self.set.insert(cid) {
            self.elements.push_back(cid);
        }
    }

    pub fn remove(&mut self, cid: &Cid) {
        self.set.remove(cid);
    }

    pub fn has(&self, cid: &Cid) -> bool {
        self.set.contains(cid)
    }

    pub fn len(&self) -> usize {
        self.set.len()
    }
}