使用 python asyncio.create_server 实例提示用户输入

Question

我正在学习 python 3 asyncio 库，我运行遇到了一个小问题。我正在尝试改编 python 文档中的 EchoServer 示例以提示用户输入，而不仅仅是回显客户端发送的内容。

我认为这就像添加对 input() 的调用一样简单，但当然 input() 会阻塞，直到出现导致问题的用户输入。

理想情况下，我想继续从客户端接收数据，即使服务器没有任何数据 "say"。有点像聊天客户端，每个连接都在与服务器聊天。我希望能够在每个单独的连接之间切换，并根据需要从 stdin 发送输入。几乎像一个P2P聊天客户端。

考虑以下修改后的 EchoServer 代码：

import asyncio

class EchoServerClientProtocol(asyncio.Protocol):
    def connection_made(self, transport):
        peername = transport.get_extra_info('peername')
        print('Connection from {}'.format(peername))
        self.transport = transport

    def data_received(self, data):
        message = data.decode()
        print('Data received: {!r}'.format(message))

        reply = input()
        print('Send: {!r}'.format(reply))
        self.transport.write(reply.encode())

        #print('Close the client socket')
        #self.transport.close()

loop = asyncio.get_event_loop()
# Each client connection will create a new protocol instance
coro = loop.create_server(EchoServerClientProtocol, '127.0.0.1', 8888)
server = loop.run_until_complete(coro)

# Serve requests until CTRL+c is pressed
print('Serving on {}'.format(server.sockets[0].getsockname()))
try:
    loop.run_forever()
except KeyboardInterrupt:
    pass

# Close the server
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()

我如何在服务器端获取输入形式的标准输入并指定将其发送到哪个连接，同时仍然从连接的客户端接收输入？

Answer 1

您可以使用 loop.add_reader schedule a callback to run when data is available on sys.stdin, and then use an asyncio.Queue 将接收到的标准输入数据传递给您的 data_received 方法：

import sys
import asyncio


def got_stdin_data(q):
    asyncio.ensure_future(q.put(sys.stdin.readline()))

class EchoServerClientProtocol(asyncio.Protocol):
   def connection_made(self, transport):
       peername = transport.get_extra_info('peername')
       print('Connection from {}'.format(peername))
       self.transport = transport

   def data_received(self, data):
       message = data.decode()
       print('Data received: {!r}'.format(message))
       fut = asyncio.ensure_future(q.get())
       fut.add_done_callback(self.write_reply)

   def write_reply(self, fut):
       reply = fut.result()
       print('Send: {!r}'.format(reply))
       self.transport.write(reply.encode())

       #print('Close the client socket')
       #self.transport.close()

q = asyncio.Queue()
loop = asyncio.get_event_loop()
loop.add_reader(sys.stdin, got_stdin_data, q)
# Each client connection will create a new protocol instance
coro = loop.create_server(EchoServerClientProtocol, '127.0.0.1', 8888)
server = loop.run_until_complete(coro)

# Serve requests until CTRL+c is pressed
print('Serving on {}'.format(server.sockets[0].getsockname()))
try:
    loop.run_forever()
except KeyboardInterrupt:
    pass

# Close the server
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()

唯一棘手的一点是我们如何调用 Queue.put/Queue.get 方法；它们都是协程，不能在回调中使用 yield from 或 Protocol 实例方法调用。相反，我们只是使用 asyncio.ensure_future 通过事件循环安排它们，然后使用 add_done_callback 方法来处理我们从 get() 调用中检索到的回复。

注意：asyncio.ensure_future 是在 Python 3.4.4 中引入的。在此之前，该方法被称为 asyncio.async. Additionally, Python 3.7 introduced asyncio.create_task，现在是首选方法。

使用 python asyncio.create_server 实例提示用户输入

Prompt for user input using python asyncio.create_server instance

python

python-asyncio