From e3d1f4fd91a5f077fda8a1976e194c378ee166d0 Mon Sep 17 00:00:00 2001
From: hozan23 <hozan23@karyontech.net>
Date: Mon, 24 Jun 2024 13:25:39 +0200
Subject: p2p/monitor: use struct instead of enum for monitor events

---
 p2p/src/peer_pool.rs | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'p2p/src/peer_pool.rs')

diff --git a/p2p/src/peer_pool.rs b/p2p/src/peer_pool.rs
index 79fd4b4..6c895a0 100644
--- a/p2p/src/peer_pool.rs
+++ b/p2p/src/peer_pool.rs
@@ -23,7 +23,7 @@ use crate::{
     config::Config,
     conn_queue::{ConnDirection, ConnQueue},
     message::{get_msg_payload, NetMsg, NetMsgCmd, VerAckMsg, VerMsg},
-    monitor::{Monitor, PeerPoolEvent},
+    monitor::{Monitor, PPEvent},
     peer::{ArcPeer, Peer, PeerID},
     protocol::{Protocol, ProtocolConstructor, ProtocolID},
     protocols::PingProtocol,
@@ -192,9 +192,7 @@ impl PeerPool {
 
         info!("Add new peer {pid}, direction: {conn_direction}, endpoint: {endpoint}");
 
-        self.monitor
-            .notify(PeerPoolEvent::NewPeer(pid.clone()))
-            .await;
+        self.monitor.notify(PPEvent::NewPeer(pid.clone())).await;
 
         Ok(())
     }
@@ -215,9 +213,7 @@ impl PeerPool {
 
         peer.shutdown().await;
 
-        self.monitor
-            .notify(PeerPoolEvent::RemovePeer(pid.clone()))
-            .await;
+        self.monitor.notify(PPEvent::RemovePeer(pid.clone())).await;
 
         let endpoint = peer.remote_endpoint();
         let direction = peer.direction();
-- 
cgit v1.2.3