修复 huey 队列不会释放数据库连接的问题,是用法不对

This commit is contained in:
virusdefender 2015-12-12 11:36:02 +08:00
parent a91d2fb4f1
commit 84d390362b

View File

@ -1,9 +1,9 @@
# coding=utf-8 # coding=utf-8
from huey.djhuey import task from huey.djhuey import db_task
from judge_dispatcher.tasks import JudgeDispatcher from judge_dispatcher.tasks import JudgeDispatcher
@task() @db_task()
def _judge(submission, time_limit, memory_limit, test_case_id, is_waiting_task=False): def _judge(submission, time_limit, memory_limit, test_case_id, is_waiting_task=False):
JudgeDispatcher(submission, time_limit, memory_limit, test_case_id).judge(is_waiting_task) JudgeDispatcher(submission, time_limit, memory_limit, test_case_id).judge(is_waiting_task)