From 7be7f59d5caf38ca0cd7a12a937a2cfdca0268d7 Mon Sep 17 00:00:00 2001
From: hozan23 <hozan23@karyontech.net>
Date: Thu, 23 May 2024 00:21:59 +0200
Subject: p2p: add rpc server to monitor example

---
 net/src/codec/websocket.rs  | 2 +-
 net/src/endpoint.rs         | 6 ++++++
 net/src/stream/websocket.rs | 5 ++++-
 3 files changed, 11 insertions(+), 2 deletions(-)

(limited to 'net')

diff --git a/net/src/codec/websocket.rs b/net/src/codec/websocket.rs
index b59a55c..8676810 100644
--- a/net/src/codec/websocket.rs
+++ b/net/src/codec/websocket.rs
@@ -19,5 +19,5 @@ pub trait WebSocketEncoder {
 
 pub trait WebSocketDecoder {
     type DeItem;
-    fn decode(&self, src: &Message) -> Result<Self::DeItem>;
+    fn decode(&self, src: &Message) -> Result<Option<Self::DeItem>>;
 }
diff --git a/net/src/endpoint.rs b/net/src/endpoint.rs
index 0c7ecd1..5aebdf9 100644
--- a/net/src/endpoint.rs
+++ b/net/src/endpoint.rs
@@ -238,6 +238,12 @@ impl ToEndpoint for String {
     }
 }
 
+impl ToEndpoint for Endpoint {
+    fn to_endpoint(&self) -> Result<Endpoint> {
+        Ok(self.clone())
+    }
+}
+
 impl ToEndpoint for &str {
     fn to_endpoint(&self) -> Result<Endpoint> {
         Endpoint::from_str(self)
diff --git a/net/src/stream/websocket.rs b/net/src/stream/websocket.rs
index 9d41626..2626d2f 100644
--- a/net/src/stream/websocket.rs
+++ b/net/src/stream/websocket.rs
@@ -47,7 +47,10 @@ where
 
     pub async fn recv(&mut self) -> Result<C::Item> {
         match self.inner.next().await {
-            Some(msg) => self.codec.decode(&msg?),
+            Some(msg) => match self.codec.decode(&msg?)? {
+                Some(m) => Ok(m),
+                None => todo!(),
+            },
             None => Err(Error::IO(std::io::ErrorKind::ConnectionAborted.into())),
         }
     }
-- 
cgit v1.2.3