如何在with/as语句中动态增删Sockets？是否需用asyncio.create_task？

为什么with/as语句不适合动态添加/移除Socket #

with/as是用来管理固定资源生命周期的语法，它会在进入代码块时一次性初始化所有上下文管理器，退出时统一关闭资源。你当前的代码把两个Socket绑定在同一个async with里，意味着它们的生命周期完全绑定，没法在循环执行过程中动态新增或单独关闭某个Socket。

推荐方案：用asyncio.create_task独立管理每个Socket #

把每个WebSocket连接封装成独立的异步任务，就能随时创建新任务、取消旧任务，轻松实现动态管理。

实现步骤 #

1. 封装通用的Socket消息处理函数，负责单个交易对的监听和消息输出
2. 用字典跟踪当前运行的Socket任务（键为交易对符号，值为对应的Task对象）
3. 编写添加/移除Socket的逻辑，通过创建任务或取消任务来实现动态操作

完整代码示例 #

import asyncio
from binance import AsyncClient, BinanceSocketManager
import config

# 存储当前运行的Socket任务：{交易对符号: 任务对象}
active_sockets = {}

async def handle_symbol_ticker(client, symbol):
    """处理单个交易对的ticker消息"""
    bm = BinanceSocketManager(client)
    async with bm.symbol_ticker_socket(symbol) as socket:
        while True:
            try:
                res = await socket.recv()
                print(f"{symbol}: {res}")
            except asyncio.CancelledError:
                # 任务被取消时退出循环，关闭Socket
                print(f"停止监听 {symbol}")
                break

async def add_socket(client, symbol):
    """添加新的Socket监听任务"""
    if symbol not in active_sockets:
        task = asyncio.create_task(handle_symbol_ticker(client, symbol))
        active_sockets[symbol] = task
        print(f"开始监听 {symbol}")

async def remove_socket(symbol):
    """移除指定交易对的Socket监听任务"""
    if symbol in active_sockets:
        active_sockets[symbol].cancel()
        await active_sockets[symbol]
        del active_sockets[symbol]
        print(f"已移除 {symbol} 的监听")

async def main():
    client = await AsyncClient.create()
    
    # 初始添加两个监听
    await add_socket(client, "BNBBTC")
    await add_socket(client, "ETHBUSD")
    
    # 模拟动态操作：5秒后新增监听，10秒后移除一个监听
    await asyncio.sleep(5)
    await add_socket(client, "BTCUSDT")
    
    await asyncio.sleep(5)
    await remove_socket("BNBBTC")
    
    # 保持主任务运行，直到所有Socket都被移除
    while active_sockets:
        await asyncio.sleep(1)
    
    await client.close()

if __name__ == "__main__":
    asyncio.run(main())

关键说明 #

• 任务跟踪：active_sockets字典用来记录所有运行中的Socket任务，方便后续查找和操作
• 任务取消：调用task.cancel()会触发asyncio.CancelledError，在处理函数中捕获该异常，确保Socket正常关闭后退出
• 动态操作：你可以根据实际业务逻辑（比如用户指令、外部事件）在任意时机调用add_socket或remove_socket来管理连接

补充：with/as的替代思路（不推荐） #

如果坚持要用上下文管理器，需要维护一个动态的上下文实例集合：

• 新增Socket时手动进入上下文，把实例加入集合
• 移除时手动关闭上下文实例，再从集合中删除
  但这种方式需要自己处理上下文的生命周期，代码复杂度高，不如任务方式直观可靠

内容的提问来源于stack exchange，提问作者noskule