OLD | NEW |
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 "base/sequenced_task_runner.h" | 7 #include "base/sequenced_task_runner.h" |
8 #include "base/thread_task_runner_handle.h" | 8 #include "base/thread_task_runner_handle.h" |
9 #include "base/threading/sequenced_worker_pool.h" | 9 #include "base/threading/sequenced_worker_pool.h" |
10 | 10 |
11 namespace base { | 11 namespace base { |
12 | 12 |
13 // static | 13 // static |
14 scoped_refptr<SequencedTaskRunner> SequencedTaskRunnerHandle::Get() { | 14 scoped_refptr<SequencedTaskRunner> SequencedTaskRunnerHandle::Get() { |
15 // If we are on a worker thread for a SequencedBlockingPool that is running a | 15 // Return the SequencedTaskRunner if found or the SingleThreadedTaskRunner for |
16 // sequenced task, return a SequencedTaskRunner for it. | 16 // the current thread otherwise. |
17 scoped_refptr<base::SequencedWorkerPool> pool = | 17 scoped_refptr<base::SequencedTaskRunner> task_runner = |
18 SequencedWorkerPool::GetWorkerPoolForCurrentThread(); | 18 SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread(); |
19 if (pool) { | 19 if (task_runner) |
20 SequencedWorkerPool::SequenceToken sequence_token = | 20 return task_runner; |
21 SequencedWorkerPool::GetSequenceTokenForCurrentThread(); | |
22 DCHECK(sequence_token.IsValid()); | |
23 DCHECK(pool->IsRunningSequenceOnCurrentThread(sequence_token)); | |
24 return pool->GetSequencedTaskRunner(sequence_token); | |
25 } | |
26 | 21 |
27 // Otherwise, return a SingleThreadTaskRunner for the current thread. | |
28 return base::ThreadTaskRunnerHandle::Get(); | 22 return base::ThreadTaskRunnerHandle::Get(); |
29 } | 23 } |
30 | 24 |
31 // static | 25 // static |
32 bool SequencedTaskRunnerHandle::IsSet() { | 26 bool SequencedTaskRunnerHandle::IsSet() { |
33 return (SequencedWorkerPool::GetWorkerPoolForCurrentThread() && | 27 return SequencedWorkerPool::GetWorkerPoolForCurrentThread() || |
34 SequencedWorkerPool::GetSequenceTokenForCurrentThread().IsValid()) || | |
35 base::ThreadTaskRunnerHandle::IsSet(); | 28 base::ThreadTaskRunnerHandle::IsSet(); |
36 } | 29 } |
37 | 30 |
38 } // namespace base | 31 } // namespace base |
OLD | NEW |