rubit 0.1.0

The bencoding decoding package for rubit Bittorrent 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

use std::{
    collections::{HashSet, VecDeque},
    fs::File,
    net::SocketAddr,
    sync::{Arc, Mutex},
    thread::{self, JoinHandle},
};


use rubit_bencode::TorrentFile;

use crate::PeerConnManager;

pub struct PeerManager {
    pub peers: Arc<Mutex<HashSet<SocketAddr>>>,
}

impl PeerManager {
    pub fn new() -> Self {
        Self {
            peers: Arc::new(Mutex::new(HashSet::new())),
        }
    }

    pub fn try_add(
        &self,
        global_queue: Arc<Mutex<VecDeque<usize>>>,
        socket_addr: SocketAddr,
        torrent_file: Arc<TorrentFile>,
        peer_id: [u8; 20],
        file: Arc<Mutex<File>>,
    ) -> Option<JoinHandle<()>> {
        let mut set = self.peers.lock().unwrap();

        if set.insert(socket_addr) {
            let peers_clone = Arc::clone(&self.peers);
            Some(thread::spawn(move || {
                let mut peer_manager = PeerConnManager::new();

                match peer_manager.handle_peer(
                    global_queue,
                    socket_addr,
                    torrent_file,
                    peer_id,
                    file,
                ) {
                    Err(_) => {
                        let mut set = peers_clone.lock().unwrap();
                        set.remove(&socket_addr);
                    }
                    _ => (),
                };
            }))
        } else {
            None
        }
    }
}