如何测试 Nginx WebSocket 代理的吞吐性能
使用 wrk 或 autobahn 工具结合自定义脚本，通过发送大量并发连接和消息来测量 nginx 处理 websocket 连接的每秒请求数（rps）和延迟。核心做法是配置 nginx 的 proxy_pass 并设置正确的 upgrade 头，然后利用压测工具模拟客户端行为，观察服务端 cpu、内存及网络 i/o 指标。
如何配置 Nginx 以支持 WebSocket 代理
Nginx 默认不支持 WebSocket，必须手动添加 Upgrade 和 Connection 头部字段才能正确转发握手请求。具体做法是在 location 块中设置 proxy_http_version 1.1;，并将 $http_upgrade 和 "connection" 映射到 Upgrade 和 Connection 头。这样客户端与后端服务之间的长连接才能建立并保持活跃。
server {
listen 80;
server_name example.com;
location /ws/ {
proxy_pass http://backend_server;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_set_header Host $host;
proxy_set_header X-Real-IP $remote_addr;
proxy_read_timeout 86400s;
proxy_send_timeout 86400s;
}
}
如何使用 wrk 进行 WebSocket 压力测试
wrk 原生不支持 WebSocket，但可以通过 Lua 脚本扩展其功能，或者使用专门针对 WebSocket 的工具如 wscat 配合循环脚本。这里推荐使用 autobahn 测试套件中的 wstest 作为服务器端，配合 wscat 或 Python 脚本作为客户端进行简单吞吐量测试。更专业的做法是使用 tsung 或 jmeter 的 WebSocket 插件。
// 简单的 Python 客户端示例用于发送消息
import websocket
import time
def on_message(ws, message):
pass
def on_error(ws, error):
print(error)
def on_close(ws, close_status_code, close_msg):
print("Connection closed")
def on_open(ws):
start_time = time.time()
for i in range(1000):
ws.send("Hello World")
end_time = time.time()
print(f"Sent 1000 messages in {end_time - start_time} seconds")
if __name__ == "__main__":
ws = websocket.WebSocketApp("ws://example.com/ws/",
on_open=on_open,
on_message=on_message,
on_error=on_error,
on_close=on_close)
ws.run_forever()
基于程序中某个输入、条件或结果的具体问题
当后端服务响应缓慢或断开连接时，Nginx 的 proxy_read_timeout 参数会直接影响 WebSocket 连接的存活时间。如果设置为默认值（通常为 60 秒），而业务逻辑需要保持长连接超过此时限，连接将被 Nginx 主动切断。因此，在测试吞吐性能时，必须确保 proxy_read_timeout 和 proxy_send_timeout 设置为足够大的值（如 86400 秒），以避免因超时导致的连接中断，从而准确测量真实的吞吐能力。
