Add ping/pong to websocket

CympleTech · May 15, 2024 · bd2b11f · bd2b11f
1 parent a46839f
commit bd2b11f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-[![crate](https://img.shields.io/badge/crates.io-v0.10.3-green.svg)](https://crates.io/crates/tdn) [![doc](https://img.shields.io/badge/docs.rs-v0.10.3-blue.svg)](https://docs.rs/tdn)
+[![crate](https://img.shields.io/badge/crates.io-v0.10.6-green.svg)](https://crates.io/crates/tdn) [![doc](https://img.shields.io/badge/docs.rs-v0.10.6-blue.svg)](https://docs.rs/tdn)
 
 <h1 align="center"><img src="https://cympletech.com/logo/tdn_words.png" alt="TDN"></h1>
 

diff --git a/tdn/Cargo.toml b/tdn/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "tdn"
-version = "0.10.5"
+version = "0.10.6"
 authors = ["Dev <[email protected]>"]
 edition = "2021"
 readme = "../README.md"

diff --git a/tdn/src/rpc/ws.rs b/tdn/src/rpc/ws.rs
@@ -70,6 +70,12 @@ async fn ws_connection(
             }
             Some(FutureResult::Stream(msg)) => {
                 let msg = msg.to_text().unwrap();
+                if msg == "ping" {
+                    let s = WsMessage::from("pong".to_owned());
+                    let _ = writer.send(s).await;
+                    continue;
+                }
+
                 match parse_jsonrpc(msg.to_owned()) {
                     Ok(rpc_param) => {
                         send.send(RpcMessage::Request(id, rpc_param, None)).await?;