Skip to content

Commit 12a1d1d

Browse files
committed
format
1 parent 70146dd commit 12a1d1d

File tree

2 files changed

+16
-13
lines changed

2 files changed

+16
-13
lines changed

src/bmp_collector.rs

+2-3
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,7 @@ use tokio::net::{TcpListener, TcpStream};
1111
use tokio::sync::mpsc;
1212
use tokio_util::codec::length_delimited::LengthDelimitedCodec;
1313
use zettabgp::bmp::prelude::{
14-
BmpMessagePeerDown, BmpMessagePeerHeader, BmpMessageRouteMonitoring,
15-
BmpMessageTermination,
14+
BmpMessagePeerDown, BmpMessagePeerHeader, BmpMessageRouteMonitoring, BmpMessageTermination,
1615
};
1716
use zettabgp::bmp::BmpMessage;
1817

@@ -172,7 +171,7 @@ pub async fn run_client(
172171
run_peer(client_addr, rm.peer.clone(), store)
173172
});
174173
channel.send(Ok(rm)).await.unwrap();
175-
},
174+
}
176175
BmpMessage::PeerUpNotification(n) => {
177176
channels.insert(n.peer.peeraddress, run_peer(client_addr, n.peer, store));
178177
}

src/main.rs

+14-10
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
use fernglas::*;
2+
use figment::providers::{Env, Format, Yaml};
3+
use figment::Figment;
24
use futures_util::future::{join_all, select_all};
35
use log::*;
46
use tokio::signal::unix::{signal, SignalKind};
5-
use figment::Figment;
6-
use figment::providers::{Yaml, Env, Format};
77

88
#[cfg(feature = "mimalloc")]
99
#[global_allocator]
@@ -42,14 +42,18 @@ async fn main() -> anyhow::Result<()> {
4242
shutdown_rx.clone(),
4343
)));
4444

45-
futures.extend(cfg.collectors.into_iter().map(|(_, collector)| match collector {
46-
CollectorConfig::Bmp(cfg) => {
47-
tokio::task::spawn(bmp_collector::run(cfg, store.clone(), shutdown_rx.clone()))
48-
}
49-
CollectorConfig::Bgp(cfg) => {
50-
tokio::task::spawn(bgp_collector::run(cfg, store.clone(), shutdown_rx.clone()))
51-
}
52-
}));
45+
futures.extend(
46+
cfg.collectors
47+
.into_iter()
48+
.map(|(_, collector)| match collector {
49+
CollectorConfig::Bmp(cfg) => {
50+
tokio::task::spawn(bmp_collector::run(cfg, store.clone(), shutdown_rx.clone()))
51+
}
52+
CollectorConfig::Bgp(cfg) => {
53+
tokio::task::spawn(bgp_collector::run(cfg, store.clone(), shutdown_rx.clone()))
54+
}
55+
}),
56+
);
5357

5458
let mut sigint = signal(SignalKind::interrupt())?;
5559
let mut sigterm = signal(SignalKind::terminate())?;

0 commit comments

Comments
 (0)