fix interval timings for concurrent pings

main
Jan Christian Grünhage 5 years ago
parent eccfe68530
commit 0448980dd9
No known key found for this signature in database
GPG Key ID: 6594C449C633D10C

@ -5,6 +5,8 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html). and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
## [Unreleased] ## [Unreleased]
### Fixed
- fix interval timings concurrent pings
## [v0.2.2] - 2020-04-06 ## [v0.2.2] - 2020-04-06
### Added ### Added

@ -18,13 +18,12 @@
* along with this program. If not, see <https://www.gnu.org/licenses/>. * * along with this program. If not, see <https://www.gnu.org/licenses/>. *
********************************************************************************/ ********************************************************************************/
use crate::config::Config; use crate::config::Config;
use futures_util::stream::StreamExt;
use lazy_static::lazy_static; use lazy_static::lazy_static;
use log::{error, info, trace}; use log::{error, info, trace};
use prometheus::*; use prometheus::*;
use std::net::IpAddr; use std::net::IpAddr;
use std::time::Duration; use std::time::Duration;
use tokio_ping::Pinger; use tokio_ping::{PingFuture, Pinger};
lazy_static! { lazy_static! {
static ref PING_HISTOGRAM: HistogramVec = register_histogram_vec!( static ref PING_HISTOGRAM: HistogramVec = register_histogram_vec!(
@ -58,21 +57,17 @@ pub(crate) async fn start_pinging_hosts(
} }
async fn ping_host(pinger: Pinger, host: IpAddr, interval: u64) { async fn ping_host(pinger: Pinger, host: IpAddr, interval: u64) {
let pingchain = pinger.chain(host).timeout(Duration::from_secs(3)); let mut pingchain = pinger.chain(host).timeout(Duration::from_secs(3));
let host = host.to_string();
let mut stream = pingchain.stream();
let mut interval = tokio::time::interval(Duration::from_millis(interval)); let mut interval = tokio::time::interval(Duration::from_millis(interval));
let host_string = host.to_string();
loop { loop {
interval.tick().await; interval.tick().await;
handle_ping_result(stream.next().await.unwrap(), &host).await; tokio::spawn(handle_ping_result(pingchain.send(), host_string.clone()));
} }
} }
async fn handle_ping_result( async fn handle_ping_result(result: PingFuture, host: String) {
result: std::result::Result<Option<Duration>, tokio_ping::Error>, let pong = match result.await {
host: &str,
) {
let pong = match result {
Ok(pong) => pong, Ok(pong) => pong,
Err(error) => { Err(error) => {
error!("Couldn't ping {}: {}", &host, error); error!("Couldn't ping {}: {}", &host, error);

Loading…
Cancel
Save