主页 / 问题 / 1599865
Accepted
Almaz
Asked:2024-11-15 20:57:59 +0000 UTC2024-11-15 20:57:59 +0000 UTC

监听位于不同子网的服务器的 websocket

  • 772

有一个特定的服务器位于以下地址的不同子网中ws://192.168.48.33:9090/ws/uplink/app，它通过 websocket 发送遥测数据。如何实现FastAPI服务器监听？我写了下面的代码，但它只在本地机器上监听ws://127.0.0.1:8000/ws。

主要.py

import ...

@app.websocket("/ws")
async def websocket(websocket: WebSocket):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        print(data)
python

1 个回答

  1. Best Answer
    import uvicorn
from contextlib import asynccontextmanager
from fastapi import FastAPI, WebSocket
import asyncio
import websockets



def create_websocket_client_task(app_state):
    """
    Establish a WebSocket client connection and handle incoming messages in a loop.
    """

    async def websocket_client():
        uri = "ws://localhost:5000/ws"  # Replace with your WebSocket server URI
        try:
            async with websockets.connect(uri) as websocket:
                print(f"Connected to WebSocket server at {uri}")

                while app_state["running"]:
                    try:
                        message = await websocket.recv()  # Receive a message
                        print(f"Received message: {message}")
                        # Process the message as needed
                    except websockets.ConnectionClosed:
                        print("WebSocket connection closed")
                        break
        except Exception as e:
            print(f"Error in WebSocket connection: {e}")

    return websocket_client


@asynccontextmanager
async def lifespan(app: FastAPI):
    app_state = {"running": True}

    # Start WebSocket client in a background asyncio task
    websocket_task = asyncio.create_task(create_websocket_client_task(app_state)())

    yield  # Yield control to FastAPI for its lifespan

    # On shutdown, clean up
    app_state["running"] = False
    await websocket_task  # Wait for the WebSocket task to complete
    print("WebSocket client shutdown")



# Create the FastAPI app with lifespan
app = FastAPI(lifespan=lifespan)
    • 0

相关问题