在Tornado异步处理程序中使用简单的python生成器作为协同例程？

Question

我有一个python生成器函数，可以生成文本块。我想get为tornado.web.RequestHandler子类编写一个方法，该方法将遍历生成器，并随即将大块写入响应。

由于这是Tornado，并且由于生成器可能要花一秒钟的时间来处理，所以我认为最好使处理程序异步，使用此生成器作为协程并将控制权在每个块之后传递给IOLoop。但是，我无法做到这一点。

这是我的示例（阻止）代码：

class TextHandler(web.RequestHandler):
    @web.asynchronous
    def get(self, n):
        generator = self.generate_text(100000)
        # Clearly, this will block. How to make it asynchronous?
        for text in generator:
            self.write(text)

    def generate_text(n):
        for x in xrange(n):
            if not x % 15:
                yield "FizzBuzz\n"
            elif not x % 5:
                yield "Buzz\n"
            elif not x % 3:
                yield "Fizz\n"
            else:
                yield "%s\n" % x

如何使此处理程序异步工作？

Answer 1

这是您所描述内容的基本版本。为了避免阻塞，您可以通过回调函数将生成器传递给IOLoop。这里的技巧是因为您没有使用执行实际IO的进程，因此没有os级别的进程/文件处理程序可通过来添加到IOLoop中add_handler，您可以使用简单的add_callback调用并从回调函数中反复调用它来保持IOLoop回调队列中的函数，直到生成器完成为止。

import tornado.httpserver
import tornado.ioloop
import tornado.web

class TextHandler(tornado.web.RequestHandler):
    @tornado.web.asynchronous
    def get(self):
        self.generator = self.generate_text(1000)
        tornado.ioloop.IOLoop.instance().add_callback(self.loop)

    def loop(self):
        try:
            text = self.generator.next()
            self.write(text)
            tornado.ioloop.IOLoop.instance().add_callback(self.loop)
        except StopIteration:
            self.finish()

    def generate_text(self, n):
        for x in xrange(n):
            if not x % 15:
                yield "FizzBuzz\n"
            elif not x % 5:
                yield "Buzz\n"
            elif not x % 3:
                yield "Fizz\n"
            else:
                yield "%s\n" % x

application = tornado.web.Application([
    (r"/text/", TextHandler),
])

http_server = tornado.httpserver.HTTPServer(application)
http_server.listen(8888)
tornado.ioloop.IOLoop.instance().start()