Workerman中WebSocket的Permessage-Deflate压缩问题分析与解决方案

2025-05-20 12:58:11作者：田桥桑Industrious

问题背景

在使用Workerman框架开发WebSocket服务时，开发者可能会遇到一个关于消息压缩的典型问题：当客户端发送短消息时一切正常，但在发送长消息后，后续的消息会出现解压失败的情况，错误提示为"inflate_add(): data error"。这个问题与WebSocket协议中的permessage-deflate扩展实现有关。

问题分析

WebSocket协议支持通过permessage-deflate扩展进行消息压缩，这可以显著减少网络传输的数据量。Workerman内置了对这一扩展的支持，但在某些情况下会出现以下现象：

短消息可以正常压缩和解压
长消息首次可以解压成功
长消息之后的所有消息都会解压失败
需要重新连接才能恢复正常

这主要是因为Workerman默认的压缩窗口设置与某些客户端实现不兼容导致的。具体来说，PHP的zlib扩展默认使用15位的窗口大小(32KB)，而某些WebSocket客户端可能期望使用更小的窗口(如9位，即512字节)。

解决方案

方案一：修改Workerman源码

在Workerman的Websocket.php文件中，修改inflate_init和deflate_init函数的window参数为15：

protected static function inflate($connection, $buffer, $is_fin_frame) {
    if (!isset($connection->context->inflator)) {
        $connection->context->inflator = \inflate_init(
            \ZLIB_ENCODING_RAW,
            [
                'level'    => -1,
                'memory'   => 8,
                'window'   => 15,  // 修改为15
                'strategy' => \ZLIB_DEFAULT_STRATEGY
            ]
        );
    }
    // ...其余代码...
}

protected static function deflate($connection, $buffer) {
    if (!isset($connection->context->deflator)) {
        $connection->context->deflator = \deflate_init(
            \ZLIB_ENCODING_RAW,
            [
                'level'    => -1,
                'memory'   => 8,
                'window'   => 15,  // 修改为15
                'strategy' => \ZLIB_DEFAULT_STRATEGY
            ]
        );
    }
    // ...其余代码...
}

这种修改强制服务端使用15位的窗口大小，与PHP的zlib默认行为一致，可以解决大多数兼容性问题。

方案二：通过响应头协商压缩参数

另一种更灵活的方式是通过WebSocket握手时的响应头来协商压缩参数：

$worker->onWebSocketConnect = function ($connection) {
    // 设置响应头，声明服务端支持permessage-deflate
    // 并指定客户端最大窗口位数为9
    $connection->headers = [
        'sec-websocket-extensions: permessage-deflate; server_no_context_takeover; client_max_window_bits=9'
    ];
    
    // 设置WebSocket类型为支持压缩的BLOB_DEFLATE
    $connection->websocketType = Websocket::BINARY_TYPE_BLOB_DEFLATE;
};