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 |