| OLD | NEW |
| 1 // Copyright 2015 the V8 project authors. All rights reserved. | 1 // Copyright 2015 the V8 project authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "src/cancelable-task.h" | 5 #include "src/cancelable-task.h" |
| 6 | 6 |
| 7 #include "src/base/platform/platform.h" | 7 #include "src/base/platform/platform.h" |
| 8 #include "src/isolate.h" | 8 #include "src/isolate.h" |
| 9 | 9 |
| 10 namespace v8 { | 10 namespace v8 { |
| 11 namespace internal { | 11 namespace internal { |
| 12 | 12 |
| 13 | 13 |
| 14 Cancelable::Cancelable(CancelableTaskManager* parent) | 14 Cancelable::Cancelable(CancelableTaskManager* parent) |
| 15 : parent_(parent), status_(kWaiting), id_(0), cancel_counter_(0) { | 15 : parent_(parent), status_(kWaiting), id_(0), cancel_counter_(0) { |
| 16 id_ = parent->Register(this); | 16 id_ = parent->Register(this); |
| 17 } | 17 } |
| 18 | 18 |
| 19 | 19 |
| 20 Cancelable::~Cancelable() { | 20 Cancelable::~Cancelable() { |
| 21 // The following check is needed to avoid calling an already terminated | 21 // The following check is needed to avoid calling an already terminated |
| 22 // manager object. This happens when the manager cancels all pending tasks | 22 // manager object. This happens when the manager cancels all pending tasks |
| 23 // in {CancelAndWait} only before destroying the manager object. | 23 // in {CancelAndWait} only before destroying the manager object. |
| 24 if (TryRun() || IsRunning()) { | 24 if (TryRun() || IsRunning()) { |
| 25 parent_->RemoveFinishedTask(id_); | 25 parent_->RemoveFinishedTask(id_); |
| 26 } | 26 } |
| 27 } | 27 } |
| 28 | 28 |
| 29 CancelableTaskManager::CancelableTaskManager() : task_id_counter_(0) {} | 29 CancelableTaskManager::CancelableTaskManager() |
| 30 : task_id_counter_(0), canceled_(false) {} |
| 30 | 31 |
| 31 uint32_t CancelableTaskManager::Register(Cancelable* task) { | 32 uint32_t CancelableTaskManager::Register(Cancelable* task) { |
| 32 base::LockGuard<base::Mutex> guard(&mutex_); | 33 base::LockGuard<base::Mutex> guard(&mutex_); |
| 33 uint32_t id = ++task_id_counter_; | 34 uint32_t id = ++task_id_counter_; |
| 34 // The loop below is just used when task_id_counter_ overflows. | 35 // The loop below is just used when task_id_counter_ overflows. |
| 35 while (cancelable_tasks_.count(id) > 0) ++id; | 36 while (cancelable_tasks_.count(id) > 0) ++id; |
| 36 cancelable_tasks_[id] = task; | 37 if (canceled_) { |
| 38 bool successfully_canceled_task = task->Cancel(); |
| 39 CHECK(successfully_canceled_task); |
| 40 } else { |
| 41 cancelable_tasks_[id] = task; |
| 42 } |
| 37 return id; | 43 return id; |
| 38 } | 44 } |
| 39 | 45 |
| 40 | 46 |
| 41 void CancelableTaskManager::RemoveFinishedTask(uint32_t id) { | 47 void CancelableTaskManager::RemoveFinishedTask(uint32_t id) { |
| 42 base::LockGuard<base::Mutex> guard(&mutex_); | 48 base::LockGuard<base::Mutex> guard(&mutex_); |
| 43 size_t removed = cancelable_tasks_.erase(id); | 49 size_t removed = cancelable_tasks_.erase(id); |
| 44 USE(removed); | 50 USE(removed); |
| 45 DCHECK_NE(0, removed); | 51 DCHECK_NE(0, removed); |
| 46 cancelable_tasks_barrier_.NotifyOne(); | 52 cancelable_tasks_barrier_.NotifyOne(); |
| (...skipping 15 matching lines...) Expand all Loading... |
| 62 return false; | 68 return false; |
| 63 } | 69 } |
| 64 | 70 |
| 65 | 71 |
| 66 void CancelableTaskManager::CancelAndWait() { | 72 void CancelableTaskManager::CancelAndWait() { |
| 67 // Clean up all cancelable fore- and background tasks. Tasks are canceled on | 73 // Clean up all cancelable fore- and background tasks. Tasks are canceled on |
| 68 // the way if possible, i.e., if they have not started yet. After each round | 74 // the way if possible, i.e., if they have not started yet. After each round |
| 69 // of canceling we wait for the background tasks that have already been | 75 // of canceling we wait for the background tasks that have already been |
| 70 // started. | 76 // started. |
| 71 base::LockGuard<base::Mutex> guard(&mutex_); | 77 base::LockGuard<base::Mutex> guard(&mutex_); |
| 78 canceled_ = true; |
| 72 | 79 |
| 73 // Cancelable tasks could be running or could potentially register new | 80 // Cancelable tasks could be running or could potentially register new |
| 74 // tasks, requiring a loop here. | 81 // tasks, requiring a loop here. |
| 75 while (!cancelable_tasks_.empty()) { | 82 while (!cancelable_tasks_.empty()) { |
| 76 for (auto it = cancelable_tasks_.begin(); it != cancelable_tasks_.end();) { | 83 for (auto it = cancelable_tasks_.begin(); it != cancelable_tasks_.end();) { |
| 77 auto current = it; | 84 auto current = it; |
| 78 // We need to get to the next element before erasing the current. | 85 // We need to get to the next element before erasing the current. |
| 79 ++it; | 86 ++it; |
| 80 if (current->second->Cancel()) { | 87 if (current->second->Cancel()) { |
| 81 cancelable_tasks_.erase(current); | 88 cancelable_tasks_.erase(current); |
| 82 } | 89 } |
| 83 } | 90 } |
| 84 // Wait for already running background tasks. | 91 // Wait for already running background tasks. |
| 85 if (!cancelable_tasks_.empty()) { | 92 if (!cancelable_tasks_.empty()) { |
| 86 cancelable_tasks_barrier_.Wait(&mutex_); | 93 cancelable_tasks_barrier_.Wait(&mutex_); |
| 87 } | 94 } |
| 88 } | 95 } |
| 89 } | 96 } |
| 90 | 97 |
| 91 | 98 |
| 92 CancelableTask::CancelableTask(Isolate* isolate) | 99 CancelableTask::CancelableTask(Isolate* isolate) |
| 93 : Cancelable(isolate->cancelable_task_manager()), isolate_(isolate) {} | 100 : Cancelable(isolate->cancelable_task_manager()), isolate_(isolate) {} |
| 94 | 101 |
| 95 | 102 |
| 96 CancelableIdleTask::CancelableIdleTask(Isolate* isolate) | 103 CancelableIdleTask::CancelableIdleTask(Isolate* isolate) |
| 97 : Cancelable(isolate->cancelable_task_manager()), isolate_(isolate) {} | 104 : Cancelable(isolate->cancelable_task_manager()), isolate_(isolate) {} |
| 98 | 105 |
| 99 } // namespace internal | 106 } // namespace internal |
| 100 } // namespace v8 | 107 } // namespace v8 |
| OLD | NEW |