Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(266)

Side by Side Diff: base/threading/sequenced_task_runner_handle.cc

Issue 2042383004: Introduce TaskRunnerHandle. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: +tests Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 1 // Copyright 2015 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_task_runner_handle.h" 5 #include "base/threading/sequenced_task_runner_handle.h"
6 6
7 #include <utility> 7 #include <utility>
8 8
9 #include "base/lazy_instance.h"
10 #include "base/logging.h"
11 #include "base/sequenced_task_runner.h" 9 #include "base/sequenced_task_runner.h"
12 #include "base/threading/sequenced_worker_pool.h"
13 #include "base/threading/thread_local.h"
14 #include "base/threading/thread_task_runner_handle.h"
15 10
16 namespace base { 11 namespace base {
17 12
18 namespace {
19
20 base::LazyInstance<base::ThreadLocalPointer<SequencedTaskRunnerHandle>>::Leaky
21 lazy_tls_ptr = LAZY_INSTANCE_INITIALIZER;
22
23 } // namespace
24
25 // static
26 scoped_refptr<SequencedTaskRunner> SequencedTaskRunnerHandle::Get() {
27 // Return the registered SequencedTaskRunner, if any.
28 const SequencedTaskRunnerHandle* handle = lazy_tls_ptr.Pointer()->Get();
29 if (handle) {
30 // Various modes of setting SequencedTaskRunnerHandle don't combine.
31 DCHECK(!base::ThreadTaskRunnerHandle::IsSet());
32 DCHECK(!SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread());
33 return handle->task_runner_;
34 }
35
36 // Return the SequencedTaskRunner obtained from SequencedWorkerPool, if any.
37 scoped_refptr<base::SequencedTaskRunner> task_runner =
38 SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread();
39 if (task_runner) {
40 DCHECK(!base::ThreadTaskRunnerHandle::IsSet());
41 return task_runner;
42 }
43
44 // Return the SingleThreadTaskRunner for the current thread otherwise.
45 return base::ThreadTaskRunnerHandle::Get();
46 }
47
48 // static
49 bool SequencedTaskRunnerHandle::IsSet() {
50 return lazy_tls_ptr.Pointer()->Get() ||
51 SequencedWorkerPool::GetWorkerPoolForCurrentThread() ||
52 base::ThreadTaskRunnerHandle::IsSet();
53 }
54
55 SequencedTaskRunnerHandle::SequencedTaskRunnerHandle( 13 SequencedTaskRunnerHandle::SequencedTaskRunnerHandle(
56 scoped_refptr<SequencedTaskRunner> task_runner) 14 scoped_refptr<SequencedTaskRunner> task_runner)
57 : task_runner_(std::move(task_runner)) { 15 : task_scope_(std::move(task_runner)) {}
58 DCHECK(task_runner_->RunsTasksOnCurrentThread());
59 DCHECK(!SequencedTaskRunnerHandle::IsSet());
60 lazy_tls_ptr.Pointer()->Set(this);
61 }
62 16
63 SequencedTaskRunnerHandle::~SequencedTaskRunnerHandle() { 17 SequencedTaskRunnerHandle::~SequencedTaskRunnerHandle() = default;
64 DCHECK(task_runner_->RunsTasksOnCurrentThread());
65 DCHECK_EQ(lazy_tls_ptr.Pointer()->Get(), this);
66 lazy_tls_ptr.Pointer()->Set(nullptr);
67 }
68 18
69 } // namespace base 19 } // namespace base
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698