Index: trunk/src/base/threading/sequenced_worker_pool.cc |
=================================================================== |
--- trunk/src/base/threading/sequenced_worker_pool.cc (revision 210433) |
+++ trunk/src/base/threading/sequenced_worker_pool.cc (working copy) |
@@ -10,21 +10,22 @@ |
#include <utility> |
#include <vector> |
-#include "base/atomicops.h" |
+#include "base/atomic_sequence_num.h" |
#include "base/callback.h" |
#include "base/compiler_specific.h" |
#include "base/critical_closure.h" |
#include "base/debug/trace_event.h" |
+#include "base/lazy_instance.h" |
#include "base/logging.h" |
#include "base/memory/linked_ptr.h" |
#include "base/message_loop/message_loop_proxy.h" |
-#include "base/metrics/histogram.h" |
#include "base/stl_util.h" |
#include "base/strings/stringprintf.h" |
#include "base/synchronization/condition_variable.h" |
#include "base/synchronization/lock.h" |
#include "base/threading/platform_thread.h" |
#include "base/threading/simple_thread.h" |
+#include "base/threading/thread_local.h" |
#include "base/threading/thread_restrictions.h" |
#include "base/time/time.h" |
#include "base/tracked_objects.h" |
@@ -33,6 +34,10 @@ |
#include "base/mac/scoped_nsautorelease_pool.h" |
#endif |
+#if !defined(OS_NACL) |
+#include "base/metrics/histogram.h" |
+#endif |
+ |
namespace base { |
namespace { |
@@ -213,6 +218,10 @@ |
static_cast<uint64>(reinterpret_cast<intptr_t>(pool)); |
} |
+base::LazyInstance<base::ThreadLocalPointer< |
+ SequencedWorkerPool::SequenceToken> > g_lazy_tls_ptr = |
+ LAZY_INSTANCE_INITIALIZER; |
+ |
} // namespace |
// Worker --------------------------------------------------------------------- |
@@ -382,8 +391,9 @@ |
// The last sequence number used. Managed by GetSequenceToken, since this |
// only does threadsafe increment operations, you do not need to hold the |
- // lock. |
- volatile subtle::Atomic32 last_sequence_number_; |
+ // lock. This is class-static to make SequenceTokens issued by |
+ // GetSequenceToken unique across SequencedWorkerPool instances. |
+ static base::StaticAtomicSequenceNumber g_last_sequence_number_; |
// This lock protects |everything in this class|. Do not read or modify |
// anything without holding this lock. Do not block while holding this |
@@ -479,6 +489,10 @@ |
} |
void SequencedWorkerPool::Worker::Run() { |
+ // Store a pointer to the running sequence in thread local storage for |
+ // static function access. |
+ g_lazy_tls_ptr.Get().Set(&running_sequence_); |
+ |
// Just jump back to the Inner object to run the thread, since it has all the |
// tracking information and queues. It might be more natural to implement |
// using DelegateSimpleThread and have Inner implement the Delegate to avoid |
@@ -497,7 +511,6 @@ |
const std::string& thread_name_prefix, |
TestingObserver* observer) |
: worker_pool_(worker_pool), |
- last_sequence_number_(0), |
lock_(), |
has_work_cv_(&lock_), |
can_shutdown_cv_(&lock_), |
@@ -532,9 +545,9 @@ |
SequencedWorkerPool::SequenceToken |
SequencedWorkerPool::Inner::GetSequenceToken() { |
- subtle::Atomic32 result = |
- subtle::NoBarrier_AtomicIncrement(&last_sequence_number_, 1); |
- return SequenceToken(static_cast<int>(result)); |
+ // Need to add one because StaticAtomicSequenceNumber starts at zero, which |
+ // is used as a sentinel value in SequenceTokens. |
+ return SequenceToken(g_last_sequence_number_.GetNext() + 1); |
} |
SequencedWorkerPool::SequenceToken |
@@ -615,7 +628,7 @@ |
ThreadMap::const_iterator found = threads_.find(PlatformThread::CurrentId()); |
if (found == threads_.end()) |
return false; |
- return found->second->running_sequence().Equals(sequence_token); |
+ return sequence_token.Equals(found->second->running_sequence()); |
} |
// See https://code.google.com/p/chromium/issues/detail?id=168415 |
@@ -674,8 +687,10 @@ |
while (!CanShutdown()) |
can_shutdown_cv_.Wait(); |
} |
+#if !defined(OS_NACL) |
UMA_HISTOGRAM_TIMES("SequencedWorkerPool.ShutdownDelayTime", |
TimeTicks::Now() - shutdown_wait_begin); |
+#endif |
} |
void SequencedWorkerPool::Inner::ThreadLoop(Worker* this_worker) { |
@@ -872,8 +887,10 @@ |
std::vector<Closure>* delete_these_outside_lock) { |
lock_.AssertAcquired(); |
+#if !defined(OS_NACL) |
UMA_HISTOGRAM_COUNTS_100("SequencedWorkerPool.TaskCount", |
static_cast<int>(pending_tasks_.size())); |
+#endif |
// Find the next task with a sequence token that's not currently in use. |
// If the token is in use, that means another thread is running something |
@@ -958,8 +975,10 @@ |
// Track the number of tasks we had to skip over to see if we should be |
// making this more efficient. If this number ever becomes large or is |
// frequently "some", we should consider the optimization above. |
+#if !defined(OS_NACL) |
UMA_HISTOGRAM_COUNTS_100("SequencedWorkerPool.UnrunnableTaskCount", |
unrunnable_tasks); |
+#endif |
return status; |
} |
@@ -1088,8 +1107,20 @@ |
blocking_shutdown_pending_task_count_ == 0; |
} |
+base::StaticAtomicSequenceNumber |
+SequencedWorkerPool::Inner::g_last_sequence_number_; |
+ |
// SequencedWorkerPool -------------------------------------------------------- |
+// static |
+SequencedWorkerPool::SequenceToken |
+SequencedWorkerPool::GetSequenceTokenForCurrentThread() { |
+ SequencedWorkerPool::SequenceToken* token = g_lazy_tls_ptr.Get().Get(); |
+ if (!token) |
+ return SequenceToken(); |
+ return *token; |
+} |
+ |
SequencedWorkerPool::SequencedWorkerPool( |
size_t max_threads, |
const std::string& thread_name_prefix) |