chromeos/services/libassistant/abortable_task_list.cc

// Copyright 2021 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chromeos/services/libassistant/abortable_task_list.h"

#include <algorithm>

namespace chromeos {
namespace libassistant {

AbortableTaskList::AbortableTaskList() = default;
AbortableTaskList::~AbortableTaskList() {
  AbortAll();
}

void AbortableTaskList::AddInternal(std::unique_ptr<AbortableTask> task) {
  // We cleanup finished tasks when a new task is added.
  RemoveFinishedTasks();
  tasks_.push_back(std::move(task));
}

void AbortableTaskList::AbortAll() {
  // Cancel all tasks that are not finished yet.
  for (auto& task : tasks_) {
    if (!task->IsFinished())
      task->Abort();
  }

  tasks_.clear();
}

void AbortableTaskList::RemoveFinishedTasks() {
  tasks_.erase(std::remove_if(tasks_.begin(), tasks_.end(),
                              [](const std::unique_ptr<AbortableTask>& task) {
                                return task->IsFinished();
                              }),
               tasks_.end());
}

}  // namespace libassistant
}  // namespace chromeos