怎么让一个阻塞的函数实现异步,不阻塞别人的请求呢,相信很多人都苦恼在这个问题上,我整理这篇文章一方面加深自己的印象,一方面就是为了正在看文章的你!
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"
所以如果你有一个耗时的操作,都可以利用这两个方式实现异步调用!
写的不对的地方非常希望你能指出!