mirror of
https://gitlab.com/veloren/veloren.git
synced 2024-08-30 18:12:32 +00:00
2a7c5807ff
fix async_recv and double block_on panic on Network::drop and participant::drop include Cargo.lock from all examples Found a bug on imbris runners with doc tests of `stream::send` and `stream::recv` As neither a backtrace, nor tracing on runners in the doc tests seems to help, i disable them and add them as unit tests
94 lines
2.8 KiB
Rust
94 lines
2.8 KiB
Rust
use lazy_static::*;
|
|
use std::{
|
|
net::SocketAddr,
|
|
sync::{
|
|
atomic::{AtomicU16, Ordering},
|
|
Arc,
|
|
},
|
|
thread,
|
|
time::Duration,
|
|
};
|
|
use tracing::*;
|
|
use tracing_subscriber::EnvFilter;
|
|
use uvth::ThreadPoolBuilder;
|
|
use veloren_network::{Address, Network, Participant, Pid, Stream, PROMISES_NONE};
|
|
|
|
#[allow(dead_code)]
|
|
pub fn setup(tracing: bool, mut sleep: u64) -> (u64, u64) {
|
|
if tracing {
|
|
sleep += 1000
|
|
}
|
|
if sleep > 0 {
|
|
thread::sleep(Duration::from_millis(sleep));
|
|
}
|
|
|
|
let _subscriber = if tracing {
|
|
let filter = EnvFilter::from_default_env()
|
|
.add_directive("trace".parse().unwrap())
|
|
.add_directive("async_std::task::block_on=warn".parse().unwrap())
|
|
.add_directive("veloren_network::tests=trace".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());
|
|
|
|
Some(
|
|
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.
|
|
.try_init(),
|
|
)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
(0, 0)
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn network_participant_stream(
|
|
addr: Address,
|
|
) -> (
|
|
Network,
|
|
Arc<Participant>,
|
|
Stream,
|
|
Network,
|
|
Arc<Participant>,
|
|
Stream,
|
|
) {
|
|
let pool = ThreadPoolBuilder::new().num_threads(2).build();
|
|
let n_a = Network::new(Pid::fake(1), &pool, None);
|
|
let n_b = Network::new(Pid::fake(2), &pool, None);
|
|
|
|
n_a.listen(addr.clone()).await.unwrap();
|
|
let p1_b = n_b.connect(addr).await.unwrap();
|
|
let p1_a = n_a.connected().await.unwrap();
|
|
|
|
let s1_a = p1_a.open(10, PROMISES_NONE).await.unwrap();
|
|
let s1_b = p1_b.opened().await.unwrap();
|
|
|
|
(n_a, p1_a, s1_a, n_b, p1_b, s1_b)
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub fn tcp() -> veloren_network::Address {
|
|
lazy_static! {
|
|
static ref PORTS: AtomicU16 = AtomicU16::new(5000);
|
|
}
|
|
let port = PORTS.fetch_add(1, Ordering::Relaxed);
|
|
veloren_network::Address::Tcp(SocketAddr::from(([127, 0, 0, 1], port)))
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub fn udp() -> veloren_network::Address {
|
|
lazy_static! {
|
|
static ref PORTS: AtomicU16 = AtomicU16::new(5000);
|
|
}
|
|
let port = PORTS.fetch_add(1, Ordering::Relaxed);
|
|
veloren_network::Address::Udp(SocketAddr::from(([127, 0, 0, 1], port)))
|
|
}
|