1

我对 asyncio 和 async def/await 语法有点陌生,所以我想问一下我应该如何做这样的事情:

import asyncio
import pygame
import logging
from pygame import *
log = logging.getLogger('')


class Client:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.loop = asyncio.get_event_loop()
        self.loop.run_until_complete(self.create_client())

    async def create_client(self):
        self.reader, self.writer = await asyncio.open_connection(self.host, 
                                                                 self.port, 
                                                                 loop=self.loop)
        asyncio.ensure_future(self._handle_packets(), loop=self.loop)

    async def _handle_packets(self):
        while True:
            data = await self.reader.read(4096)
            if not data:
                continue
            message = data.decode()
            log.debug("(NET) Received "+message)

    def send(self, data):
        self.loop.run_until_complete(asyncio.ensure_future(self._send(data),
                                                           loop=self.loop))

    async def _send(self, data):
        self.writer.write(data)
        await self.writer.drain()
        print("_send done")

    def disconnect(self):
        print("DC")
        self.loop.close()


def main():
    pygame.init()
    screen = pygame.display.set_mode((640, 480))
    pygame.display.set_caption("Pyond client")
    bg = Surface((640, 480))
    bg.fill(Color("#004400"))
    client = Client('127.0.0.1', 2508)
    while True:
        pygame.event.pump()
        for e in pygame.event.get():
            if e.type == QUIT:
                raise SystemExit
            elif e.type == KEYUP:
                if e.key == K_UP:
                    client.send(b"{'Hello':'World'}")
        screen.blit(bg, (0, 0))
        pygame.display.update()
    client.disconnect()


if __name__ == "__main__":
    main()

此代码使用 pygame 创建 640x480 窗口,然后读取传入的 K_UP(向上箭头)键。按下后,将类似 json 的字符串发送到服务器。_handle_packets应该从服务器读取任何传入数据并打印它。我正在测试此代码并且发送工作正常,但接收延迟很大。我确定我需要将处理程序放在其他地方,那么到底在哪里?顺便说一句,发送只工作一次。在这方面也需要帮助。

4

1 回答 1

1

这里有几个问题。

第一个是相当基本的。事件asycnio循环在完成后停止运行create_client(),并且仅send()数据时再次运行。所以,它唯一能够运行的时间_handle_packets就是你在send()ing 的时候。理想情况下,您应该在更高的范围内启动一次事件循环,并在完成所有操作后关闭它。

第二个问题是,无论何时client.send(b"{'Hello':'World'}"),你都会阻塞外pygame while True循环,阻止任何其他事件被处理,直到前一个事件被发送。您应该使用 aasyncio.Queue将事件排队并从Client班级发送。

以下是我要做的一些更改(抱歉,未经测试;我没有pygame安装 ATM):

# vim: tabstop=4 expandtab

import asyncio
import pygame
import logging
from pygame import *
log = logging.getLogger('')


class Client:
    def __init__(self, host, port, loop):
        self.host = host
        self.port = port
        self.loop = loop
        self.send_q = asyncio.Queue()

    async def connect(self):
        self.reader, self.writer = await asyncio.open_connection(self.host, 
                                                                 self.port, 
                                                                 loop=self.loop)
        self.loop.create_task(self._handle_packets())
        self.loop.create_task(self._send())

    async def _handle_packets(self):
        while True:
            data = await self.reader.read(4096)
            if not data:
                continue
            message = data.decode()
            log.debug("(NET) Received "+message)

    def send(self, data):
        self.send_q.put_nowait(data)

    async def _send(self):
        while True:
            data = await self.send_q.get()
            self.writer.write(data)
            await self.writer.drain()

    def disconnect(self):
        print("DC")
        self.writer.close()


async def main(loop):
    pygame.init()
    screen = pygame.display.set_mode((640, 480))
    pygame.display.set_caption("Pyond client")
    bg = Surface((640, 480))
    bg.fill(Color("#004400"))
    client = Client('127.0.0.1', 2508, loop)
    await client.connect()
    while True:
        pygame.event.pump()
        for e in pygame.event.get():
            if e.type == QUIT:
                raise SystemExit
            elif e.type == KEYUP:
                if e.key == K_UP:
                    client.send(b"{'Hello':'World'}")
        screen.blit(bg, (0, 0))
        pygame.display.update()
    client.disconnect()


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main(loop))
    loop.close()

要记住的另一件重要的事情是,你永远不应该用 阻塞asyncio事件循环pygame,否则Client后台的网络处理将停止。我从来没有使用过pygame,所以我不熟悉哪个pygame函数可能是“阻塞”的,但应该使用调用它们result = await loop.run_in_executor(None, blocking_func, *func_args)。这将在另一个线程中调用阻塞函数。

于 2015-11-03T00:02:14.237 回答