Question

所以从技术上讲，我已经创建了一个TCP侦听套接字，并且一切正常。

它只是获取tcp json消息并将其吐出。

class TCPListener(Base):
    def __init__(self,
                 address,
                 port,
                 buffer,
                 backlog,
                 **kwargs):
        self.tcp_address = address
        self.tcp_port = port
        self.tcp_buffer = buffer
        self.tcp_asyncio_backlog = backlog
        self.queue = queue.Queue()

    def __await__(self):
        self._server = yield from asyncio.start_server(
            self.__on_message,
            host=self.tcp_address,
            port=self.tcp_port,
            backlog=self.tcp_asyncio_backlog
        )
        return self

    async def __on_message(self, reader, writer):
        message = await reader.read(self.tcp_buffer)
        try:
            json_encoded = json.loads(message.decode('UTF-8').strip())
            self.queue.put(json_encoded)
        writer.close()

    def start(self):
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        asyncio.ensure_future(self, loop=loop)


    def receive(self):
        while self.queue.empty():
            sleep_random = random.randint(0, 1000)
            time.sleep(sleep_random / 1000.0)

        if not self.queue.empty():
            return self.queue.get(block=False)
        return '', ''

现在，我正在尝试使此代码适用于TCP和UDP侦听器。我想知道我可以保留什么功能，以及如何在两个侦听器之间共享它们，这对于如何解决这个问题一无所知！如果我没记错的话，我应该使用“ create_datagram_endpoint”，但仍然不知道如何将其与此类和函数粘合在一起。

如何使用AsyncIo

0 个答案: