Mongoose WebSocket服务中的发送缓冲区无限增长问题分析

问题背景

在使用Mongoose库开发WebSocket服务时，开发者发现当客户端浏览器进入"休眠标签页"状态后，服务器端的内存使用量会持续增长，最终导致内存耗尽。这个问题特别容易在使用Microsoft Edge浏览器时复现，因为Edge默认会启用"非活动标签页休眠"功能。

问题现象

当WebSocket连接建立后，服务器会定期向客户端发送数据。如果客户端浏览器标签页进入休眠状态，会出现以下现象：

1. 服务器TCP发送缓冲区逐渐填满
2. Mongoose检测到TCP缓冲区满后，停止直接调用send系统调用
3. Mongoose内部发送缓冲区开始积累数据
4. 内部缓冲区大小持续线性增长，直到耗尽系统内存

技术原理分析

Mongoose库设计了一个高效的IO缓冲机制，其工作流程如下：

1. 当应用层调用mg_ws_send发送数据时，Mongoose会先将数据存入连接结构体的发送缓冲区(c->send)
2. 在每次事件循环中，Mongoose通过poll/select/epoll检查每个连接的写状态
3. 如果TCP缓冲区有空间，Mongoose会将内部缓冲区的数据写入TCP层
4. 如果TCP缓冲区满，Mongoose会暂停写入，保留数据在内部缓冲区

当客户端进入休眠状态时，TCP层无法确认收到的数据包(ACK)，导致：

1. 服务器TCP发送缓冲区无法被清空
2. Mongoose检测到TCP缓冲区满状态(is_writeable=false)
3. 但应用层仍持续调用mg_ws_send添加新数据
4. 内部缓冲区(c->send)不断增长，没有上限限制

解决方案

针对这个问题，开发者可以采取以下几种解决方案：

方案一：监控发送缓冲区大小

在发送数据前检查c->send.len的大小，如果超过阈值则暂停发送：

if (c->send.len < MAX_BUFFER_SIZE) {
    mg_ws_send(c, data, len, WEBSOCKET_OP_TEXT);
} else {
    // 记录警告或采取其他措施
}

方案二：使用Mongoose内置的流量控制机制

Mongoose本身已经实现了类似的流量控制机制，特别是在处理大文件传输时。可以参考HTTP静态文件服务的实现方式：

// 类似Mongoose内部处理大文件的方式
if (c->send.len == 0) {
    // 只有当发送缓冲区为空时才添加新数据
    mg_send(c, data, len);
}

方案三：优化数据发送方式

利用Mongoose提供的格式化输出功能，减少中间缓冲区的使用：

// 直接格式化输出到WebSocket，避免中间缓冲区
mg_ws_printf(c, WEBSOCKET_OP_TEXT, "格式化消息: %d", value);

最佳实践建议

1. 对于持续数据流服务，必须实现发送缓冲区监控
2. 设置合理的发送缓冲区大小阈值(如1MB)
3. 考虑实现连接健康检查机制，检测长时间不响应的连接
4. 使用Mongoose提供的高级API(mg_ws_printf等)减少内存拷贝
5. 在定时器回调中增加连接状态检查逻辑

总结

这个问题揭示了在网络编程中处理慢客户端的重要性。Mongoose提供了灵活的缓冲机制，但开发者需要根据应用场景实现适当的流量控制。通过监控发送缓冲区大小和合理使用API，可以有效避免内存无限增长的问题，构建更健壮的WebSocket服务。