python fastapi 实现websocket服务

版本:
fastapi 0.112.2
uvicorn 0.30.6
websocket-clients 0.58.0

服务端

# server.py
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_text() # 接收来自客户端的信息,也可以使用其他方式接收。
await websocket.send_text(f"Message text was: {data}") # 发送给客户端的信息,支持发送json,bytes等。
if __name__ == "__main__":
import uvicorn
uvicorn.run("server:app", host="0.0.0.0", port=8000, reload=True)

客户端

# websocket_client.py
import websocket
import threading
def on_message(ws, message):
print("Received message from server: " + message)
def on_error(ws, error):
print("Error: " + str(error))
def on_close(ws, close_status_code, close_msg):
print("### Connection closed ###")
def on_open(ws):
def run(*args):
# 在连接打开后发送消息
ws.send("Hello, Server!")
# 你可以在这里发送更多的消息或执行其他操作
threading.Thread(target=run).start()
if __name__ == "__main__":
websocket.enableTrace(True)
ws = websocket.WebSocketApp("ws://localhost:8000/ws",
on_message=on_message,
on_error=on_error,
on_close=on_close)
ws.on_open = on_open
ws.run_forever()
posted @   Kevinarcsin001  阅读(27)  评论(0编辑  收藏  举报
编辑推荐:
· 一个奇形怪状的面试题:Bean中的CHM要不要加volatile?
· [.NET]调用本地 Deepseek 模型
· 一个费力不讨好的项目,让我损失了近一半的绩效!
· .NET Core 托管堆内存泄露/CPU异常的常见思路
· PostgreSQL 和 SQL Server 在统计信息维护中的关键差异
阅读排行:
· CSnakes vs Python.NET:高效嵌入与灵活互通的跨语言方案对比
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· Plotly.NET 一个为 .NET 打造的强大开源交互式图表库
· 上周热点回顾(2.17-2.23)
点击右上角即可分享
微信分享提示