fix(mongodb): cursor timeout bug and warnings

Author: truongnguyen3

src/error.rs

@@ -2,6 +2,7 @@ use thiserror::Error;
 
 /// Errors that can occur during migration
 #[derive(Error, Debug)]
+#[allow(dead_code)]
 pub enum MigrationError {
     /// MongoDB connection error
     #[error("MongoDB connection error: {0}")]
@@ -53,8 +54,10 @@ pub enum MigrationError {
 }
 
 /// Result type alias for migration operations
+#[allow(dead_code)]
 pub type MigrationResult<T> = Result<T, MigrationError>;
 
+#[allow(dead_code)]
 impl MigrationError {
     /// Create a new schema inference error
     pub fn schema_inference<S: Into<String>>(msg: S) -> Self {

src/libsql_client.rs

@@ -5,13 +5,16 @@ use tracing::{debug, info};
 
 /// LibSQL client wrapper supporting both local and remote (Turso) databases
 pub struct LibSqlClient {
+    #[allow(dead_code)]
     database: Database,
     connection: Connection,
+    #[allow(dead_code)]
     mode: ConnectionMode,
 }
 
 /// Connection mode for LibSQL
 #[derive(Debug, Clone)]
+#[allow(dead_code)]
 enum ConnectionMode {
     Local(String),
     Remote { url: String },
@@ -105,6 +108,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// Total number of rows affected
+    #[allow(dead_code)]
     pub async fn execute_batch(&self, statements: Vec<String>) -> Result<u64> {
         debug!("Executing batch of {} statements", statements.len());
 
@@ -143,6 +147,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// Total number of rows inserted
+    #[allow(dead_code)]
     pub async fn execute_batch_inserts<P>(
         &self,
         sql: &str,
@@ -198,6 +203,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// Rows result set
+    #[allow(dead_code)]
     pub async fn query(&self, sql: &str) -> Result<libsql::Rows> {
         debug!("Querying: {}", sql);
         let rows = self.connection.query(sql, ()).await?;
@@ -208,6 +214,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// String describing the connection mode
+    #[allow(dead_code)]
     pub fn connection_info(&self) -> String {
         match &self.mode {
             ConnectionMode::Local(path) => format!("Local file: {}", path),
@@ -219,6 +226,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// True if using local SQLite file, false if using Turso
+    #[allow(dead_code)]
     pub fn is_local(&self) -> bool {
         matches!(self.mode, ConnectionMode::Local(_))
     }
@@ -227,6 +235,7 @@ impl LibSqlClient {
     ///
     /// # Returns
     /// Optional path to the local SQLite file
+    #[allow(dead_code)]
     pub fn output_path(&self) -> Option<String> {
         match &self.mode {
             ConnectionMode::Local(path) => Some(path.clone()),

src/mongodb_client.rs

@@ -135,7 +135,13 @@ impl MongoClient {
         let db = self.client.database(database_name);
         let collection = db.collection::<Document>(collection_name);
 
-        let cursor = collection.find(doc! {}, None).await?;
+        // Configure find options to prevent cursor timeout
+        let find_options = mongodb::options::FindOptions::builder()
+            .no_cursor_timeout(true)  // Prevent 10-minute cursor timeout
+            .batch_size(1000)          // Process in batches
+            .build();
+
+        let cursor = collection.find(doc! {}, find_options).await?;
 
         Ok(cursor)
     }
@@ -147,6 +153,7 @@ impl MongoClient {
     ///
     /// # Returns
     /// True if the database exists, false otherwise
+    #[allow(dead_code)]
     pub async fn database_exists(&self, database_name: &str) -> Result<bool> {
         let db_names = self.client.list_database_names(doc! {}, None).await?;
         Ok(db_names.contains(&database_name.to_string()))
@@ -160,6 +167,7 @@ impl MongoClient {
     ///
     /// # Returns
     /// True if the collection exists, false otherwise
+    #[allow(dead_code)]
     pub async fn collection_exists(
         &self,
         database_name: &str,
@@ -173,6 +181,7 @@ impl MongoClient {
     ///
     /// # Returns
     /// Reference to the MongoDB client
+    #[allow(dead_code)]
     pub fn client(&self) -> &Client {
         &self.client
     }