OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #include "base/threading/sequenced_worker_pool.h" | 5 #include "base/threading/sequenced_worker_pool.h" |
6 | 6 |
7 #include <list> | 7 #include <list> |
8 #include <map> | 8 #include <map> |
9 #include <set> | 9 #include <set> |
10 #include <utility> | 10 #include <utility> |
11 #include <vector> | 11 #include <vector> |
12 | 12 |
13 #include "base/atomicops.h" | 13 #include "base/atomicops.h" |
14 #include "base/callback.h" | 14 #include "base/callback.h" |
15 #include "base/compiler_specific.h" | 15 #include "base/compiler_specific.h" |
16 #include "base/logging.h" | 16 #include "base/logging.h" |
17 #include "base/memory/linked_ptr.h" | 17 #include "base/memory/linked_ptr.h" |
18 #include "base/message_loop_proxy.h" | 18 #include "base/message_loop_proxy.h" |
19 #include "base/metrics/histogram.h" | 19 #include "base/metrics/histogram.h" |
20 #include "base/stl_util.h" | 20 #include "base/stl_util.h" |
21 #include "base/stringprintf.h" | 21 #include "base/stringprintf.h" |
22 #include "base/synchronization/condition_variable.h" | 22 #include "base/synchronization/condition_variable.h" |
23 #include "base/synchronization/lock.h" | 23 #include "base/synchronization/lock.h" |
24 #include "base/threading/platform_thread.h" | 24 #include "base/threading/platform_thread.h" |
25 #include "base/threading/sequenced_worker_pool_task_runner.h" | |
26 #include "base/threading/simple_thread.h" | 25 #include "base/threading/simple_thread.h" |
27 #include "base/time.h" | 26 #include "base/time.h" |
28 #include "base/tracked_objects.h" | 27 #include "base/tracked_objects.h" |
29 | 28 |
30 #if defined(OS_MACOSX) | 29 #if defined(OS_MACOSX) |
31 #include "base/mac/scoped_nsautorelease_pool.h" | 30 #include "base/mac/scoped_nsautorelease_pool.h" |
32 #endif | 31 #endif |
33 | 32 |
34 namespace base { | 33 namespace base { |
35 | 34 |
36 namespace { | 35 namespace { |
37 | 36 |
38 struct SequencedTask { | 37 struct SequencedTask { |
39 SequencedTask() | 38 SequencedTask() |
40 : sequence_token_id(0), | 39 : sequence_token_id(0), |
41 shutdown_behavior(SequencedWorkerPool::BLOCK_SHUTDOWN) {} | 40 shutdown_behavior(SequencedWorkerPool::BLOCK_SHUTDOWN) {} |
42 | 41 |
43 ~SequencedTask() {} | 42 ~SequencedTask() {} |
44 | 43 |
45 int sequence_token_id; | 44 int sequence_token_id; |
46 SequencedWorkerPool::WorkerShutdown shutdown_behavior; | 45 SequencedWorkerPool::WorkerShutdown shutdown_behavior; |
47 tracked_objects::Location location; | 46 tracked_objects::Location location; |
48 Closure task; | 47 Closure task; |
49 }; | 48 }; |
50 | 49 |
| 50 // SequencedWorkerPoolSequencedTaskRunner ------------------------------------ |
| 51 // A SequencedTaskRunner which posts tasks to a SequencedWorkerPool with a |
| 52 // fixed sequence token. |
| 53 // |
| 54 // Note that this class is RefCountedThreadSafe (inherited from TaskRunner). |
| 55 class SequencedWorkerPoolSequencedTaskRunner : public SequencedTaskRunner { |
| 56 public: |
| 57 SequencedWorkerPoolSequencedTaskRunner( |
| 58 const scoped_refptr<SequencedWorkerPool>& pool, |
| 59 SequencedWorkerPool::SequenceToken token); |
| 60 |
| 61 // TaskRunner implementation |
| 62 virtual bool PostDelayedTask(const tracked_objects::Location& from_here, |
| 63 const Closure& task, |
| 64 int64 delay_ms) OVERRIDE; |
| 65 virtual bool PostDelayedTask(const tracked_objects::Location& from_here, |
| 66 const Closure& task, |
| 67 TimeDelta delay) OVERRIDE; |
| 68 virtual bool RunsTasksOnCurrentThread() const OVERRIDE; |
| 69 |
| 70 // SequencedTaskRunner implementation |
| 71 virtual bool PostNonNestableDelayedTask( |
| 72 const tracked_objects::Location& from_here, |
| 73 const Closure& task, |
| 74 int64 delay_ms) OVERRIDE; |
| 75 virtual bool PostNonNestableDelayedTask( |
| 76 const tracked_objects::Location& from_here, |
| 77 const Closure& task, |
| 78 TimeDelta delay) OVERRIDE; |
| 79 |
| 80 private: |
| 81 virtual ~SequencedWorkerPoolSequencedTaskRunner(); |
| 82 |
| 83 // Helper function for posting a delayed task. Asserts that the delay is |
| 84 // zero because non-zero delays are not yet supported. |
| 85 bool PostDelayedTaskAssertZeroDelay( |
| 86 const tracked_objects::Location& from_here, |
| 87 const Closure& task, |
| 88 int64 delay_ms); |
| 89 bool PostDelayedTaskAssertZeroDelay( |
| 90 const tracked_objects::Location& from_here, |
| 91 const Closure& task, |
| 92 TimeDelta delay); |
| 93 |
| 94 const scoped_refptr<SequencedWorkerPool> pool_; |
| 95 |
| 96 const SequencedWorkerPool::SequenceToken token_; |
| 97 |
| 98 DISALLOW_COPY_AND_ASSIGN(SequencedWorkerPoolSequencedTaskRunner); |
| 99 }; |
| 100 |
| 101 SequencedWorkerPoolSequencedTaskRunner::SequencedWorkerPoolSequencedTaskRunner( |
| 102 const scoped_refptr<SequencedWorkerPool>& pool, |
| 103 SequencedWorkerPool::SequenceToken token) |
| 104 : pool_(pool), |
| 105 token_(token) { |
| 106 } |
| 107 |
| 108 SequencedWorkerPoolSequencedTaskRunner:: |
| 109 ~SequencedWorkerPoolSequencedTaskRunner() { |
| 110 } |
| 111 |
| 112 bool SequencedWorkerPoolSequencedTaskRunner::PostDelayedTask( |
| 113 const tracked_objects::Location& from_here, |
| 114 const Closure& task, |
| 115 int64 delay_ms) { |
| 116 return PostDelayedTaskAssertZeroDelay(from_here, task, delay_ms); |
| 117 } |
| 118 |
| 119 bool SequencedWorkerPoolSequencedTaskRunner::PostDelayedTask( |
| 120 const tracked_objects::Location& from_here, |
| 121 const Closure& task, |
| 122 TimeDelta delay) { |
| 123 return PostDelayedTaskAssertZeroDelay(from_here, task, delay); |
| 124 } |
| 125 |
| 126 bool SequencedWorkerPoolSequencedTaskRunner::RunsTasksOnCurrentThread() const { |
| 127 return pool_->IsRunningSequenceOnCurrentThread(token_); |
| 128 } |
| 129 |
| 130 bool SequencedWorkerPoolSequencedTaskRunner::PostNonNestableDelayedTask( |
| 131 const tracked_objects::Location& from_here, |
| 132 const Closure& task, |
| 133 int64 delay_ms) { |
| 134 return PostDelayedTaskAssertZeroDelay(from_here, task, delay_ms); |
| 135 } |
| 136 |
| 137 bool SequencedWorkerPoolSequencedTaskRunner::PostNonNestableDelayedTask( |
| 138 const tracked_objects::Location& from_here, |
| 139 const Closure& task, |
| 140 TimeDelta delay) { |
| 141 return PostDelayedTaskAssertZeroDelay(from_here, task, delay); |
| 142 } |
| 143 |
| 144 bool SequencedWorkerPoolSequencedTaskRunner::PostDelayedTaskAssertZeroDelay( |
| 145 const tracked_objects::Location& from_here, |
| 146 const Closure& task, |
| 147 int64 delay_ms) { |
| 148 // TODO(francoisk777@gmail.com): Change the following two statements once |
| 149 // SequencedWorkerPool supports non-zero delays. |
| 150 DCHECK_EQ(delay_ms, 0) |
| 151 << "SequencedWorkerPoolSequencedTaskRunner does not yet support non-zero" |
| 152 " delays"; |
| 153 return pool_->PostSequencedWorkerTask(token_, from_here, task); |
| 154 } |
| 155 |
| 156 bool SequencedWorkerPoolSequencedTaskRunner::PostDelayedTaskAssertZeroDelay( |
| 157 const tracked_objects::Location& from_here, |
| 158 const Closure& task, |
| 159 TimeDelta delay) { |
| 160 return PostDelayedTaskAssertZeroDelay(from_here, |
| 161 task, |
| 162 delay.InMillisecondsRoundedUp()); |
| 163 } |
| 164 |
51 } // namespace | 165 } // namespace |
52 | 166 |
53 // Worker --------------------------------------------------------------------- | 167 // Worker --------------------------------------------------------------------- |
54 | 168 |
55 class SequencedWorkerPool::Worker : public SimpleThread { | 169 class SequencedWorkerPool::Worker : public SimpleThread { |
56 public: | 170 public: |
57 // Hold a (cyclic) ref to |worker_pool|, since we want to keep it | 171 // Hold a (cyclic) ref to |worker_pool|, since we want to keep it |
58 // around as long as we are running. | 172 // around as long as we are running. |
59 Worker(const scoped_refptr<SequencedWorkerPool>& worker_pool, | 173 Worker(const scoped_refptr<SequencedWorkerPool>& worker_pool, |
60 int thread_number, | 174 int thread_number, |
(...skipping 693 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
754 return inner_->GetSequenceToken(); | 868 return inner_->GetSequenceToken(); |
755 } | 869 } |
756 | 870 |
757 SequencedWorkerPool::SequenceToken SequencedWorkerPool::GetNamedSequenceToken( | 871 SequencedWorkerPool::SequenceToken SequencedWorkerPool::GetNamedSequenceToken( |
758 const std::string& name) { | 872 const std::string& name) { |
759 return inner_->GetNamedSequenceToken(name); | 873 return inner_->GetNamedSequenceToken(name); |
760 } | 874 } |
761 | 875 |
762 scoped_refptr<SequencedTaskRunner> SequencedWorkerPool::GetSequencedTaskRunner( | 876 scoped_refptr<SequencedTaskRunner> SequencedWorkerPool::GetSequencedTaskRunner( |
763 SequenceToken token) { | 877 SequenceToken token) { |
764 return new SequencedWorkerPoolTaskRunner(this, token); | 878 return new SequencedWorkerPoolSequencedTaskRunner(this, token); |
765 } | 879 } |
766 | 880 |
767 bool SequencedWorkerPool::PostWorkerTask( | 881 bool SequencedWorkerPool::PostWorkerTask( |
768 const tracked_objects::Location& from_here, | 882 const tracked_objects::Location& from_here, |
769 const Closure& task) { | 883 const Closure& task) { |
770 return inner_->PostTask(NULL, SequenceToken(), BLOCK_SHUTDOWN, | 884 return inner_->PostTask(NULL, SequenceToken(), BLOCK_SHUTDOWN, |
771 from_here, task); | 885 from_here, task); |
772 } | 886 } |
773 | 887 |
774 bool SequencedWorkerPool::PostWorkerTaskWithShutdownBehavior( | 888 bool SequencedWorkerPool::PostWorkerTaskWithShutdownBehavior( |
(...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
839 void SequencedWorkerPool::SignalHasWorkForTesting() { | 953 void SequencedWorkerPool::SignalHasWorkForTesting() { |
840 inner_->SignalHasWorkForTesting(); | 954 inner_->SignalHasWorkForTesting(); |
841 } | 955 } |
842 | 956 |
843 void SequencedWorkerPool::Shutdown() { | 957 void SequencedWorkerPool::Shutdown() { |
844 DCHECK(constructor_message_loop_->BelongsToCurrentThread()); | 958 DCHECK(constructor_message_loop_->BelongsToCurrentThread()); |
845 inner_->Shutdown(); | 959 inner_->Shutdown(); |
846 } | 960 } |
847 | 961 |
848 } // namespace base | 962 } // namespace base |
OLD | NEW |