| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium 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 // CancelableTaskTracker posts tasks (in the form of a Closure) to a | 5 // CancelableTaskTracker posts tasks (in the form of a Closure) to a |
| 6 // TaskRunner, and is able to cancel the task later if it's not needed | 6 // TaskRunner, and is able to cancel the task later if it's not needed |
| 7 // anymore. On destruction, CancelableTaskTracker will cancel all | 7 // anymore. On destruction, CancelableTaskTracker will cancel all |
| 8 // tracked tasks. | 8 // tracked tasks. |
| 9 // | 9 // |
| 10 // Each cancelable task can be associated with a reply (also a Closure). After | 10 // Each cancelable task can be associated with a reply (also a Closure). After |
| (...skipping 25 matching lines...) Expand all Loading... |
| 36 #ifndef BASE_TASK_CANCELABLE_TASK_TRACKER_H_ | 36 #ifndef BASE_TASK_CANCELABLE_TASK_TRACKER_H_ |
| 37 #define BASE_TASK_CANCELABLE_TASK_TRACKER_H_ | 37 #define BASE_TASK_CANCELABLE_TASK_TRACKER_H_ |
| 38 | 38 |
| 39 #include <stdint.h> | 39 #include <stdint.h> |
| 40 | 40 |
| 41 #include <utility> | 41 #include <utility> |
| 42 | 42 |
| 43 #include "base/base_export.h" | 43 #include "base/base_export.h" |
| 44 #include "base/bind.h" | 44 #include "base/bind.h" |
| 45 #include "base/callback.h" | 45 #include "base/callback.h" |
| 46 #include "base/containers/hash_tables.h" | 46 #include "base/containers/small_map.h" |
| 47 #include "base/macros.h" | 47 #include "base/macros.h" |
| 48 #include "base/memory/weak_ptr.h" | 48 #include "base/memory/weak_ptr.h" |
| 49 #include "base/post_task_and_reply_with_result_internal.h" | 49 #include "base/post_task_and_reply_with_result_internal.h" |
| 50 #include "base/sequence_checker.h" | 50 #include "base/sequence_checker.h" |
| 51 | 51 |
| 52 namespace tracked_objects { | 52 namespace tracked_objects { |
| 53 class Location; | 53 class Location; |
| 54 } // namespace tracked_objects | 54 } // namespace tracked_objects |
| 55 | 55 |
| 56 namespace base { | 56 namespace base { |
| (...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 135 void TryCancelAll(); | 135 void TryCancelAll(); |
| 136 | 136 |
| 137 // Returns true iff there are in-flight tasks that are still being | 137 // Returns true iff there are in-flight tasks that are still being |
| 138 // tracked. | 138 // tracked. |
| 139 bool HasTrackedTasks() const; | 139 bool HasTrackedTasks() const; |
| 140 | 140 |
| 141 private: | 141 private: |
| 142 void Track(TaskId id, CancellationFlag* flag); | 142 void Track(TaskId id, CancellationFlag* flag); |
| 143 void Untrack(TaskId id); | 143 void Untrack(TaskId id); |
| 144 | 144 |
| 145 hash_map<TaskId, CancellationFlag*> task_flags_; | 145 // Typically the number of tasks are 0-2 and occationally 3-4. But since |
| 146 // this is a general API that could be used in unexpected ways, use a |
| 147 // small_map instead of a flat_map to avoid falling over if there are many |
| 148 // tasks. |
| 149 small_map<std::map<TaskId, CancellationFlag*>, 4> task_flags_; |
| 146 | 150 |
| 147 TaskId next_id_; | 151 TaskId next_id_; |
| 148 SequenceChecker sequence_checker_; | 152 SequenceChecker sequence_checker_; |
| 149 | 153 |
| 150 WeakPtrFactory<CancelableTaskTracker> weak_factory_; | 154 WeakPtrFactory<CancelableTaskTracker> weak_factory_; |
| 151 | 155 |
| 152 DISALLOW_COPY_AND_ASSIGN(CancelableTaskTracker); | 156 DISALLOW_COPY_AND_ASSIGN(CancelableTaskTracker); |
| 153 }; | 157 }; |
| 154 | 158 |
| 155 } // namespace base | 159 } // namespace base |
| 156 | 160 |
| 157 #endif // BASE_TASK_CANCELABLE_TASK_TRACKER_H_ | 161 #endif // BASE_TASK_CANCELABLE_TASK_TRACKER_H_ |
| OLD | NEW |