使用 Python 代理实现可扩展的外汇 WebSocket

本指南将教您如何使用 Python 创建 WebSocket 代理服务器。

服务器将执行以下操作：

• 验证客户端身份：在允许客户端连接之前，会检查每个客户端是否拥有唯一的“用户密钥（API Key）”。
• 连接到另一个 WebSocket： 服务器将连接到一个单独的 WebSocket 服务器。
• 中继消息：服务器将从连接的 WebSocket 接收消息并将其发送给所有经过验证的客户端。

开始之前：

• 确保您已安装Python 3.6或更高版本。 WebSockets 需要 Python 3.6 或更高版本。
• 安装 WebSockets 库：您可以在终端中使用以下命令来安装它。

pip install websockets

1. 入门

• 为您的项目创建一个新文件夹。
• 在文件夹中创建一个新的 Python 文件并将其命名为“websocket_proxy_server.py”。该文件将保存您服务器的所有代码。

2. 创建WebSocket服务器

• 导入所需的库。您将需要之前安装的库。
• 构建服务器的基本结构。使用 WebSockets 库为您的服务器创建基础。

import asyncio
import websockets
import json

class WebSocketProxy:

    def init(self, source_url, symbols):

        self.source_url = source_url
        self.clients = set()
        self.symbols = symbols
        self.valid_user_key = "yourValidUserKey"  # Single valid user key for authentication

    async def on_open(self, ws):

        print("Connected to source")
        symbols_str = ",".join(self.symbols.keys())
        init_message = f"{{"userKey":"your_api_key", "symbol":"{symbols_str}"}}"
        await ws.send(init_message)

3. 连接并验证客户端

• 确保服务器已全部设置为接受来自客户端的连接。
• 添加检查以验证每个客户的身份。当客户端尝试连接时，服务器应要求提供“用户密钥”。只有具有正确密钥的客户端才被允许连接。

async def client_handler(self, websocket, path):

        try:

            # Wait for a message that should contain the authentication key
            auth_message = await asyncio.wait_for(websocket.recv(), timeout=10)
            auth_data = json.loads(auth_message)
            user_key = auth_data.get("userKey")

            if user_key == self.valid_user_key:
                self.clients.add(websocket)
                print(f"Client authenticated with key: {user_key}")

                try:
                    await websocket.wait_closed()

                finally:
                    self.clients.remove(websocket)

            else:

                print("Authentication failed")
                await websocket.close(reason="Authentication failed")
        except (asyncio.TimeoutError, json.JSONDecodeError, KeyError):
            print("Failed to authenticate")
            await websocket.close(reason="Failed to authenticate")

4. 连接到源并共享消息

• 创建一个函数，使服务器保持与原始 WebSocket 的连接。
• 此函数应该自动将从原始 WebSocket 接收到的消息发送到所有成功验证的客户端。

async def source_handler(self):
        async with websockets.connect(self.source_url) as websocket:
            await self.on_open(websocket)
            async for message in websocket:
                await self.broadcast(message)

    async def broadcast(self, message):
        if self.clients:
            await asyncio.gather(*(client.send(message) for client in self.clients))

5. 启动服务器

• 创建一个函数来启动服务器并监听连接。
• 添加代码来运行此函数，启动您的 WebSocket 代理服务器。

def run(self, host="localhost", port=8765):
        start_server = websockets.serve(self.client_handler, host, port)
        asyncio.get_event_loop().run_until_complete(start_server)
        asyncio.get_event_loop().run_until_complete(self.source_handler())
        asyncio.get_event_loop().run_forever()

if name == "main":
    symbols = {"EURUSD": {}, "GBPUSD": {}, "USDJPY": {}, "AUDUSD": {}, "USDCAD": {}}
    source_url = "ws://example.com/source"
    proxy = WebSocketProxy(source_url, symbols)
    proxy.run()

总之

您已经成功开发了一个基于Python的WebSocket代理服务器。该服务器可以验证客户端身份，维护与指定数据源的持久连接，并将从源接收到的消息有效地分发给所有经过验证的客户端。事实证明，对于需要将数据从单一来源安全即时传播到不同用户群的应用程序来说，此功能非常宝贵。

下一步

彻底的服务器测试对于确保最佳性能和可靠性至关重要。它验证其对连接和消息传输的正确处理。为了提高效率，请考虑实施负载平衡机制和自定义连接标头。最后，建议将服务器部署到适合生产部署的环境，例如专门为容纳长期网络连接而设计的云服务。

另外，请查看我们网站上最初发布的教程：使用 Python 代理扩展外汇 WebSocket

以上是使用 Python 代理实现可扩展的外汇 WebSocket的详细内容。更多信息请关注PHP中文网其他相关文章！