tunnet 0.1.0

Embed a Tunnet mesh node in any Rust application
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

//! Inbound mesh stream listener.

use tokio::sync::mpsc;

use crate::error::{Error, Result};
use crate::peer::Peer;
use crate::stream::TunnetStream;
use crate::types::StreamHeader;

pub(crate) struct InboundConnection {
    pub stream: TunnetStream,
    pub peer: Peer,
    pub header: StreamHeader,
}

/// Accepts inbound application streams from mesh peers.
///
/// Similar in shape to [`tokio::net::TcpListener`]: call [`StreamListener::accept`]
/// in a loop. Only available when this process is the coordinator (or standalone).
pub struct StreamListener {
    rx: mpsc::Receiver<InboundConnection>,
}

impl StreamListener {
    pub(crate) fn new(rx: mpsc::Receiver<InboundConnection>) -> Self {
        Self { rx }
    }

    /// Accept the next inbound stream.
    ///
    /// Returns an error when the node has shut down and the acceptor closed.
    pub async fn accept(&mut self) -> Result<(TunnetStream, Peer, StreamHeader)> {
        let inbound = self
            .rx
            .recv()
            .await
            .ok_or_else(|| Error::Internal("stream listener closed (node shut down)".into()))?;
        Ok((inbound.stream, inbound.peer, inbound.header))
    }
}