WebSocket 基础：六回调写双向通信

本章导读

21.1 引入

手写 WebSocket 的痛苦清单：Upgrade 握手校验（Sec-WebSocket-Key 的 SHA-1+Base64）、帧头解析（掩码位、扩展长度）、分片重组、控制帧穿插、close 状态机、permessage-deflate 扩展协商。XServer 的分工是：宿主做握手与校验（失败 400），xrtWsStreamAttach 做帧层——回调拿到的是"整条消息"，发送的是一个函数调用。

21.2 握手与回调

• 握手按 vhost 路由（Host/SNI 同 HTTP）；路由失败 404
• host 未导出 WsText/WsBinary → 503（服务存在但没写业务）
• Pong 由 xrt 层自动回应——WsPing/WsPong 回调只做观察（心跳统计）
• 握手期线路硬边界：防止读满 ReadLimit 后永久等待的慢速攻击

21.3 发送族与 echo

static void WsOpen(XS_HostInfo* pHost, xwsstream* pWs)
{
    (void)xrtWsStreamText(pWs, xrtStrView("connected"));
}

static void WsText(XS_HostInfo* pHost, xwsstream* pWs, xstrview tText)
{
    (void)xrtWsStreamText(pWs, tText);      /* echo */
}

static void WsBinary(XS_HostInfo* pHost, xwsstream* pWs, xbytesview tData)
{
    (void)xrtWsStreamBinary(pWs, tData);
}

刚才发生了什么：发送族 xrtWsStreamText / Binary（拷贝发送）为主；另有 ...Ref（借用发送，函数返回前缓冲不可动）与 ...Take（转交所有权，零拷贝大消息）变体——按消息来源选。

21.4 连接上下文

WS 业务几乎都需要"每个连接自己的状态"（用户身份、订阅列表）。推荐做法：连接注册表（自建 map：连接指针 → 上下文），WsOpen 建、WsClose 删：

static xmutex  G_WsLock;
static xmap*   G_WsConns;    /* xwsstream* → ConnCtx*（跨回调借用需自管）*/

typedef struct ConnCtx {
    char sUser[64];
    uint64 iLastSeen;
} ConnCtx;

static void WsOpen(XS_HostInfo* pHost, xwsstream* pWs)
{
    ConnCtx* pCtx = xrtCalloc(1, sizeof(ConnCtx));
    /* 入表（锁内）... */
}
static void WsClose(XS_HostInfo* pHost, xwsstream* pWs,
    uint16 iCode, xstrview tReason)
{
    /* 出表并释放（锁内）；WsClose 保证是终态回调 */
}

注意：连接生命周期宿主管，你的上下文生命周期你管——注册表模式让两者精确同步。这个表也是 ch22 广播的基础设施。

21.5 契约

• 回调线：六回调在 worker 线程可并发；消息视图（tText/tData）回调内有效。
• 终态线：WsClose 是每连接的终态回调——注册表清理放这里最可靠。
• 发送线：Text/Binary 拷贝、Ref 借用、Take 转交——按缓冲所有权选变体。

常见错误

编程练习