a629486ad3
后端新增 load_messages_for_topic_page（seq < cursor + limit 分页，走 (session_id, seq) 索引替代 OFFSET 深翻页），ChatMessage 增加 seq 游标；历史批次消息带 topic_id 下发，前端以 seq+topic_id 双重判定批次归属，规避切话题瞬间在途旧批次污染。 前端触顶增量加载：批次缓存在 pendingHistoryRef，收到 topic_history_end 一次性去重 prepend，scrollTop 按新增高度补偿锚定原头部消息；不足一屏自动补页，loading 超时 10s 自愈；流式输出中加载历史不清空流式累加器，避免已流出文本丢失。
343 lines
14 KiB
Rust
343 lines
14 KiB
Rust
use crate::command::Command;
|
||
use crate::command::adapter::{AdapterError, InputAdapter, OutputAdapter};
|
||
use crate::command::context::AdapterContext;
|
||
use crate::command::response::{CommandResponse, MessageKind};
|
||
use crate::protocol::WsOutbound;
|
||
|
||
/// WebSocket 输入适配器
|
||
///
|
||
/// 将 WebSocket 的 JSON 输入直接反序列化为 Command
|
||
pub struct WebSocketInputAdapter;
|
||
|
||
impl WebSocketInputAdapter {
|
||
/// 创建新的 WebSocket 输入适配器
|
||
pub fn new() -> Self {
|
||
Self
|
||
}
|
||
}
|
||
|
||
impl Default for WebSocketInputAdapter {
|
||
fn default() -> Self {
|
||
Self::new()
|
||
}
|
||
}
|
||
|
||
impl InputAdapter for WebSocketInputAdapter {
|
||
fn try_parse(
|
||
&self,
|
||
input: &str,
|
||
_ctx: AdapterContext,
|
||
) -> Result<Option<Command>, AdapterError> {
|
||
// 尝试将 JSON 反序列化为 Command
|
||
// 如果失败，说明不是 Command 消息，返回 None
|
||
match serde_json::from_str(input) {
|
||
Ok(cmd) => Ok(Some(cmd)),
|
||
Err(_) => Ok(None),
|
||
}
|
||
}
|
||
}
|
||
|
||
/// WebSocket 输出适配器
|
||
///
|
||
/// 将 CommandResponse 转换为 WsOutbound 消息列表
|
||
pub struct WebSocketOutputAdapter;
|
||
|
||
impl WebSocketOutputAdapter {
|
||
/// 创建新的 WebSocket 输出适配器
|
||
pub fn new() -> Self {
|
||
Self
|
||
}
|
||
}
|
||
|
||
impl Default for WebSocketOutputAdapter {
|
||
fn default() -> Self {
|
||
Self::new()
|
||
}
|
||
}
|
||
|
||
impl OutputAdapter for WebSocketOutputAdapter {
|
||
type Output = Vec<WsOutbound>;
|
||
|
||
fn adapt(&self, response: CommandResponse) -> Vec<WsOutbound> {
|
||
let mut outbounds = Vec::new();
|
||
|
||
// 如果出错，返回错误消息
|
||
if let Some(error) = response.error {
|
||
outbounds.push(WsOutbound::Error {
|
||
code: error.code,
|
||
message: error.message,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
subagent_task_id: None,
|
||
});
|
||
return outbounds;
|
||
}
|
||
|
||
// 转换响应消息为 WsOutbound
|
||
for msg in &response.messages {
|
||
let outbound = match msg.kind {
|
||
MessageKind::Text => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
MessageKind::Notification => {
|
||
// 根据元数据判断具体类型
|
||
// 优先识别话题重命名（同时含 topics + topic_id + title）
|
||
if let (Some(topics_json), Some(topic_id), Some(title)) = (
|
||
response.metadata.get("topics"),
|
||
response.metadata.get("topic_id"),
|
||
response.metadata.get("title"),
|
||
) {
|
||
match serde_json::from_str::<Vec<crate::protocol::TopicSummary>>(
|
||
topics_json,
|
||
) {
|
||
Ok(topics) => {
|
||
let session_id = response
|
||
.metadata
|
||
.get("session_id")
|
||
.cloned()
|
||
.unwrap_or_default();
|
||
WsOutbound::TopicRenamed {
|
||
topics,
|
||
session_id,
|
||
topic_id: topic_id.clone(),
|
||
title: title.clone(),
|
||
}
|
||
}
|
||
Err(_) => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
}
|
||
} else if let Some(topics_json) = response.metadata.get("topics") {
|
||
// Topic 列表响应 - 优先检查 topics
|
||
match serde_json::from_str::<Vec<crate::protocol::TopicSummary>>(
|
||
topics_json,
|
||
) {
|
||
Ok(topics) => {
|
||
let session_id = response
|
||
.metadata
|
||
.get("session_id")
|
||
.cloned()
|
||
.unwrap_or_default();
|
||
WsOutbound::TopicList { topics, session_id }
|
||
}
|
||
Err(_) => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
}
|
||
} else if let Some(session_id) = response.metadata.get("session_id") {
|
||
// 有 session_id 但没有 topic_id 的是创建会话
|
||
if response.metadata.get("topic_id").is_none() {
|
||
WsOutbound::SessionCreated {
|
||
session_id: session_id.clone(),
|
||
title: msg.content.clone(),
|
||
}
|
||
} else {
|
||
// 加载会话
|
||
let message_count = response
|
||
.metadata
|
||
.get("message_count")
|
||
.and_then(|s| s.parse().ok())
|
||
.unwrap_or(0);
|
||
WsOutbound::SessionLoaded {
|
||
session_id: session_id.clone(),
|
||
title: msg.content.clone(),
|
||
message_count,
|
||
}
|
||
}
|
||
} else if let Some(topic_id) = response.metadata.get("topic_id") {
|
||
// 只有 topic_id，可能是加载话题
|
||
let message_count = response
|
||
.metadata
|
||
.get("message_count")
|
||
.and_then(|s| s.parse().ok())
|
||
.unwrap_or(0);
|
||
WsOutbound::SessionLoaded {
|
||
session_id: topic_id.clone(),
|
||
title: msg.content.clone(),
|
||
message_count,
|
||
}
|
||
} else if let Some(channels_json) = response.metadata.get("channels") {
|
||
// 通道列表响应
|
||
match serde_json::from_str::<Vec<crate::protocol::Channel>>(channels_json) {
|
||
Ok(channels) => WsOutbound::ChannelList { channels },
|
||
Err(_) => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
}
|
||
} else if let Some(sessions_json) = response.metadata.get("sessions") {
|
||
// 会话列表响应
|
||
match serde_json::from_str::<Vec<crate::protocol::SessionSummary>>(
|
||
sessions_json,
|
||
) {
|
||
Ok(sessions) => {
|
||
let channel_name = response.metadata.get("channel_name").cloned();
|
||
WsOutbound::SessionList {
|
||
sessions,
|
||
current_session_id: None,
|
||
channel_name,
|
||
}
|
||
}
|
||
Err(_) => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
}
|
||
} else if let Some(topics_json) = response.metadata.get("topics") {
|
||
// Topic 列表响应
|
||
match serde_json::from_str::<Vec<crate::protocol::TopicSummary>>(
|
||
topics_json,
|
||
) {
|
||
Ok(topics) => {
|
||
let session_id = response
|
||
.metadata
|
||
.get("session_id")
|
||
.cloned()
|
||
.unwrap_or_default();
|
||
WsOutbound::TopicList { topics, session_id }
|
||
}
|
||
Err(_) => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
}
|
||
} else {
|
||
// 默认通知
|
||
WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
}
|
||
}
|
||
}
|
||
MessageKind::Error => WsOutbound::Error {
|
||
code: "RESPONSE_ERROR".to_string(),
|
||
message: msg.content.clone(),
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
subagent_task_id: None,
|
||
},
|
||
_ => WsOutbound::AssistantResponse {
|
||
id: response.request_id.to_string(),
|
||
content: msg.content.clone(),
|
||
role: "assistant".to_string(),
|
||
attachments: Vec::new(),
|
||
subagent_task_id: None,
|
||
topic_id: None,
|
||
timestamp: Some(crate::protocol::now_timestamp()),
|
||
reasoning_content: None,
|
||
user_message_id: None,
|
||
seq: None,
|
||
},
|
||
};
|
||
outbounds.push(outbound);
|
||
}
|
||
|
||
outbounds
|
||
}
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn test_websocket_input_adapter_valid_command() {
|
||
let adapter = WebSocketInputAdapter::new();
|
||
let ctx = AdapterContext::new("test");
|
||
|
||
let json = r#"{"type":"create_session","title":"my session"}"#;
|
||
let result = adapter.try_parse(json, ctx).unwrap();
|
||
|
||
assert!(result.is_some());
|
||
let cmd = result.unwrap();
|
||
assert!(matches!(
|
||
cmd,
|
||
Command::CreateSession {
|
||
title: Some(ref t)
|
||
} if t == "my session"
|
||
));
|
||
}
|
||
|
||
#[test]
|
||
fn test_websocket_input_adapter_invalid_json() {
|
||
let adapter = WebSocketInputAdapter::new();
|
||
let ctx = AdapterContext::new("test");
|
||
|
||
let json = "not a command";
|
||
let result = adapter.try_parse(json, ctx).unwrap();
|
||
|
||
assert!(result.is_none());
|
||
}
|
||
|
||
#[test]
|
||
fn test_websocket_output_adapter_session_created() {
|
||
let adapter = WebSocketOutputAdapter::new();
|
||
let request_id = uuid::Uuid::new_v4();
|
||
let response = CommandResponse::success(request_id)
|
||
.with_message(MessageKind::Notification, "My Session")
|
||
.with_metadata("session_id", "abc123");
|
||
|
||
let outbounds = adapter.adapt(response);
|
||
|
||
assert_eq!(outbounds.len(), 1);
|
||
assert!(matches!(outbounds[0], WsOutbound::SessionCreated { .. }));
|
||
}
|
||
}
|