天天看點

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"
           

是以如果你有一個耗時的操作,都可以利用這兩個方式實作異步調用!

寫的不對的地方非常希望你能指出!