Python-trio中KeyboardInterrupt后连接清理及会话收尾逻辑问题

我完全懂你的痛点：现在的逻辑是等netcat监听器关闭才发注销，但此时连接已经断了，完全没意义。我们需要在用户触发KeyboardInterrupt（也就是按Ctrl+C）的时候，先主动发送注销字符串，再优雅地清理套接字连接。下面是具体的解决方案：

核心思路 #

利用trio的异步上下文管理器和异常捕获机制，把套接字生命周期、登录/注销逻辑绑定在一起：

• 连接建立后立即发送登录字符串
• 用try-except捕获KeyboardInterrupt，在用户中断时优先发送注销指令
• 借助trio.open_tcp_stream的async with语法，自动保证套接字的优雅关闭，即使出现异常

完整示例代码 #

import trio

async def manage_server_session(host: str, port: int, login_msg: str, logout_msg: str):
    # 用async with自动管理套接字生命周期，确保连接最终被关闭
    async with await trio.open_tcp_stream(host, port) as stream:
        # 连接建立后立即发送登录字符串
        await stream.send_all(login_msg.encode())
        print(f"已发送登录字符串: {login_msg}")
        
        try:
            # 这里是你的会话逻辑，比如保持连接、处理消息等
            # 示例：无限等待（模拟会话持续），直到用户触发中断
            print("会话已启动，按Ctrl+C结束会话...")
            await trio.sleep_forever()
            
        except KeyboardInterrupt:
            # 用户触发中断，优先发送注销字符串
            print("\n收到中断信号，准备发送注销字符串...")
            try:
                await stream.send_all(logout_msg.encode())
                print(f"已发送注销字符串: {logout_msg}")
            except Exception as e:
                print(f"发送注销字符串时出错: {e}")
            # 这里不需要手动关闭套接字，async with会自动处理
        finally:
            print("会话已结束，连接已清理")

if __name__ == "__main__":
    # 配置参数
    HOST = "127.0.0.1"
    PORT = 9999
    LOGIN_STR = "USER_LOGIN:my_username"
    LOGOUT_STR = "USER_LOGOUT:my_username"
    
    try:
        trio.run(manage_server_session, HOST, PORT, LOGIN_STR, LOGOUT_STR)
    except KeyboardInterrupt:
        # 外层捕获防止未处理的中断（实际内层已经处理）
        print("\n程序已退出")

关键细节解释 #

1. 异步上下文管理器：async with await trio.open_tcp_stream(...)会在代码块结束（不管是正常结束还是异常）时自动调用stream.close()，保证连接被正确清理，不需要手动管理套接字的关闭逻辑。
2. 登录时机：连接建立后立刻调用await stream.send_all(login_msg.encode())，确保登录字符串在会话开始时就被发送到服务器。
3. KeyboardInterrupt处理：在会话逻辑的try块里捕获中断，此时连接还处于活跃状态，发送注销字符串能被服务器正常接收，避免之前“连接断了才发注销”的无效操作。
4. 异常容错：发送注销时额外加了一层try-except，防止因为网络问题导致发送失败时程序崩溃，保证后续的清理逻辑能正常执行。

这样修改后，当用户按Ctrl+C时，程序会先尝试发送注销字符串，再由trio的上下文管理器自动关闭套接字，完美解决你遇到的问题。

内容的提问来源于stack exchange，提问作者NarūnasK