feat: 添加 SessionLifecycleService 模块，重构会话管理逻辑以优化会话生命周期处理

Co-authored-by: Copilot <copilot@github.com>

src/gateway/mod.rs

@@ -12,6 +12,7 @@ pub mod prompt;
 pub mod prompt_injector;
 pub mod session;
 pub mod session_factory;
+pub mod session_lifecycle;
 pub mod session_pool;
 pub mod tool_registry_factory;
 pub mod ws;

src/gateway/session.rs

@@ -31,7 +31,7 @@ use super::memory_maintenance::{
 };
 use super::prompt_injector::PromptInjector;
 use super::session_factory::SessionFactory;
-use super::session_pool::SessionPool;
+use super::session_lifecycle::SessionLifecycleService;
 use super::tool_registry_factory::ToolRegistryFactory;
 
 fn preview_text(content: &str, max_chars: usize) -> String {
@@ -455,7 +455,7 @@ pub struct SessionManager {
     skills: Arc<SkillRuntime>,
     store: Arc<SessionStore>,
     show_tool_results: bool,
-    session_pool: SessionPool,
+    lifecycle: SessionLifecycleService,
     cli_sessions: CliSessionService,
 }
 
@@ -499,7 +499,7 @@ impl SessionManager {
             prompt_injector,
             store.clone(),
         );
-        let session_pool = SessionPool::new(session_ttl_hours, session_factory);
+        let lifecycle = SessionLifecycleService::new(session_ttl_hours, session_factory);
         let cli_sessions = CliSessionService::new(store.clone());
 
         Ok(Self {
@@ -509,7 +509,7 @@ impl SessionManager {
             skills,
             store,
             show_tool_results,
-            session_pool,
+            lifecycle,
             cli_sessions,
         })
     }
@@ -569,21 +569,21 @@ impl SessionManager {
 
     /// 确保 session 存在且未超时，超时则重建
     pub async fn ensure_session(&self, channel_name: &str) -> Result<(), AgentError> {
-        self.session_pool.ensure_session(channel_name).await
+        self.lifecycle.ensure_session(channel_name).await
     }
 
     /// 获取 session（不检查超时）
     pub async fn get(&self, channel_name: &str) -> Option<Arc<Mutex<Session>>> {
-        self.session_pool.get(channel_name).await
+        self.lifecycle.get(channel_name).await
     }
 
     /// 更新最后活跃时间
     pub async fn touch(&self, channel_name: &str) {
-        self.session_pool.touch(channel_name).await;
+        self.lifecycle.touch(channel_name).await;
     }
 
     pub async fn cleanup_expired_sessions(&self) -> usize {
-        self.session_pool.cleanup_expired_sessions().await
+        self.lifecycle.cleanup_expired_sessions().await
     }
 
     /// 处理消息：路由到对应 session 的 agent
@@ -610,17 +610,7 @@ impl SessionManager {
             }
         }
 
-        // 确保 session 存在（可能需要重建）
+        let session = self.lifecycle.active_session(channel_name).await?;
-        self.ensure_session(channel_name).await?;
-
-        // 更新活跃时间
-        self.touch(channel_name).await;
-
-        // 获取 session
-        let session = self
-            .get(channel_name)
-            .await
-            .ok_or_else(|| AgentError::Other("Session not found".to_string()))?;
 
         let outbound_messages = AgentExecutionService::new(self.show_tool_results)
             .prepare_and_execute_message(MessageExecutionRequest {
@@ -652,13 +642,7 @@ impl SessionManager {
         prompt: &str,
         options: ScheduledAgentTaskOptions,
     ) -> Result<Vec<OutboundMessage>, AgentError> {
-        self.ensure_session(channel_name).await?;
+        let session = self.lifecycle.active_session(channel_name).await?;
-        self.touch(channel_name).await;
-
-        let session = self
-            .get(channel_name)
-            .await
-            .ok_or_else(|| AgentError::Other("Session not found".to_string()))?;
 
         let sender_id = options
             .sender_id

src/gateway/session_lifecycle.rs

@@ -0,0 +1,49 @@
+use std::sync::Arc;
+
+use tokio::sync::Mutex;
+
+use crate::agent::AgentError;
+
+use super::session::Session;
+use super::session_factory::SessionFactory;
+use super::session_pool::SessionPool;
+
+#[derive(Clone)]
+pub(crate) struct SessionLifecycleService {
+    session_pool: SessionPool,
+}
+
+impl SessionLifecycleService {
+    pub(crate) fn new(session_ttl_hours: u64, session_factory: SessionFactory) -> Self {
+        Self {
+            session_pool: SessionPool::new(session_ttl_hours, session_factory),
+        }
+    }
+
+    pub(crate) async fn ensure_session(&self, channel_name: &str) -> Result<(), AgentError> {
+        self.session_pool.ensure_session(channel_name).await
+    }
+
+    pub(crate) async fn get(&self, channel_name: &str) -> Option<Arc<Mutex<Session>>> {
+        self.session_pool.get(channel_name).await
+    }
+
+    pub(crate) async fn touch(&self, channel_name: &str) {
+        self.session_pool.touch(channel_name).await;
+    }
+
+    pub(crate) async fn active_session(
+        &self,
+        channel_name: &str,
+    ) -> Result<Arc<Mutex<Session>>, AgentError> {
+        self.ensure_session(channel_name).await?;
+        self.touch(channel_name).await;
+        self.get(channel_name)
+            .await
+            .ok_or_else(|| AgentError::Other("Session not found".to_string()))
+    }
+
+    pub(crate) async fn cleanup_expired_sessions(&self) -> usize {
+        self.session_pool.cleanup_expired_sessions().await
+    }
+}