perf: optimize store batch serialization, zero-clone semantic search, and lock contention

server/src/api/ws.rs

@@ -105,8 +105,8 @@ pub async fn handle_socket(socket: WebSocket, state: Arc<AppState>, _client_type
     });
 
     let handler = Arc::clone(&state.handler);
-    let state_clone = Arc::clone(&state);
     let session_id_clone = session_id.clone();
+    let response_tx = tx.clone();
 
     let recv_task = tokio::spawn(async move {
         while let Some(msg_result) = receiver.next().await {
@@ -121,24 +121,11 @@ pub async fn handle_socket(socket: WebSocket, state: Arc<AppState>, _client_type
                         // Process MCP request
                         if let Some(response) = handler.handle_request(payload).await {
                             let res_str = serde_json::to_string(&response).unwrap_or_else(|e| format!(r#"{{\"jsonrpc\":\"2.0\",\"id\":null,\"error\":{{\"code\":-32603,\"message\":\"{}\"}}}}"#, e));
-                            let tx_opt = state_clone
-                                .clients
-                                .read()
-                                .unwrap_or_else(|e| e.into_inner())
-                                .get(&session_id_clone)
-                                .cloned();
-                            if let Some(client_tx) = tx_opt {
-                                if let Err(e) = client_tx.send(res_str).await {
-                                    tracing::error!(
-                                        "Failed to send response to client channel for session {}: {}",
-                                        session_id_clone,
-                                        e
-                                    );
-                                }
-                            } else {
-                                tracing::warn!(
-                                    "Could not find client_tx for session_id {} when trying to send response",
-                                    session_id_clone
+                            if let Err(e) = response_tx.send(res_str).await {
+                                tracing::error!(
+                                    "Failed to send response to client channel for session {}: {}",
+                                    session_id_clone,
+                                    e
                                 );
                             }
                         }