| 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 |