base: Change WeakPtr to use SequenceChecker instead of ThreadChecker.

base/sequence_checker_impl.cc

Index: base/sequence_checker_impl.cc
diff --git a/base/sequence_checker_impl.cc b/base/sequence_checker_impl.cc
index 24d9ed99a147c7c5a8d2db69557b3f891851cfa4..ff2811da8ae95330a2e215f9101d100bbe3a16e6 100644
--- a/base/sequence_checker_impl.cc
+++ b/base/sequence_checker_impl.cc
@@ -1,31 +1,52 @@
-// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/sequence_checker_impl.h"
 
-#include "base/sequenced_task_runner.h"
-
 namespace base {
 
-SequenceCheckerImpl::SequenceCheckerImpl(
-    const scoped_refptr<SequencedTaskRunner>& sequenced_task_runner)
-    : sequenced_task_runner_(sequenced_task_runner) {}
+SequenceCheckerImpl::SequenceCheckerImpl()
+    : sequence_token_assigned_(false),
+      associated_pool_(NULL) {
+  EnsureSequenceTokenAssigned();
+}
 
 SequenceCheckerImpl::~SequenceCheckerImpl() {}
 
 bool SequenceCheckerImpl::CalledOnValidSequence() const {
+  EnsureSequenceTokenAssigned();
   AutoLock auto_lock(lock_);
-  return sequenced_task_runner_.get() ?
-      sequenced_task_runner_->RunsTasksOnCurrentThread() :
-      thread_checker_.CalledOnValidThread();
+
+  if (!associated_pool_)
+    return thread_checker_.CalledOnValidThread();
+
+  if (SequencedWorkerPool::Get() != associated_pool_)
+    return false;
+
+  return associated_pool_->IsRunningSequenceOnCurrentThread(sequence_token_);
 }
 
-void SequenceCheckerImpl::ChangeSequence(
-    const scoped_refptr<SequencedTaskRunner>& sequenced_task_runner) {
+void SequenceCheckerImpl::DetachFromSequence() {
   AutoLock auto_lock(lock_);
-  sequenced_task_runner_ = sequenced_task_runner;
   thread_checker_.DetachFromThread();
+  sequence_token_assigned_ = false;
+  associated_pool_ = NULL;
+  sequence_token_ = SequencedWorkerPool::SequenceToken();
+}
+
+void SequenceCheckerImpl::EnsureSequenceTokenAssigned() const {
+  AutoLock auto_lock(lock_);
+  if (sequence_token_assigned_)
+    return;
+
+  sequence_token_assigned_ = true;
+  associated_pool_ = SequencedWorkerPool::Get();
+
+  // If can't get sequence token, lock to the thread ID.

[vandebo (ex-Chrome), 2013/06/28 21:00:06]

If you got a pool, probably should DCHECK that you get a token.

[tommycli, 2013/06/28 22:37:07]

Done.

+  if (!associated_pool_ ||
+      !associated_pool_->GetCurrentThreadSequenceToken(&sequence_token_))
+    associated_pool_ = NULL;
 }
 
 }  // namespace base