python-trio echo server的确切版本是什么 - asyncio中的多个客户端？

问题描述

我正在尝试将以下类似 echo 服务器 IRC 的聊天终端转换为 asyncio ，但我不认为关于 trio 的信息太多，因为它是新的，但是将其转换为 asyncio 的内容是什么？

import trio
from itertools import count
from datetime import datetime

PORT = 9999
BUFSIZE = 16384
CONNECTION_COUNTER = count()

class ServerProtocol:

    def __init__(self,server_stream):
        self.ident = next(CONNECTION_COUNTER)
        self.stream = server_stream

    async def listen(self):
        while True:
            data = await self.stream.receive_some(BUFSIZE)
            print("echo_server {}: received data {!r}".format(self.ident,data))
            if not data:
                print("echo_server {}: connection closed".format(self.ident))
                return
            print("echo_server {}: sending data {!r}".format(self.ident,data))
            await self.stream.send_all('success'.encode())


class Server:

    def __init__(self):
        self.protocols = []

    async def receive_connection(self,server_stream):
        print('{} - {} CONNECTED.'.format(datetime.Now(),dir(server_stream)))
        sp: ServerProtocol = ServerProtocol(server_stream)
        self.protocols.append(sp)
        await sp.listen()


async def main():
    await trio.serve_tcp(Server().receive_connection,PORT)

trio.run(main)

解决方法

感谢 anyio，您无需更改大量代码即可使其与 asyncio 配合使用；)

import anyio
from itertools import count
from datetime import datetime

PORT = 9999
BUFSIZE = 16384
CONNECTION_COUNTER = count()


class ServerProtocol:

    def __init__(self,server_stream):
        self.ident = next(CONNECTION_COUNTER)
        self.stream = server_stream

    async def listen(self):
        async with self.stream:
            data = await self.stream.receive(BUFSIZE)
            print("echo_server {}: received data {!r}".format(self.ident,data))
            if not data:
                print("echo_server {}: connection closed".format(self.ident))
                return
            print("echo_server {}: sending data {!r}".format(self.ident,data))
            await self.stream.send('success'.encode())


class Server:

    def __init__(self):
        self.protocols = []

    async def receive_connection(self,server_stream):
        print('{} - {} CONNECTED.'.format(datetime.now(),dir(server_stream)))
        sp: ServerProtocol = ServerProtocol(server_stream)
        self.protocols.append(sp)
        await sp.listen()


async def main():
    listener = await anyio.create_tcp_listener(local_port=PORT)
    await listener.serve(Server().receive_connection)


anyio.run(main)

你绝对应该看看 anyio :)

,

我会选择：

import asyncio
from itertools import count

PORT = 9999
BUFSIZE = 16384
CONNECTION_COUNTER = count()

class ServerProtocol:
    def __init__(self,read,write):
        self.ident = next(CONNECTION_COUNTER)
        self.read = read
        self.write = write

    async def listen(self):
        while True:
            data = await self.read.read(BUFSIZE)
            print("echo_server {}: received data {!r}".format(self.ident,data))
            self.write.write('success'.encode())
            await self.write.drain()

class Server:
    def __init__(self):
        self.protocols = []

    async def receive_connection(self,write):
        sp: ServerProtocol = ServerProtocol(read,write)
        self.protocols.append(sp)
        await sp.listen()

async def main():
    server = await asyncio.start_server(Server().receive_connection,'127.0.0.1',PORT)
    async with server:
        await server.serve_forever()

asyncio.run(main())

请注意，我保留了您的名字 ServerProtocol，但请记住，“协议”类在 asyncio 中具有不同的含义。代码使用 asyncio“流”层（很像原始的三重奏）而不是较低级别的传输/协议层，正如类名中的“协议”后缀所理解的那样。