Compare commits
No commits in common. "92720a7156cf9840d238221c51ba15b54dbda353" and "39fb40e0a0b41c30544a713d0a6d2119c5e002a5" have entirely different histories.
92720a7156
...
39fb40e0a0
@ -100,14 +100,8 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
|
|||||||
|
|
||||||
if let Some(_) = &self.udp.socket {
|
if let Some(_) = &self.udp.socket {
|
||||||
for (addr, conn) in &mut self.connections {
|
for (addr, conn) in &mut self.connections {
|
||||||
let interval = if conn.last_sent_ping > conn.last_recv_pong {
|
|
||||||
Duration::from_millis(2000)
|
|
||||||
} else {
|
|
||||||
self.config.ping_interval
|
|
||||||
};
|
|
||||||
|
|
||||||
let duration = now - conn.last_sent_ping;
|
let duration = now - conn.last_sent_ping;
|
||||||
if duration > interval {
|
if duration > self.config.ping_interval {
|
||||||
match conn.state {
|
match conn.state {
|
||||||
ConnectionState::ReceivingConnection | ConnectionState::Connecting => {
|
ConnectionState::ReceivingConnection | ConnectionState::Connecting => {
|
||||||
match self
|
match self
|
||||||
@ -185,7 +179,8 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
|
|||||||
}
|
}
|
||||||
match self.udp.send_delayed_messages() {
|
match self.udp.send_delayed_messages() {
|
||||||
Ok(_) => {}
|
Ok(_) => {}
|
||||||
Err(_) => {
|
Err(err) => {
|
||||||
|
println!("Error: {}", err);
|
||||||
self.close();
|
self.close();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -329,7 +324,6 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
|
|||||||
Package::Pong => {
|
Package::Pong => {
|
||||||
if let Some(conn) = self.connections.get_mut(addr) {
|
if let Some(conn) = self.connections.get_mut(addr) {
|
||||||
if conn.state == ConnectionState::Connected {
|
if conn.state == ConnectionState::Connected {
|
||||||
conn.last_recv_pong = Instant::now();
|
|
||||||
conn.ping = Instant::now() - conn.last_sent_ping;
|
conn.ping = Instant::now() - conn.last_sent_ping;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -536,28 +530,24 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn send_delayed_messages(&mut self) -> Result<(), SendError> {
|
fn send_delayed_messages(&mut self) -> Result<(), SendError> {
|
||||||
let now = Instant::now();
|
let queue_iter = self.delayed_queue.clone().into_iter();
|
||||||
|
let treshold = Instant::now() - self.additional_ping;
|
||||||
|
let messages = queue_iter.take_while(|m| treshold >= m.created);
|
||||||
|
self.delayed_queue.retain(|m| treshold < m.created);
|
||||||
|
|
||||||
if let Some(socket) = &self.socket {
|
if let Some(socket) = &self.socket {
|
||||||
for i in (0..self.delayed_queue.len()).rev() {
|
for message in messages {
|
||||||
if let Some(message) = self.delayed_queue.get(i) {
|
let mut buf = Vec::new();
|
||||||
if now < (message.created + self.additional_ping) {
|
let cursor = Cursor::new(&mut buf);
|
||||||
continue;
|
let encoder = GzEncoder::new(cursor, Compression::fast());
|
||||||
}
|
ciborium::into_writer(&message.package, encoder)
|
||||||
}
|
.map_err(|err| SendError::SerializationError(err))?;
|
||||||
if let Some(message) = self.delayed_queue.remove(i) {
|
if buf.len() > DATAGRAM_SIZE {
|
||||||
let mut buf = Vec::new();
|
return Err(SendError::DatagramTooLarge(buf.len()));
|
||||||
let cursor = Cursor::new(&mut buf);
|
|
||||||
let encoder = GzEncoder::new(cursor, Compression::fast());
|
|
||||||
ciborium::into_writer(&message.package, encoder)
|
|
||||||
.map_err(|err| SendError::SerializationError(err))?;
|
|
||||||
if buf.len() > DATAGRAM_SIZE {
|
|
||||||
return Err(SendError::DatagramTooLarge(buf.len()));
|
|
||||||
}
|
|
||||||
socket
|
|
||||||
.send_to(&buf, message.addr)
|
|
||||||
.map_err(|e| SendError::SendError(e))?;
|
|
||||||
}
|
}
|
||||||
|
socket
|
||||||
|
.send_to(&buf, message.addr)
|
||||||
|
.map_err(|e| SendError::SendError(e))?;
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
} else {
|
} else {
|
||||||
@ -576,7 +566,6 @@ pub struct Connection<T: Clone> {
|
|||||||
|
|
||||||
last_sent_ping: Instant,
|
last_sent_ping: Instant,
|
||||||
last_recv_ping: Instant,
|
last_recv_ping: Instant,
|
||||||
last_recv_pong: Instant,
|
|
||||||
last_recv_close: Instant,
|
last_recv_close: Instant,
|
||||||
closing_since: Instant,
|
closing_since: Instant,
|
||||||
|
|
||||||
@ -604,7 +593,6 @@ impl<T: Clone> Clone for Connection<T> {
|
|||||||
ping: self.ping.clone(),
|
ping: self.ping.clone(),
|
||||||
last_sent_ping: self.last_sent_ping.clone(),
|
last_sent_ping: self.last_sent_ping.clone(),
|
||||||
last_recv_ping: self.last_recv_ping.clone(),
|
last_recv_ping: self.last_recv_ping.clone(),
|
||||||
last_recv_pong: self.last_recv_pong.clone(),
|
|
||||||
last_recv_close: self.last_recv_close.clone(),
|
last_recv_close: self.last_recv_close.clone(),
|
||||||
closing_since: self.closing_since.clone(),
|
closing_since: self.closing_since.clone(),
|
||||||
|
|
||||||
@ -633,7 +621,6 @@ impl<T: Clone> Connection<T> {
|
|||||||
ping: Duration::default(),
|
ping: Duration::default(),
|
||||||
last_sent_ping: Instant::now() - Duration::from_hours(1),
|
last_sent_ping: Instant::now() - Duration::from_hours(1),
|
||||||
last_recv_ping: Instant::now(),
|
last_recv_ping: Instant::now(),
|
||||||
last_recv_pong: Instant::now(),
|
|
||||||
last_recv_close: Instant::now(),
|
last_recv_close: Instant::now(),
|
||||||
closing_since: Instant::now(),
|
closing_since: Instant::now(),
|
||||||
|
|
||||||
|
|||||||
20
src/lib.rs
20
src/lib.rs
@ -51,7 +51,6 @@ pub enum PeerMessage<T: Clone + Serialize + DeserializeOwned + Send + Sync + 'st
|
|||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct PeerConfig {
|
pub struct PeerConfig {
|
||||||
ping_interval: Duration,
|
ping_interval: Duration,
|
||||||
max_ping_interval: Duration,
|
|
||||||
timeout: Duration,
|
timeout: Duration,
|
||||||
disconnect_timeout: Duration,
|
disconnect_timeout: Duration,
|
||||||
message_retry: Duration,
|
message_retry: Duration,
|
||||||
@ -63,8 +62,7 @@ impl Default for PeerConfig {
|
|||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
ping_interval: Duration::from_millis(100),
|
ping_interval: Duration::from_millis(100),
|
||||||
max_ping_interval: Duration::from_millis(2000),
|
timeout: Duration::from_millis(2000),
|
||||||
timeout: Duration::from_millis(5000),
|
|
||||||
disconnect_timeout: Duration::from_millis(500),
|
disconnect_timeout: Duration::from_millis(500),
|
||||||
message_retry: Duration::from_millis(100),
|
message_retry: Duration::from_millis(100),
|
||||||
identifier: "ExamplePeer".to_string(),
|
identifier: "ExamplePeer".to_string(),
|
||||||
@ -74,8 +72,7 @@ impl Default for PeerConfig {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl PeerConfig {
|
impl PeerConfig {
|
||||||
/// Set the desired interval between "ping"s (assuming pongs arrive in due
|
/// Set the desired interval between "ping"s
|
||||||
/// time)
|
|
||||||
pub fn with_ping_interval(self, interval: Duration) -> PeerConfig {
|
pub fn with_ping_interval(self, interval: Duration) -> PeerConfig {
|
||||||
PeerConfig {
|
PeerConfig {
|
||||||
ping_interval: interval,
|
ping_interval: interval,
|
||||||
@ -83,15 +80,6 @@ impl PeerConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set the desired maximum interval between "ping"s (assuming pongs aren't
|
|
||||||
/// being received in due time)
|
|
||||||
pub fn with_max_ping_interval(self, interval: Duration) -> PeerConfig {
|
|
||||||
PeerConfig {
|
|
||||||
max_ping_interval: interval,
|
|
||||||
..self
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set the length of timeout; that is to say the duration of time which is
|
/// Set the length of timeout; that is to say the duration of time which is
|
||||||
/// acceptable to occur between pings before a Timeout-error occurs
|
/// acceptable to occur between pings before a Timeout-error occurs
|
||||||
pub fn with_timeout(self, timeout: Duration) -> PeerConfig {
|
pub fn with_timeout(self, timeout: Duration) -> PeerConfig {
|
||||||
@ -131,7 +119,7 @@ impl PeerConfig {
|
|||||||
/// connections.
|
/// connections.
|
||||||
pub fn with_additional_ping(self, ping: Duration) -> PeerConfig {
|
pub fn with_additional_ping(self, ping: Duration) -> PeerConfig {
|
||||||
PeerConfig {
|
PeerConfig {
|
||||||
additional_ping: ping / 2,
|
additional_ping: ping,
|
||||||
..self
|
..self
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -216,7 +204,7 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
|
|||||||
let now = Instant::now();
|
let now = Instant::now();
|
||||||
for i in (0..self.delayed_messages.len()).rev() {
|
for i in (0..self.delayed_messages.len()).rev() {
|
||||||
if let Some(msg) = self.delayed_messages.get(i) {
|
if let Some(msg) = self.delayed_messages.get(i) {
|
||||||
if now >= (msg.created + self.config.additional_ping) {
|
if (msg.created + self.config.additional_ping) > now {
|
||||||
if let Some(msg) = self.delayed_messages.remove(i) {
|
if let Some(msg) = self.delayed_messages.remove(i) {
|
||||||
self.messages.extend(self.connection_mgr.handle_package(
|
self.messages.extend(self.connection_mgr.handle_package(
|
||||||
msg.package,
|
msg.package,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user