add support for tokio & improve net crate api

1 files changed, 59 insertions, 0 deletions

diff --git a/net/examples/tcp_codec.rs b/net/examples/tcp_codec.rs
new file mode 100644
index 0000000..93deaae
--- /dev/null
+++ b/net/examples/tcp_codec.rs
@@ -0,0 +1,59 @@
+use std::time::Duration;
+
+use karyon_core::async_util::sleep;
+
+use karyon_net::{
+    codec::{Codec, Decoder, Encoder},
+    tcp, ConnListener, Connection, Endpoint, Result,
+};
+
+#[derive(Clone)]
+struct NewLineCodec {}
+
+impl Codec for NewLineCodec {
+    type Item = String;
+}
+
+impl Encoder for NewLineCodec {
+    type EnItem = String;
+    fn encode(&self, src: &Self::EnItem, dst: &mut [u8]) -> Result<usize> {
+        dst[..src.len()].copy_from_slice(src.as_bytes());
+        Ok(src.len())
+    }
+}
+
+impl Decoder for NewLineCodec {
+    type DeItem = String;
+    fn decode(&self, src: &mut [u8]) -> Result<Option<(usize, Self::DeItem)>> {
+        match src.iter().position(|&b| b == b'\n') {
+            Some(i) => Ok(Some((i + 1, String::from_utf8(src[..i].to_vec()).unwrap()))),
+            None => Ok(None),
+        }
+    }
+}
+
+fn main() {
+    smol::block_on(async {
+        let endpoint: Endpoint = "tcp://127.0.0.1:3000".parse().unwrap();
+
+        let config = tcp::TcpConfig::default();
+
+        let listener = tcp::listen(&endpoint, config.clone(), NewLineCodec {})
+            .await
+            .unwrap();
+        smol::spawn(async move {
+            if let Ok(conn) = listener.accept().await {
+                loop {
+                    let msg = conn.recv().await.unwrap();
+                    println!("Receive a message: {:?}", msg);
+                }
+            };
+        })
+        .detach();
+
+        let conn = tcp::dial(&endpoint, config, NewLineCodec {}).await.unwrap();
+        conn.send("hello".to_string()).await.unwrap();
+        conn.send(" world\n".to_string()).await.unwrap();
+        sleep(Duration::from_secs(1)).await;
+    });
+}