hyperdht 0.1.1

Rust implementation of hyperdht, peer announcements and connections on top of dht-rpc
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

use std::{
    net::SocketAddr,
    pin::Pin,
    sync::{Arc, RwLock},
    task::{Context, Poll},
};

use dht_rpc::{IdBytes, generic_hash};
use futures::Stream;
use tokio::sync::mpsc;

use crate::{
    Keypair, Result, adht::DhtInner, announcer::Announcer, next_router::connection::Connection,
};

pub struct Server {
    rx: mpsc::Receiver<Result<Connection>>,
    dht: Arc<RwLock<DhtInner>>,
    announcer: Announcer,
}

impl Server {
    pub fn new(
        rx: mpsc::Receiver<Result<Connection>>,
        keypair: Keypair,
        dht: Arc<RwLock<DhtInner>>,
    ) -> Self {
        let target = IdBytes(generic_hash(&*keypair.public));
        let announcer = Announcer::new(dht.read().unwrap().get_rpc(), keypair, target);
        Self { rx, dht, announcer }
    }

    /// Current relay addresses from the announcer (newest generation).
    pub fn relay_addresses(&self) -> Vec<SocketAddr> {
        self.announcer.relay_addresses()
    }
}

impl Stream for Server {
    type Item = Result<Connection>;

    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
        _ = Pin::new(&mut *self.dht.write().unwrap()).poll_next(cx);
        _ = Pin::new(&mut self.announcer).poll_next(cx);
        Pin::new(&mut self.rx).poll_recv(cx)
    }
}