Flask SocketIO经Nginx反向代理发送1MB数据时连接断开问题求解

问题背景 #

直接连接Flask-SocketIO服务时传输数据正常，但通过Docker容器中的Nginx反向代理转发后，传输约1MB的大数据会导致连接断开。当前使用的Nginx location配置如下：

location /socket.io { 
    proxy_buffering off;
    proxy_pass http://backend;
    proxy_http_version 1.1;
    large_client_header_buffers 4 16k;
    client_body_buffer_size 100M;
    proxy_buffering off;
    proxy_buffers 8 5M;
    proxy_buffer_size 5M;
    proxy_read_timeout 300s;
    proxy_send_timeout 300s;
    proxy_buffering off;  # Disable buffering for WebSocket

    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_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    proxy_set_header X-Forwarded-Proto $scheme;
}

解决方案 #

1. 增加Nginx全局请求体大小限制 #

Nginx默认client_max_body_size为1MB，刚好触发你的场景限制，需在http块或当前location块中添加该参数：

# 可添加在http块全局生效，或location /socket.io内仅针对该路径
client_max_body_size 100M;

2. 动态处理Connection请求头 #

SocketIO在初始polling阶段使用keep-alive而非Upgrade连接头，固定设置Connection "Upgrade"会导致兼容性问题。需在Nginx的http块中添加映射规则，再修改location内的配置：

# 在http块中添加
map $http_upgrade $connection_upgrade {
    default upgrade;
    '' close;
}

# 修改location内的Connection头设置
proxy_set_header Connection $connection_upgrade;

3. 调整Flask-SocketIO的缓冲区配置 #

Flask-SocketIO默认的HTTP缓冲区大小可能不足以处理1MB数据，初始化SocketIO实例时增大缓冲区：

from flask import Flask
from flask_socketio import SocketIO

app = Flask(__name__)
# 设置最大HTTP缓冲区为10MB，根据需求调整
socketio = SocketIO(app, max_http_buffer_size=10*1024*1024)

4. 检查Docker容器MTU限制 #

Docker默认MTU为1500，大数据传输时可能因分片问题导致连接中断。启动Nginx容器时调整MTU：

docker run --mtu=9000 [你的Nginx镜像和其他参数]

若使用docker-compose，在compose文件中添加网络MTU配置：

version: '3'
services:
  nginx:
    image: nginx:latest
    networks:
      default:
        mtu: 9000
    # 其他配置...
networks:
  default:
    driver: bridge
    driver_opts:
      com.docker.network.driver.mtu: 9000

5. 验证Nginx配置并重启容器 #

修改配置后，进入Nginx容器验证配置合法性：

docker exec -it [你的Nginx容器名] nginx -t

验证通过后重启容器：

docker restart [你的Nginx容器名]

内容的提问来源于stack exchange，提问作者Edward