fix(ai): SSE 帧优先投给在途流式请求,修「生成后按钮永久不可点」
WebSocketServer.send 原本 WS 优先:只要同一账号还有 WS 会话(问答助手页会建 连接),文稿工作台那条 /stream 的帧就全被投到 WS 上,SSE 一帧收不到——正文只能 等末帧,收尾的 __END__ 永远不来,前端 loading 不复位,生成按钮卡死。 改为 SSE 优先:ChatStreamRegistry 是按请求登记的,登记期间发出的帧都属于那条 请求;没有在途请求时 emit 返回 false,照旧走 WS。握手回执「连接成功」改由 register() 直接写给该会话,避免被派发到别的请求。 测试:补两条回归(有在途 SSE 时即使 WS 在线也投 SSE;没有在途 SSE 时仍走 WS), ai 包 158 条全绿。
This commit is contained in:
@@ -58,7 +58,9 @@ public class WebSocketServer {
|
||||
logger.info("websocket connected, userId={}", userId);
|
||||
|
||||
try {
|
||||
send(userId, "连接成功");
|
||||
// 握手回应属于这条连接本身,不能走 {@link #send}:那条路会优先投给
|
||||
// 在途的 HTTP 流,可能把「连接成功」送到别的请求上,端上就等不到握手回执。
|
||||
session.getBasicRemote().sendText("连接成功");
|
||||
} catch (IOException e) {
|
||||
logger.warn("websocket 握手回应发送失败, userId={}", userId, e);
|
||||
}
|
||||
@@ -124,17 +126,25 @@ public class WebSocketServer {
|
||||
/**
|
||||
* 向指定用户推送一条文本;用户不在线时静默丢弃。
|
||||
*
|
||||
* <p>优先走 WebSocket;没有截活会话时退到 HTTP 流式通道
|
||||
* ({@link ChatStreamRegistry})。生产网关在 {@code /api} 前丢 Upgrade 头,
|
||||
* WS 握手根本进不来,端上因此一直在用 HTTP 流那条路。
|
||||
* <p><b>先投在途的 HTTP 流通道(SSE),没有才退到 WebSocket。</b>
|
||||
*
|
||||
* <p>HTTP 流通道是按请求登记的:登记期间发出的每一帧,都属于那条正在跑的
|
||||
* {@code /stream} 请求。反过来按「这个用户有没有 WS 会话」优先就会出事——同一个账号
|
||||
* 在问答助手开着(WS 会话在 {@code SESSIONS} 里)时,再到文稿工作台生成,文稿页
|
||||
* 那条 SSE 一帧都收不到:正文只能等末帧,收尾的 {@code __END__} 永远不来,
|
||||
* 生成按钮就卡在加载态、点不动(见 issue 20)。
|
||||
*
|
||||
* <p>没有在途请求时 {@link ChatStreamRegistry#emit} 返回 {@code false},照旧走 WS。
|
||||
* 生产网关在 {@code /api} 前丢 Upgrade 头、WS 握手进不来,端上用的正是 SSE 那条路。
|
||||
*/
|
||||
static void send(String userId, String message) throws IOException {
|
||||
if (ChatStreamRegistry.emit(userId, message)) {
|
||||
return;
|
||||
}
|
||||
Session target = SESSIONS.get(userId);
|
||||
if (target != null) {
|
||||
target.getBasicRemote().sendText(message);
|
||||
return;
|
||||
}
|
||||
ChatStreamRegistry.emit(userId, message);
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user