Update src/mcp.rs

src/mcp.rs

@@ -4,7 +4,6 @@ use anyhow::Result;
 use axum::{
     extract::State,
     response::sse::{Event, KeepAlive, Sse},
-    response::IntoResponse,
     routing::{get, post},
     Json, Router,
 };
@@ -17,7 +16,8 @@ use std::path::Path;
 use std::sync::Arc;
 use std::time::Duration;
 use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
-use tracing::{error, info};
+use tokio_stream::StreamExt;
+use tracing::info;
 use walkdir::WalkDir;
 
 #[derive(Serialize, Deserialize, Debug, Clone)]
@@ -198,7 +198,6 @@ impl McpServer {
                     return Err(anyhow::anyhow!("Directory does not exist: {:?}", dir_path));
                 }
 
-                // 1. Collect all valid candidate files
                 let paths: Vec<_> = WalkDir::new(&dir_path)
                     .into_iter()
                     .filter_map(|e| e.ok())
@@ -208,7 +207,6 @@ impl McpServer {
 
                 let total_found = paths.len();
 
-                // 2. Parallel ingestion with Rayon
                 let parsed_docs: Vec<ParsedDocument> = paths
                     .par_iter()
                     .filter_map(|path| match DocumentParser::parse_file(path) {
@@ -221,8 +219,6 @@ impl McpServer {
                     .collect();
 
                 let parsed_count = parsed_docs.len();
-
-                // 3. Batch commit to Tantivy engine
                 let indexed_count = self.engine.add_documents(collection, parsed_docs).await?;
 
                 Ok(json!({
@@ -292,7 +288,6 @@ impl McpServer {
         }
     }
 
-    /// Runs the server loop over standard I/O (STDIO) transport
     pub async fn run_stdio(self: Arc<Self>) -> Result<()> {
         let stdin = tokio::io::stdin();
         let mut reader = BufReader::new(stdin).lines();
@@ -323,7 +318,6 @@ impl McpServer {
         Ok(())
     }
 
-    /// Runs the server over Axum SSE/HTTP transport
     pub async fn run_http(self: Arc<Self>, host: &str, port: u16) -> Result<()> {
         let app = Router::new()
             .route("/rpc", post(handle_http_rpc))
@@ -351,11 +345,14 @@ async fn handle_http_rpc(
 async fn handle_sse(
     State(_server): State<Arc<McpServer>>,
 ) -> Sse<impl Stream<Item = Result<Event, Infallible>>> {
-    let stream = tokio_stream::iter(vec![
-        Ok(Event::default().event("endpoint").data("/rpc"))
-    ])
-    .chain(tokio_stream::wrappers::IntervalStream::new(tokio::time::interval(Duration::from_secs(15)))
-        .map(|_| Ok(Event::default().comment("keep-alive"))));
+    let initial = tokio_stream::iter(vec![
+        Ok(Event::default().event("endpoint").data("/rpc")),
+    ]);
+
+    let interval = tokio_stream::wrappers::IntervalStream::new(tokio::time::interval(Duration::from_secs(15)))
+        .map(|_| Ok(Event::default().comment("keep-alive")));
+
+    let stream = initial.chain(interval);
 
     Sse::new(stream).keep_alive(KeepAlive::default())
 }