如何使用asyncio创建TCP代理服务器？

Question

我发现了一个使用asyncio编写的TCP客户端和服务端示例：tcp服务器示例。但是如何连接它们以创建TCP代理服务器，从而接收数据并将其发送到其他地址呢？

- Evgeny

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

- Vincent

---

如何使用传输方式和create_server/create_connection创建此代理？ - Evgeny

在打开到 127.0.0.1:8889 的连接之前，能否从实际请求中读取 local_reader 以获取特定的标头值？ - Sreenadh T C

@SreenadhTC 当然，客户端处理程序可以在转发流量之前执行任何类型的操作。 - Vincent

@Vincent，嘿，是的，那就是我想做的事情，但我遇到了一个问题。 我在SO上发布了同样的问题此处。 - Sreenadh T C