veloren/network/src/lib.rs

#![feature(trait_alias)]
mod api;
mod channel;
mod controller;
mod message;
mod metrics;
mod mpsc;
mod tcp;
mod types;
mod udp;
mod worker;

#[cfg(test)]
pub mod tests {
    use crate::api::*;
    use futures::executor::block_on;
    use std::{net::SocketAddr, sync::Arc};
    use tracing::*;
    use tracing_subscriber::EnvFilter;
    use uuid::Uuid;
    use uvth::ThreadPoolBuilder;

    struct N {
        _id: u8,
    }

    impl Events for N {
        fn on_remote_connection_open(_net: &Network<N>, _con: &Connection) {}

        fn on_remote_connection_close(_net: &Network<N>, _con: &Connection) {}

        fn on_remote_stream_open(_net: &Network<N>, _st: &Stream) {}

        fn on_remote_stream_close(_net: &Network<N>, _st: &Stream) {}
    }

    pub fn test_tracing() {
        let filter = EnvFilter::from_default_env()
            //.add_directive("[worker]=trace".parse().unwrap())
            //.add_directive("trace".parse().unwrap())
            .add_directive("veloren_network::worker=debug".parse().unwrap())
            .add_directive("veloren_network::controller=trace".parse().unwrap())
            .add_directive("veloren_network::channel=trace".parse().unwrap())
            .add_directive("veloren_network::message=trace".parse().unwrap())
            .add_directive("veloren_network::metrics=trace".parse().unwrap())
            .add_directive("veloren_network::types=trace".parse().unwrap())
            .add_directive("veloren_network::mpsc=debug".parse().unwrap())
            .add_directive("veloren_network::udp=debug".parse().unwrap())
            .add_directive("veloren_network::tcp=debug".parse().unwrap());

        tracing_subscriber::FmtSubscriber::builder()
            // all spans/events with a level higher than TRACE (e.g, info, warn, etc.)
            // will be written to stdout.
            .with_max_level(Level::TRACE)
            .with_env_filter(filter)
            // sets this to be the default, global subscriber for this application.
            .init();
    }

    #[test]
    fn it_works() {
        assert_eq!(2 + 2, 4);
    }

    /*
        #[test]
        #[ignore]
        fn client_server() {
            let thread_pool = Arc::new(
                ThreadPoolBuilder::new()
                    .name("veloren-network-test".into())
                    .build(),
            );
            test_tracing();
            let n1 = Network::<N>::new(Uuid::new_v4(), thread_pool.clone());
            let n2 = Network::<N>::new(Uuid::new_v4(), thread_pool.clone());
            let a1 = Address::Tcp(SocketAddr::from(([127, 0, 0, 1], 52000)));
            let a2 = Address::Tcp(SocketAddr::from(([127, 0, 0, 1], 52001)));
            n1.listen(&a1); //await
            n2.listen(&a2); // only requiered here, but doesnt hurt on n1
            std::thread::sleep(std::time::Duration::from_millis(20));

            let p1 = n1.connect(&a2); //await
            //n2.OnRemoteConnectionOpen triggered
            std::thread::sleep(std::time::Duration::from_millis(20));

            let s1 = n1.open(&p1, 16, Promise::InOrder | Promise::NoCorrupt);
            std::thread::sleep(std::time::Duration::from_millis(20));
            //n2.OnRemoteStreamOpen triggered

            n1.send("Hello World", &s1);
            std::thread::sleep(std::time::Duration::from_millis(20));
            // receive on n2 now

            let s: Option<String> = n2.recv(&s1);
            for _ in 1..4 {
                error!("{:?}", s);
            }
            assert_eq!(s, Some("Hello World".to_string()));

            n1.close(s1);
            //n2.OnRemoteStreamClose triggered

            std::thread::sleep(std::time::Duration::from_millis(20000));
        }
    */

    #[test]
    fn client_server_stream() {
        let thread_pool = Arc::new(
            ThreadPoolBuilder::new()
                .name("veloren-network-test".into())
                .build(),
        );
        test_tracing();
        let n1 = Network::<N>::new(Uuid::new_v4(), thread_pool.clone());
        let n2 = Network::<N>::new(Uuid::new_v4(), thread_pool.clone());
        let a1 = Address::Tcp(SocketAddr::from(([127, 0, 0, 1], 52010)));
        let a2 = Address::Tcp(SocketAddr::from(([127, 0, 0, 1], 52011)));
        block_on(n1.listen(&a1)).unwrap(); //await
        block_on(n2.listen(&a2)).unwrap(); // only requiered here, but doesnt hurt on n1
        std::thread::sleep(std::time::Duration::from_millis(20));

        let p1 = block_on(n1.connect(&a2)); //await
        let p1 = p1.unwrap();
        std::thread::sleep(std::time::Duration::from_millis(20));

        let s1 = block_on(n1.open(&p1, 16, Promise::InOrder | Promise::NoCorrupt));
        //let s2 = n1.open(&p1, 16, Promise::InOrder | Promise::NoCorrupt);
        std::thread::sleep(std::time::Duration::from_millis(20));

        n1.send("Hello World", &s1);
        std::thread::sleep(std::time::Duration::from_millis(20));

        std::thread::sleep(std::time::Duration::from_millis(1000));

        let s: Option<String> = n2.recv(&s1);
        assert_eq!(s, Some("Hello World".to_string()));

        n1.close(s1);
    }
}