refactor: merge set_pool and start_reconnect_task into set_pool_and_start_reconnect_task

src/connection.rs

@@ -180,6 +180,13 @@ impl ConnectionManager {
         self.pool = Some(pool);
     }
 
+    pub fn set_pool_and_start_reconnect_task(&mut self, pool: Arc<sqlx::PgPool>) {
+        self.pool = Some(pool.clone());
+        // 将 self 转换为不可变引用以调用 start_reconnect_task
+        let manager = self.clone();
+        manager.start_reconnect_task();
+    }
+
     pub fn set_reconnect_tx(&mut self, tx: tokio::sync::mpsc::UnboundedSender<Uuid>) {
         self.reconnect_tx = Some(tx);
     }

src/main.rs

@@ -46,12 +46,11 @@ async fn main() {
         Some(ws_manager.clone()),
     ));
     connection_manager.set_event_manager(event_manager.clone());
-    connection_manager.set_pool(Arc::new(pool.clone()));
+    connection_manager.set_pool_and_start_reconnect_task(Arc::new(pool.clone()));
     
     let connection_manager = Arc::new(connection_manager);
     
-    // 启动重连任务
-    connection_manager.start_reconnect_task();
+
 
     // Connect to all enabled sources concurrently
     let sources = service::get_all_enabled_sources(&pool)