Fix database_is_empty() checking wrong database and add timeout

- Connect to target_db_url instead of target_url to check if correct
  database is empty (was checking default db, not target db)
- Add 30-second timeout to prevent indefinite hang on stale serverless
  connections (SerenDB connection pool exhaustion)

Fixes #25
Fixes #26

Author: taariq

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "database-replicator"
-version = "5.3.18"
+version = "5.3.19"
 edition = "2021"
 license = "Apache-2.0"
 description = "Universal database-to-PostgreSQL replication CLI. Supports PostgreSQL, SQLite, MongoDB, and MySQL."

src/commands/init.rs

@@ -405,8 +405,15 @@ pub async fn init(
                                 db_info.name
                             );
 
-                            // Check if empty (reuse existing connection to avoid pool exhaustion)
-                            if database_is_empty(&target_client).await? {
+                            // Check if empty by connecting to the SPECIFIC database
+                            // (target_client is connected to default db, not the target db)
+                            let is_empty = {
+                                let db_client =
+                                    postgres::connect_with_retry(&target_db_url).await?;
+                                database_is_empty(&db_client).await?
+                            }; // db_client dropped here
+
+                            if is_empty {
                                 tracing::info!(
                                     "  Database '{}' is empty, proceeding with restore",
                                     db_info.name
@@ -676,18 +683,25 @@ fn confirm_replication(sizes: &[migration::DatabaseSizeInfo]) -> Result<bool> {
     Ok(input.trim().to_lowercase() == "y")
 }
 
-/// Checks if a database is empty (no user tables)
+/// Checks if the currently connected database is empty (has no user tables).
 ///
-/// Uses the existing connection to avoid connection pool exhaustion on
-/// serverless PostgreSQL providers (SerenDB, Neon) that have strict limits.
+/// Includes a 30-second timeout to prevent hanging on stale serverless connections.
 async fn database_is_empty(client: &tokio_postgres::Client) -> Result<bool> {
     let query = "
         SELECT COUNT(*)
         FROM information_schema.tables
         WHERE table_schema NOT IN ('pg_catalog', 'information_schema')
     ";
 
-    let row = client.query_one(query, &[]).await?;
+    // Add timeout to prevent indefinite hang on stale serverless connections
+    let row = tokio::time::timeout(
+        std::time::Duration::from_secs(30),
+        client.query_one(query, &[]),
+    )
+    .await
+    .context("database_is_empty query timed out after 30 seconds")?
+    .context("Failed to query information_schema.tables")?;
+
     let count: i64 = row.get(0);
 
     Ok(count == 0)