muxado 0.5.3

The muxado stream multiplexing 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
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

use std::env;

use futures::prelude::*;
use muxado::*;
use tokio::{
    io::{
        AsyncReadExt,
        AsyncWriteExt,
    },
    net::TcpListener,
};
use tracing::{
    info,
    span,
    subscriber::Interest,
    Event,
    Metadata,
};

/// Subscriber that claims that it's always enabled, but does nothing.
#[allow(dead_code)]
struct AlwaysSubscriber;
impl tracing::Subscriber for AlwaysSubscriber {
    #[inline]
    fn register_callsite(&self, _: &'static Metadata<'static>) -> Interest {
        Interest::always()
    }

    fn new_span(&self, _: &span::Attributes<'_>) -> span::Id {
        span::Id::from_u64(0xDEAD)
    }

    fn event(&self, _event: &Event<'_>) {}

    fn record(&self, _span: &span::Id, _values: &span::Record<'_>) {}

    fn record_follows_from(&self, _span: &span::Id, _follows: &span::Id) {}

    #[inline]
    fn enabled(&self, _metadata: &Metadata<'_>) -> bool {
        true
    }

    fn enter(&self, _span: &span::Id) {}
    fn exit(&self, _span: &span::Id) {}
}

#[tokio::main]
async fn main() -> Result<(), anyhow::Error> {
    tracing_subscriber::fmt()
        .pretty()
        // .with_span_events(FmtSpan::ENTER)
        .with_env_filter(env::var("RUST_LOG").unwrap_or("info".into()))
        .init();

    let l = TcpListener::bind("127.0.0.1:1234").await?;

    loop {
        let (conn, _addr) = l.accept().await?;

        let res = async move {
            let mut sess = SessionBuilder::new(conn).start();
            let sess = &mut sess;

            loop {
                let mut stream = if let Some(stream) = sess.accept().await {
                    stream
                } else {
                    break;
                };

                tokio::spawn(
                    async move {
                        let mut buf = vec![0u8; 512];

                        loop {
                            let n = stream.read(&mut buf).await?;
                            if n == 0 {
                                info!("hit eof");
                                break;
                            }
                            info!(
                                msg = String::from_utf8_lossy(&buf[..n]).as_ref(),
                                "read data from stream"
                            );
                            stream.write_all(&buf[0..n]).await?;
                            info!(
                                msg = String::from_utf8_lossy(&buf[..n]).as_ref(),
                                "echoed data to stream"
                            );
                        }

                        Result::<(), anyhow::Error>::Ok(())
                    }
                    .then(|res| async {
                        if let Err(err) = res {
                            tracing::warn!(err = %err, "stream closed with error");
                        } else {
                            tracing::info!("stream closed");
                        }
                    }),
                );
            }
            Result::<(), anyhow::Error>::Ok(())
        }
        .await;

        if let Err(err) = res {
            tracing::warn!(err = %err, "session closed with error");
        } else {
            tracing::warn!("session closed");
        }
    }
}