feat: Send pong

Cogru · May 23, 2024 · 511ce95 · 511ce95
1 parent a2922f6
commit 511ce95
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 6 deletions.
diff --git a/src/handler/mod.rs b/src/handler/mod.rs
@@ -18,6 +18,9 @@ pub async fn handle(connection: &mut server::Connection, json: &str) {
     println!("{}: {:?}", "val", val["method"]);
 
     match method {
+        "ping" => {
+            ping::handle(connection, &val).await;
+        }
         "enter" => {
             enter::handle(connection, &val).await;
         }
@@ -30,15 +33,29 @@ pub async fn handle(connection: &mut server::Connection, json: &str) {
     }
 }
 
+/// Ping pong
+mod ping {
+    use crate::server;
+
+    pub async fn handle(connection: &mut server::Connection, json: &serde_json::Value) {
+        tracing::trace!("method: {:?}", json["method"]);
+        connection
+            .send(serde_json::json!({
+                "method": "pong",
+            }))
+            .await;
+    }
+}
+
 /// Enter session
 mod enter {
     use crate::server;
 
     pub async fn handle(connection: &mut server::Connection, json: &serde_json::Value) {
-        tracing::trace!("method: {:?}", json["method"]);
-        connection.send(serde_json::json!({
-            "jsonrpc": "2.0",
-            "method": "enter",
-        })).await;
+        connection
+            .send(serde_json::json!({
+                "method": "enter",
+            }))
+            .await;
     }
 }
diff --git a/src/server.rs b/src/server.rs
@@ -148,7 +148,8 @@ impl Connection {
     }
 
     pub async fn send(&mut self, params: serde_json::Value) {
-        let data_str = params.to_string();
+        let json_str = params.to_string();
+        let data_str = format!("Content-Length: {}\r\n\r\n{}", json_str.len(), json_str);
         let data = data_str.as_bytes();
         self.write(&data).await;
     }