0

我正在尝试将以下 echo server 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)
4

2 回答 2

3

感谢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 :)

于 2021-05-15T09:00:14.367 回答
0

我会选择类似的东西：

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))
            if not data:
                print("echo_server {}: connection closed".format(self.ident))
                return
            print("echo_server {}: sending 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, read, 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“流”层（很像 trio 原始层），而不是从类名称中的“协议”后缀可以理解的较低级别的传输/协议层。

于 2021-05-14T09:55:32.133 回答