fastapi的websocket如何设置最大消息值的大小
时间: 2024-05-05 21:16:48 浏览: 334
在FastAPI中,可以使用`WebSocket`类的`receive`方法的`max_size`参数来设置最大消息值的大小。该参数的默认值为`2^20`(1MB)。要设置不同的值,只需在创建WebSocket连接时传递一个整数值即可。例如,以下代码将设置最大消息值的大小为10MB:
```python
from fastapi import FastAPI, WebSocket
app = FastAPI()
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
while True:
data = await websocket.receive(max_size=10 * 1024 * 1024) # 10MB
# 处理消息
```
在上面的代码中,`max_size=10 * 1024 * 1024`将`max_size`参数设置为10MB。如果收到的消息大小超过了这个值,`WebSocket`类将抛出一个`ValueError`异常。
阅读全文