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