fix interval timings for concurrent pings
This commit is contained in:
parent
eccfe68530
commit
0448980dd9
2 changed files with 8 additions and 11 deletions
|
@ -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
|
||||||
|
|
17
src/ping.rs
17
src/ping.rs
|
@ -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…
Reference in a new issue