From 044548bcfff4af4fdf1150ba6e65501b55feaa94 Mon Sep 17 00:00:00 2001 From: Stephanie Wang Date: Sat, 2 Dec 2017 14:42:16 -0800 Subject: [PATCH] Mark the killed as done outside of loop (#1284) --- src/local_scheduler/local_scheduler_algorithm.cc | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/local_scheduler/local_scheduler_algorithm.cc b/src/local_scheduler/local_scheduler_algorithm.cc index 0e90d3af4..1c00b83c2 100644 --- a/src/local_scheduler/local_scheduler_algorithm.cc +++ b/src/local_scheduler/local_scheduler_algorithm.cc @@ -415,11 +415,12 @@ void finish_killed_task(LocalSchedulerState *state, ARROW_CHECK_OK(status); ARROW_CHECK_OK(state->plasma_conn->Seal(object_id.to_plasma_id())); } - if (state->db != NULL) { - Task *task = Task_alloc(spec, task_spec_size, TASK_STATUS_DONE, - get_db_client_id(state->db)); - task_table_update(state->db, task, NULL, NULL, NULL); - } + } + /* Mark the task as done. */ + if (state->db != NULL) { + Task *task = Task_alloc(spec, task_spec_size, TASK_STATUS_DONE, + get_db_client_id(state->db)); + task_table_update(state->db, task, NULL, NULL, NULL); } }