如何在Python中使用tulip/asyncio創建中繼服務器？

Question

我有example echo server

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() 

我想要做的就是添加一個客戶端片而言，我的評論將連接到一個新的服務器併發送數據關閉該-A-方式。還有的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方法在我的服務器部分和客戶端部分之間進行通信。

我需要在這裏做什麼？

Answer 1

下面是一個簡單的代理，讓你可以wget 127.0.0.1:8888，並從谷歌一個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()