Celery+Django+jQuery -- 在jQuery中报告success/failure的celery任务

Celery+Django+jQuery -- Report success/failure of celery task in jQuery

我问了一个类似的问题,关于如何检测和报告 celery 任务何时完成(或失败)。这个问题的重点是如何使用 Django 消息框架来做到这一点,结果证明这对我的目的来说是不必要的。我目前的方法是简单地发送 JSON HttpResponse,并根据收到的 JSON 呈现成功或失败消息。下面是我轮询任务状态的代码:

views.py:

def poll_state(request):
    if request.is_ajax():
        if 'task_id' in request.POST.keys() and request.POST['task_id']:
            task_id = request.POST['task_id']
            task = AsyncResult(task_id)
            if task.successful():
                logger.debug("Successful upload")
                return HttpResponse(json.dumps({"message":"Upload successful!", "state":"SUCCESS"}), content_type='application/json')
            elif task.failed():
                logger.debug("Error in upload")
                return HttpResponse(json.dumps({"message":"Error in upload!", "state":"FAILURE"}), content_type='application/json')
        else:
            logger.info('No task_id in the request')
    else:
        logger.info('Not an ajax request')

    result = task.result
    state = task.state
    response = {}
    response["percent"] = result["percent"]
    response["state"] = state
    json_data = json.dumps(response)
    return HttpResponse(json_data, content_type='application/json')

和相应的jQuery(和Django模板):

{% if task_id %}
    var PollState = function(task_id) {
        jQuery.ajax({
            url: "poll_state",
            type: "POST",
            data: "task_id=" + task_id,
        }).done(function(task) {
            if (task.percent) {
                console.log(task.percent);
                jQuery('.bar').css({'width': task.percent + '%'});
                jQuery('.bar').html(task.percent + '%');
            }
            else if (task.state == "SUCCESS" || task.state == "FAILURE") {
                console.log(task.state);
                jQuery('.status').html(task);
            };
            else {
                jQuery('.status').html(task);
            }
            setTimeout(function () {
                PollState(task_id);
            }, 300);
        });
    }
    PollState('{{ task_id }}');
{% endif %}

下面是任务本身的代码:

tasks.py:

@task(base=DBTask) # a custom base task class
def upload_task(datapoints, user, description): # datapoints is a list of dictionaries, user and description are strings
    from utils.db.databaseinserter import insertIntoDatabase
    for dp_count in insertIntoDatabase(datapoints, user, description): # insertIntoDatabase yields the number of datapoints inserted into the database so far at the end of each iteration
        percent_completion = int(100 * (float(dp_count) / float(len(datapoints))))
        current_task.update_state(state='PROGRESS', meta={'percent':percent_completion})

发生的情况是,如果上传成功,那么一旦上传完成,控制台日志就会被 JSON 成功消息弄得乱七八糟。如果上传有错误,则重新加载会使错误消息使日志混乱。在这两种情况下,poll_state 都在不断发送成功或失败的 HttpResponse。此外,在开始一项新任务后,我仍然收到旧任务信息而不是当前任务信息。唯一一次收到当前任务信息是在服务器首次启动时的第一个任务;对后续任务的任何轮询只会获得第一个任务的信息。

我在这里做错了什么?一段时间以来,我一直在挠头,但想不通。这里的最终目标是在网页上显示某种任务完成或任务失败的通知,并且不会在完成后重复发送 success/failure HttpResponse。

我尝试删除 task.successful()task.failed() 块中的 task_id 会话密钥 poll_state。这确实可以防止 success/failure HttpResponse 被重复发送,但我仍然遇到后续任务没有发送 HttpResponse 的问题;发生的情况是,一旦在第一个任务之后启动了一个新任务,就会发送第一个任务的 HttpResponse,并且当前任务的百分比跟踪不再有效。

经过一番努力和挣扎,事实证明我所有问题的答案实际上是一行。

这个特定问题的解决方案是更改

task_id = request.POST['task_id']

task_id = request.session['task_id']

poll_state 中。