天天看点

tornado 改同步阻塞为异步的方法!

怎么让一个阻塞的函数实现异步,不阻塞别人的请求呢,相信很多人都苦恼在这个问题上,我整理这篇文章一方面加深自己的印象,一方面就是为了正在看文章的你!

class SleepHandler(tornado.web.RequestHandler):
    def get(self):
        result = self.sleep(5)
        print result

    def sleep(self, seconds):
        time.sleep(seconds)
        return "sleep 5s"
           

这个方法是阻塞的,就算你在get上使用装饰器@tornado.web.asynchronous、@tornado.gen.coroutine都没用,哪怎么办?

下面介绍2种方法:

1、利用celery,把阻塞的操作放到队列里,可以用rabbitmq做代理也可以用redis,可以看这篇文章(原理)

有个tornado的封装:tornado-celery, 前提是必须先安装celery、代理用redis或者rabbitmq都行

from celery import Celery
import tcelery
from tornado import gen.coroutine
tcelery.setup_nonblocking_producer()
celery = Celery("tasks", broker="amqp://guest:[email protected]:5672")
celery.conf.CELERY_RESULT_BACKEND = "amqp"

@celery.task
def sleep(seconds):
    time.sleep(seconds)
    return "sleep 5s"
class SleepHandler(tornado.web.RequestHandler):
    @tornado.gen.coroutine
    def get(self):
        result = yield tcelery.celery_task(sleep, args=[5])
        print result
           

 或者把任务task分开写也可以

tasks:

from celery import Celery
celery = Celery("tasks", broker="amqp://guest:[email protected]:5672")
celery.conf.CELERY_RESULT_BACKEND = "amqp"

@celery.task
def sleep(seconds): 
    time.sleep(seconds) 
    return "sleep 5s"
    
if __name__ == "__main__": celery.start()
           

hander:

import tcelery
from tornado import gen.coroutine
class SleepHandler(tornado.web.RequestHandler):
    @tornado.gen.coroutine
    def get(self):
        result = yield gen.Task(tasks.sleep.apply_async, args=[5])
        print result
           

2、利用线程或者进程方式:future,python3.0以上自带,2.7需要安装sudo pip install futures

他有2个属性,ThreadPoolExecutor和ProcessPoolExecutor,使用是必须在你的处理类里实例化一个对象来设置线程/进程数executor = ThreadPoolExecutor(2)

from concurrent.futures import ThreadPoolExecutor
from tornado.concurrent import run_on_executor
class SleepHandler(tornado.web.RequestHandler):
    executor = ThreadPoolExecutor(2)

    @tornado.web.asynchronous
    @tornado.gen.coroutine
    def get(self):
        result = yield self.sleep(5) #现在的sleep()已经实现了异步,所以可以这样用,现在你知道为什么不是所有的yield xxx()都能实现异步的原因了吧
        print result
        self.finish()#使用了tornado.web.asynchronous装饰器时,必须显示关闭链接
  
    @run_on_executor
    def sleep(self,seconds):
        time.sleep(seconds)
        return "sleep 5s"
           

所以如果你有一个耗时的操作,都可以利用这两个方式实现异步调用!

写的不对的地方非常希望你能指出!