3

我有示例回声服务器

import asyncio

class EchoServer(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):
        self.transport.write(data)
        # Client piece goes here

loop = asyncio.get_event_loop()
coro = loop.create_server(EchoServer, '127.0.0.1', 8888)
server = loop.run_until_complete(coro)
print('serving on {}'.format(server.sockets[0].getsockname()))

try:
    loop.run_forever()
except KeyboardInterrupt:
    print("exit")
finally:
    server.close()
    loop.close()

我想要做的是添加一个我评论过的客户端，它将连接到一个新的服务器并以这种方式发送数据。有echo client，但我需要一个看起来像这样的过程：

    +-----------+    +-----------+    +--------------+
    | My Server |    | My Client |    | Other Server |
    +-----------+    +-----------+    +--------------+
          |                |                 |
 ===>Get some data         |                 |
          |                |                 |
      Send data ---------->|                 |
          |                |                 |
          |            Send data ----------->|
          |                |                 |
          |                |              Do Stuff
          |                |                 |
          |                | <-----------Send Data
          |                |                 |
          | <--------- Send data             |
          |                |                 |
 <=== Send data            |                 |
          |                |                 |
          |                |                 |
          |                |                 |
          |                |                 |

显然我可以同步执行此操作，但我正在尝试使client -> other server位异步，并且我并没有真正弄清楚如何使用这些asyncio方法在我的服务器部件和客户端部件之间进行通信。

我需要在这里做什么？

4

1 回答 1

12

这是一个简单的代理，它允许您wget 127.0.0.1:8888从 google 获取 html 响应：

import asyncio

class Client(asyncio.Protocol):

    def connection_made(self, transport):
        self.connected = True
        # save the transport
        self.transport = transport

    def data_received(self, data):
        # forward data to the server
        self.server_transport.write(data)

    def connection_lost(self, *args):
        self.connected = False

class Server(asyncio.Protocol):
    clients = {}
    def connection_made(self, transport):
        # save the transport
        self.transport = transport

    @asyncio.coroutine
    def send_data(self, data):
        # get a client by its peername
        peername = self.transport.get_extra_info('peername')
        client = self.clients.get(peername)
        # create a client if peername is not known or the client disconnect
        if client is None or not client.connected:
            protocol, client = yield from loop.create_connection(
                Client, 'google.com', 80)
            client.server_transport = self.transport
            self.clients[peername] = client
        # forward data to the client
        client.transport.write(data)

    def data_received(self, data):
        # use a task so this is executed async
        asyncio.Task(self.send_data(data))

@asyncio.coroutine
def initialize(loop):
    # use a coroutine to use yield from and get the async result of
    # create_server
    server = yield from loop.create_server(Server, '127.0.0.1', 8888)

loop = asyncio.get_event_loop()

# main task to initialize everything
asyncio.Task(initialize(loop))

# run
loop.run_forever()
于 2014-01-23T01:05:05.507 回答