使用python asyncio构建的简单TCP服务器，为何Telnet连接会立即断开？本文分析一个使用asyncio库编写的简单TCP服务器示例，该服务器在Telnet客户端连接后立即断开连接，并解释其原因及解决方案。代码使用asyncio.StreamReader和asyncio.StreamWriter处理客户端连接，但存在一个关键问题导致Telnet客户端连接后立即断开。
原始代码如下：
import asyncio from asyncio.base_events import server async def handle_client(reader, writer): while True: data = await reader.readline() if not data: break writer.write(data) writer.close() async def main(): server: server = await asyncio.get_running_loop().create_server( handle_client, '127.0.0.1', 8888) async with server: await server.serve_forever() asyncio.run(main())
使用Telnet连接该服务器时，会立即显示“connection closed by foreign host.” 这是因为reader.readline()方法会阻塞，等待完整的行结束符（通常是rn），而Telnet客户端可能不会立即发送完整的行。服务器在未收到完整行之前不会执行任何操作，客户端也收不到任何响应，最终导致服务器关闭连接。
问题的关键在于：服务器需要主动发送欢迎信息，使用writer.drain()确保数据写入，并修改读取数据的方式避免阻塞等待完整行。改进后的handle_client函数如下：
立即学习“Python免费学习笔记（深入）”；
async def handle_client(reader, writer): welcome_message = "Welcome to the server!n" writer.write(welcome_message.encode()) await writer.drain() while True: data = await reader.read(100) if not data: break writer.write(data) await writer.drain()
通过添加欢迎信息并使用reader.read(100)读取固定大小的数据，避免了阻塞等待完整行的问题，从而解决了Telnet连接立即断开的问题。await writer.drain()确保数据写入网络缓冲区，避免数据丢失。这使得服务器能够及时响应客户端连接，并保持连接的持续性。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
