riz/mcp-memory
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: implement native UDP ping/pong health check pattern

Browse Source
This commit is contained in:
Riz Ashraf committed 2026-10-08 12:46:55 +01:00
1 parent 0da84c804c
commit 8c6bf05269
1 file changed
+6 -4
Download Patch File Download Diff File Expand all files Collapse all files
server/src/lib.rs
+6 -4
View File
@@ -368,11 +368,13 @@ pub async fn run_server(state: Arc<MemoryState>) -> Result<(), Box<dyn std::erro
for tx in senders { for tx in senders {
let _ = tx.try_send(msg_str.clone()); let _ = tx.try_send(msg_str.clone());
} }
} else if let Ok(gate_wait) = serde_json::from_slice::<serde_json::Value>(&buf[..len]) { } else if let Ok(json_payload) = serde_json::from_slice::<serde_json::Value>(&buf[..len]) {
if gate_wait.get("type").and_then(|t| t.as_str()) == Some("gate_wait") { if json_payload.get("type").and_then(|t| t.as_str()) == Some("ping") {
let _ = socket.send_to(b"pong", addr).await;
} else if json_payload.get("type").and_then(|t| t.as_str()) == Some("gate_wait") {
if let (Some(action), Some(target)) = ( if let (Some(action), Some(target)) = (
gate_wait.get("action").and_then(|a| a.as_str()), json_payload.get("action").and_then(|a| a.as_str()),
gate_wait.get("target").and_then(|t| t.as_str()) json_payload.get("target").and_then(|t| t.as_str())
) { ) {
subscribers.insert((action.to_string(), target.to_string()), addr); subscribers.insert((action.to_string(), target.to_string()), addr);
} }