{"id":8944,"date":"2023-09-13T01:41:45","date_gmt":"2023-09-12T17:41:45","guid":{"rendered":"\/?p=8944"},"modified":"2023-09-13T01:41:45","modified_gmt":"2023-09-12T17:41:45","slug":"fastapi%e5%9f%ba%e7%a1%80%ef%bc%9a8-websocket-%e9%a1%b9%e7%9b%ae%e9%83%a8%e7%bd%b2","status":"publish","type":"post","link":"\/?p=8944","title":{"rendered":"FastAPI\u57fa\u7840\uff1a8.WebSocket & \u9879\u76ee\u90e8\u7f72"},"content":{"rendered":"

WebSocket<\/h2>\n

WebSockets - FastAPI (tiangolo.com)<\/a><\/p>\n

encode\/broadcaster: Broadcast channels for async web apps. \ud83d\udce2 (github.com)<\/a><\/p>\n

\u76f4\u63a5\u7528\u7684\u5b98\u65b9\u6587\u6863\u4f8b\u5b50<\/strong>:<\/p>\n

from uvicorn import run\n\n"""\nwebsocket\n pip install websockets\n"""\nfrom fastapi import FastAPI\n\napp = FastAPI(title="websocket")\nfrom typing import List\n\nfrom fastapi import FastAPI, WebSocket, WebSocketDisconnect\nfrom fastapi.responses import HTMLResponse\n\nhtml = """\n<!DOCTYPE html>\n<html>\n    <head>\n        <title>Chat<\/title>\n    <\/head>\n    <body>\n        <h1>WebSocket Chat<\/h1>\n        <h2>Your ID: <span id="ws-id"><\/span><\/h2>\n        <form action="" onsubmit="sendMessage(event)">\n            <input type="text" id="messageText" autocomplete="off"\/>\n            <button>Send<\/button>\n        <\/form>\n        <ul id='messages'>\n        <\/ul>\n        <script>\n            var client_id = Date.now()\n            document.querySelector("#ws-id").textContent = client_id;\n            var ws = new WebSocket(`ws:\/\/localhost:8000\/ws\/${client_id}`);\n            ws.onmessage = function(event) {\n                var messages = document.getElementById('messages')\n                var message = document.createElement('li')\n                var content = document.createTextNode(event.data)\n                message.appendChild(content)\n                messages.appendChild(message)\n            };\n            function sendMessage(event) {\n                var input = document.getElementById("messageText")\n                ws.send(input.value)\n                input.value = ''\n                event.preventDefault()\n            }\n        <\/script>\n    <\/body>\n<\/html>\n"""\n\nclass ConnectionManager:\n    def __init__(self):\n        # \u5b58\u50a8\u7528\u6237\n        self.active_connections: List[WebSocket] = []\n\n    async def connect(self, websocket: WebSocket):\n        # \u8fde\u63a5\u4e8b\u4ef6\n        await websocket.accept()\n        self.active_connections.append(websocket)\n\n    def disconnect(self, websocket: WebSocket):\n        # \u65ad\u5f00\n        self.active_connections.remove(websocket)\n\n    async def send_personal_message(self, message: str, websocket: WebSocket):\n        # \u53d1\u9001\u6d88\u606f\n        await websocket.send_text(message)\n\n    async def broadcast(self, message: str):\n        # \u5149\u64ad\u6d88\u606f\n        for connection in self.active_connections:\n            await connection.send_text(message)\n\nmanager = ConnectionManager()\n\n@app.get("\/")\nasync def get():\n    return HTMLResponse(html)\n\n@app.websocket("\/ws\/{client_id}")\nasync def websocket_endpoint(websocket: WebSocket, client_id: int):\n    await manager.connect(websocket)\n    try:\n        while True:\n            data = await websocket.receive_text()\n            await manager.send_personal_message(f"You wrote: {data}", websocket)\n            await manager.broadcast(f"Client #{client_id} says: {data}")\n    except WebSocketDisconnect:\n        manager.disconnect(websocket)\n        await manager.broadcast(f"Client #{client_id} left the chat")\n\nif __name__ == "__main__":\n    # \u542f\u7528web\u670d\u52a1\uff0c reload -> \u5f53\u6539\u53d8\u4ee3\u7801\u65f6\u81ea\u52a8\u91cd\u542f # fastapi_02:app\n    run("__main__:app", reload=True)\n<\/code><\/pre>\n
\u6548\u679c<\/h2>\n
\"image.png\"<\/p>\n
\u90e8\u7f72<\/h2>\n
tiangolo\/uvicorn-gunicorn-fastapi-docker: Docker image with Uvicorn managed by Gunicorn for high-performance FastAPI web applications in Python with performance auto-tuning. Optionally with Alpine Linux. (github.com)<\/a><\/p>\n
https:\/\/hub.docker.com\/r\/tiangolo\/uvicorn-gunicorn-fastapi<\/a><\/p>\n
FROM tiangolo\/uvicorn-gunicorn-fastapi:python3.9\nCOPY . \/app\nRUN ln -sf \/usr\/share\/zoneinfo\/Asia\/Shanghai \/etc\/localtime && \\\n    echo 'Asia\/Shanghai' >\/etc\/timezone && \\\n    pip install --no-cache-dir --upgrade -r requirements.txt -i https:\/\/pypi.tuna.tsinghua.edu.cn\/simple\/<\/code><\/pre>\n
ps: \u8fd9\u4e2a\u955c\u50cf\u6700\u7ec8\u4f1a\u6267\u884cmain.py\uff0c\u6240\u4ee5\u4fdd\u8bc1main.py\u5b58\u5728FastAPI \u5b9e\u4f8b, \u5e76\u4e14\u542f\u52a8\u7aef\u53e3\u4e3a 80<\/p>\n
version: "3"\nservices:\n  backend:\n    build:\n      context: .\/\n      # dockerfile\u6587\u4ef6\u540d\n      dockerfile: Dockerfile\n    # \u5bb9\u5668\u540d\u79f0\n    container_name: fastapi-serve\n    # \u7aef\u53e3\u6620\u5c04\n    ports:\n      - "1855:80"<\/code><\/pre>\n
\"image.png\"<\/p>\n
docker-compose up -d<\/code><\/pre>\n
\"image.png\"<\/p>\n","protected":false},"excerpt":{"rendered":"
WebSocket WebSockets – FastAPI (tiangolo.com) encode\/broadcaster: Br   \u2026 …<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"ngg_post_thumbnail":0},"categories":[204],"tags":[116],"_links":{"self":[{"href":"\/index.php?rest_route=\/wp\/v2\/posts\/8944"}],"collection":[{"href":"\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=8944"}],"version-history":[{"count":1,"href":"\/index.php?rest_route=\/wp\/v2\/posts\/8944\/revisions"}],"predecessor-version":[{"id":8945,"href":"\/index.php?rest_route=\/wp\/v2\/posts\/8944\/revisions\/8945"}],"wp:attachment":[{"href":"\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=8944"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=8944"},{"taxonomy":"post_tag","embeddable":true,"href":"\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=8944"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}