如何使用asyncio創建TCP代理服務器？

Question

我在asyncio： tcp服務器示例中找到了TCP客戶端和服務器的這些示例 。 但是如何連接它們以獲取將接收數據並將其發送到其他地址的TCP代理服務器？

Answer 1

您可以將用戶文檔中 的TCP客戶端和服務器示例組合在一起。

然后，您需要使用這種幫助程序將流連接在一起：

async def pipe(reader, writer):
    try:
        while not reader.at_eof():
            writer.write(await reader.read(2048))
    finally:
        writer.close()

這是一個可能的客戶端處理程序：

async def handle_client(local_reader, local_writer):
    try:
        remote_reader, remote_writer = await asyncio.open_connection(
            '127.0.0.1', 8889)
        pipe1 = pipe(local_reader, remote_writer)
        pipe2 = pipe(remote_reader, local_writer)
        await asyncio.gather(pipe1, pipe2)
    finally:
        local_writer.close()

和服務器代碼：

# Create the server
loop = asyncio.get_event_loop()
coro = asyncio.start_server(handle_client, '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()