实际上 TornadoFutures
不支持取消(文档)。此外,即使使用with_timeout
,超时的作业仍在运行,只是没有等待其结果。
唯一的方法,正如我如何在龙卷风中取消挂起的异步任务,超时?, 是以这种方式实现逻辑,它可以被取消(带有一些标志或其他)。
例子:
- 工作是一个简单的异步睡眠
/
列出工作
/add/TIME
添加新作业 - 以秒为单位的 TIME - 指定睡眠时间
/cancel/ID
取消工作
代码可能如下所示:
from tornado.ioloop import IOLoop
from tornado import gen, web
from time import time
class Job():
def __init__(self, run_sec):
self.run_sec = int(run_sec)
self.start_time = None
self.end_time = None
self._cancelled = False
@gen.coroutine
def run(self):
""" Some job
The job is simple: sleep for a given number of seconds.
It could be implemented as:
yield gen.sleep(self.run_sec)
but this way makes it not cancellable, so
it is divided: run 1s sleep, run_sec times
"""
self.start_time = time()
deadline = self.start_time + self.run_sec
while not self._cancelled:
yield gen.sleep(1)
if time() >= deadline:
break
self.end_time = time()
def cancel(self):
""" Cancels job
Returns None on success,
raises Exception on error:
if job is already cancelled or done
"""
if self._cancelled:
raise Exception('Job is already cancelled')
if self.end_time is not None:
raise Exception('Job is already done')
self._cancelled = True
def get_state(self):
if self._cancelled:
if self.end_time is None:
# job might be running still
# and will be stopped on the next while check
return 'CANCELING...'
else:
return 'CANCELLED'
elif self.end_time is None:
return 'RUNNING...'
elif self.start_time is None:
# actually this never will shown
# as after creation, job is immediately started
return 'NOT STARTED'
else:
return 'DONE'
class MainHandler(web.RequestHandler):
def get(self, op=None, param=None):
if op == 'add':
# add new job
new_job = Job(run_sec=param)
self.application.jobs.append(new_job)
new_job.run()
self.write('Job added')
elif op == 'cancel':
# cancel job - stop running
self.application.jobs[int(param)].cancel()
self.write('Job cancelled')
else:
# list jobs
self.write('<pre>') # this is so ugly... ;P
self.write('ID\tRUNSEC\tSTART_TIME\tSTATE\tEND_TIME\n')
for idx, job in enumerate(self.application.jobs):
self.write('%s\t%s\t%s\t%s\t%s\n' % (
idx, job.run_sec, job.start_time,
job.get_state(), job.end_time
))
class MyApplication(web.Application):
def __init__(self):
# to store tasks
self.jobs = []
super(MyApplication, self).__init__([
(r"/", MainHandler),
(r"/(add)/(\d*)", MainHandler),
(r"/(cancel)/(\d*)", MainHandler),
])
if __name__ == "__main__":
MyApplication().listen(8888)
IOLoop.current().start()
添加几个工作:
for a in `seq 12 120`; do curl http://127.0.0.1:8888/add/$a; done
然后取消一些......注意 - 它只需要龙卷风。
这个例子很简单,gen.sleep
意味着你的任务很繁重。当然,并非所有工作都像以可取消方式实现的那样简单。