使用 Python 代理实现可扩展的外汇 WebSocket-Python教程-PHP中文网

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

Susan Sarandon

发布： 2025-01-04 18:55:40

原创

556 人浏览过

Implementing a Scalable Forex WebSocket Using a Python Proxy

本指南将教您如何使用 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()

登录后复制