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:
2026-10-23 12:57:03 +08:00
parent 206adbff6c
commit da5c5c1e09
2 changed files with 82 additions and 6 deletions
@@ -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);
}