| 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 "platform/scheduler/child/webthread_impl_for_worker_scheduler.h" | 5 #include "platform/scheduler/child/webthread_impl_for_worker_scheduler.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
| 10 #include "base/single_thread_task_runner.h" | 10 #include "base/single_thread_task_runner.h" |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 59 } | 59 } |
| 60 | 60 |
| 61 void WebThreadImplForWorkerScheduler::InitOnThread( | 61 void WebThreadImplForWorkerScheduler::InitOnThread( |
| 62 base::WaitableEvent* completion) { | 62 base::WaitableEvent* completion) { |
| 63 // TODO(alexclarke): Do we need to unify virtual time for workers and the | 63 // TODO(alexclarke): Do we need to unify virtual time for workers and the |
| 64 // main thread? | 64 // main thread? |
| 65 task_runner_delegate_ = SchedulerTqmDelegateImpl::Create( | 65 task_runner_delegate_ = SchedulerTqmDelegateImpl::Create( |
| 66 thread_->message_loop(), base::MakeUnique<base::DefaultTickClock>()); | 66 thread_->message_loop(), base::MakeUnique<base::DefaultTickClock>()); |
| 67 worker_scheduler_ = CreateWorkerScheduler(); | 67 worker_scheduler_ = CreateWorkerScheduler(); |
| 68 worker_scheduler_->Init(); | 68 worker_scheduler_->Init(); |
| 69 task_runner_ = worker_scheduler_->DefaultTaskQueue(); | 69 task_queue_ = worker_scheduler_->DefaultTaskQueue(); |
| 70 idle_task_runner_ = worker_scheduler_->IdleTaskRunner(); | 70 idle_task_runner_ = worker_scheduler_->IdleTaskRunner(); |
| 71 web_scheduler_.reset(new WebSchedulerImpl( | 71 web_scheduler_.reset(new WebSchedulerImpl( |
| 72 worker_scheduler_.get(), worker_scheduler_->IdleTaskRunner(), | 72 worker_scheduler_.get(), worker_scheduler_->IdleTaskRunner(), |
| 73 worker_scheduler_->DefaultTaskQueue(), | 73 worker_scheduler_->DefaultTaskQueue(), |
| 74 worker_scheduler_->DefaultTaskQueue())); | 74 worker_scheduler_->DefaultTaskQueue())); |
| 75 base::MessageLoop::current()->AddDestructionObserver(this); | 75 base::MessageLoop::current()->AddDestructionObserver(this); |
| 76 web_task_runner_ = WebTaskRunnerImpl::Create(task_runner_); | 76 web_task_runner_ = WebTaskRunnerImpl::Create(task_queue_); |
| 77 completion->Signal(); | 77 completion->Signal(); |
| 78 } | 78 } |
| 79 | 79 |
| 80 void WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread( | 80 void WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread( |
| 81 base::WaitableEvent* completion) { | 81 base::WaitableEvent* completion) { |
| 82 task_runner_delegate_->RestoreDefaultTaskRunner(); | 82 task_runner_delegate_->RestoreDefaultTaskRunner(); |
| 83 completion->Signal(); | 83 completion->Signal(); |
| 84 } | 84 } |
| 85 | 85 |
| 86 void WebThreadImplForWorkerScheduler::WillDestroyCurrentMessageLoop() { | 86 void WebThreadImplForWorkerScheduler::WillDestroyCurrentMessageLoop() { |
| 87 task_runner_ = nullptr; | 87 task_queue_ = nullptr; |
| 88 idle_task_runner_ = nullptr; | 88 idle_task_runner_ = nullptr; |
| 89 web_scheduler_.reset(); | 89 web_scheduler_.reset(); |
| 90 worker_scheduler_.reset(); | 90 worker_scheduler_.reset(); |
| 91 web_task_runner_.Reset(); | 91 web_task_runner_.Reset(); |
| 92 } | 92 } |
| 93 | 93 |
| 94 std::unique_ptr<scheduler::WorkerScheduler> | 94 std::unique_ptr<scheduler::WorkerScheduler> |
| 95 WebThreadImplForWorkerScheduler::CreateWorkerScheduler() { | 95 WebThreadImplForWorkerScheduler::CreateWorkerScheduler() { |
| 96 return WorkerScheduler::Create(task_runner_delegate_); | 96 return WorkerScheduler::Create(task_runner_delegate_); |
| 97 } | 97 } |
| 98 | 98 |
| 99 blink::PlatformThreadId WebThreadImplForWorkerScheduler::ThreadId() const { | 99 blink::PlatformThreadId WebThreadImplForWorkerScheduler::ThreadId() const { |
| 100 return thread_->GetThreadId(); | 100 return thread_->GetThreadId(); |
| 101 } | 101 } |
| 102 | 102 |
| 103 blink::WebScheduler* WebThreadImplForWorkerScheduler::Scheduler() const { | 103 blink::WebScheduler* WebThreadImplForWorkerScheduler::Scheduler() const { |
| 104 return web_scheduler_.get(); | 104 return web_scheduler_.get(); |
| 105 } | 105 } |
| 106 | 106 |
| 107 base::SingleThreadTaskRunner* WebThreadImplForWorkerScheduler::GetTaskRunner() | 107 base::SingleThreadTaskRunner* WebThreadImplForWorkerScheduler::GetTaskRunner() |
| 108 const { | 108 const { |
| 109 return task_runner_.get(); | 109 return task_queue_.get(); |
| 110 } | 110 } |
| 111 | 111 |
| 112 SingleThreadIdleTaskRunner* WebThreadImplForWorkerScheduler::GetIdleTaskRunner() | 112 SingleThreadIdleTaskRunner* WebThreadImplForWorkerScheduler::GetIdleTaskRunner() |
| 113 const { | 113 const { |
| 114 return idle_task_runner_.get(); | 114 return idle_task_runner_.get(); |
| 115 } | 115 } |
| 116 | 116 |
| 117 blink::WebTaskRunner* WebThreadImplForWorkerScheduler::GetWebTaskRunner() { | 117 blink::WebTaskRunner* WebThreadImplForWorkerScheduler::GetWebTaskRunner() { |
| 118 return web_task_runner_.Get(); | 118 return web_task_runner_.Get(); |
| 119 } | 119 } |
| 120 | 120 |
| 121 void WebThreadImplForWorkerScheduler::AddTaskObserverInternal( | 121 void WebThreadImplForWorkerScheduler::AddTaskObserverInternal( |
| 122 base::MessageLoop::TaskObserver* observer) { | 122 base::MessageLoop::TaskObserver* observer) { |
| 123 worker_scheduler_->AddTaskObserver(observer); | 123 worker_scheduler_->AddTaskObserver(observer); |
| 124 } | 124 } |
| 125 | 125 |
| 126 void WebThreadImplForWorkerScheduler::RemoveTaskObserverInternal( | 126 void WebThreadImplForWorkerScheduler::RemoveTaskObserverInternal( |
| 127 base::MessageLoop::TaskObserver* observer) { | 127 base::MessageLoop::TaskObserver* observer) { |
| 128 worker_scheduler_->RemoveTaskObserver(observer); | 128 worker_scheduler_->RemoveTaskObserver(observer); |
| 129 } | 129 } |
| 130 | 130 |
| 131 } // namespace scheduler | 131 } // namespace scheduler |
| 132 } // namespace blink | 132 } // namespace blink |
| OLD | NEW |