itsscb/embassy
1
0
Fork 0
mirror of https://github.com/embassy-rs/embassy.git synced 2025-09-26 20:00:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: configurable rate_limit for the ping utility

Browse Source
This commit is contained in:
skkeye 2025-02-09 23:16:14 -05:00 committed by Ulf Lilleengen
parent bdb1b81213
commit 6d7a5f949c
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
embassy-net/src/icmp.rs
View File

@ -407,8 +407,8 @@ pub mod ping {
// 1 sec min per ping
let rate_limit_end = rate_limit_start.elapsed();
if rate_limit_end <= Duration::from_secs(1) {
Timer::after(Duration::from_secs(1).checked_sub(rate_limit_end).unwrap()).await;
if rate_limit_end <= params.rate_limit {
Timer::after(params.rate_limit.checked_sub(rate_limit_end).unwrap()).await;
}
}
// calculate and return the average duration
@ -579,6 +579,7 @@ pub mod ping {
/// * `hop_limit` - The hop limit to be used by the socket.
/// * `count` - The number of pings to be sent in one ping operation.
/// * `timeout` - The timeout duration before returning a [`PingError::DestinationHostUnreachable`] error.
/// * `rate_limit` - The minimum time per echo request
pub struct PingParams<'a> {
target: Option<IpAddr>,
#[cfg(feature = "proto-ipv6")]
@ -587,6 +588,7 @@ pub mod ping {
hop_limit: Option<u8>,
count: u16,
timeout: Duration,
rate_limit: Duration,
}
impl Default for PingParams<'_> {
@ -599,6 +601,7 @@ pub mod ping {
hop_limit: None,
count: 4,
timeout: Duration::from_secs(4),
rate_limit: Duration::from_secs(1),
}
}
}
@ -614,6 +617,7 @@ pub mod ping {
hop_limit: None,
count: 4,
timeout: Duration::from_secs(4),
rate_limit: Duration::from_secs(1),
}
}
@ -701,5 +705,16 @@ pub mod ping {
pub fn timeout(&self) -> Duration {
self.timeout
}
/// Sets the `rate_limit`: minimum time per echo request
pub fn set_rate_limit(&mut self, rate_limit: Duration) -> &mut Self {
self.rate_limit = rate_limit;
self
}
/// Retrieve the rate_limit
pub fn rate_limit(&self) -> Duration {
self.rate_limit
}
}
}