| 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 "components/scheduler/child/webthread_impl_for_worker_scheduler.h" | 5 #include "components/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/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
| 10 #include "base/synchronization/waitable_event.h" | 10 #include "base/synchronization/waitable_event.h" |
| (...skipping 11 matching lines...) Expand all Loading... |
| 22 const char* name) | 22 const char* name) |
| 23 : WebThreadImplForWorkerScheduler(name, base::Thread::Options()) {} | 23 : WebThreadImplForWorkerScheduler(name, base::Thread::Options()) {} |
| 24 | 24 |
| 25 WebThreadImplForWorkerScheduler::WebThreadImplForWorkerScheduler( | 25 WebThreadImplForWorkerScheduler::WebThreadImplForWorkerScheduler( |
| 26 const char* name, | 26 const char* name, |
| 27 base::Thread::Options options) | 27 base::Thread::Options options) |
| 28 : thread_(new base::Thread(name ? name : std::string())) { | 28 : thread_(new base::Thread(name ? name : std::string())) { |
| 29 bool started = thread_->StartWithOptions(options); | 29 bool started = thread_->StartWithOptions(options); |
| 30 CHECK(started); | 30 CHECK(started); |
| 31 thread_task_runner_ = thread_->task_runner(); | 31 thread_task_runner_ = thread_->task_runner(); |
| 32 } |
| 32 | 33 |
| 34 void WebThreadImplForWorkerScheduler::Init() { |
| 33 base::WaitableEvent completion(false, false); | 35 base::WaitableEvent completion(false, false); |
| 34 thread_task_runner_->PostTask( | 36 thread_task_runner_->PostTask( |
| 35 FROM_HERE, base::Bind(&WebThreadImplForWorkerScheduler::InitOnThread, | 37 FROM_HERE, base::Bind(&WebThreadImplForWorkerScheduler::InitOnThread, |
| 36 base::Unretained(this), &completion)); | 38 base::Unretained(this), &completion)); |
| 37 completion.Wait(); | 39 completion.Wait(); |
| 38 } | 40 } |
| 39 | 41 |
| 40 WebThreadImplForWorkerScheduler::~WebThreadImplForWorkerScheduler() { | 42 WebThreadImplForWorkerScheduler::~WebThreadImplForWorkerScheduler() { |
| 41 base::WaitableEvent completion(false, false); | 43 base::WaitableEvent completion(false, false); |
| 42 // Restore the original task runner so that the thread can tear itself down. | 44 // Restore the original task runner so that the thread can tear itself down. |
| 43 thread_task_runner_->PostTask( | 45 thread_task_runner_->PostTask( |
| 44 FROM_HERE, | 46 FROM_HERE, |
| 45 base::Bind(&WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread, | 47 base::Bind(&WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread, |
| 46 base::Unretained(this), &completion)); | 48 base::Unretained(this), &completion)); |
| 47 completion.Wait(); | 49 completion.Wait(); |
| 48 thread_->Stop(); | 50 thread_->Stop(); |
| 49 } | 51 } |
| 50 | 52 |
| 51 void WebThreadImplForWorkerScheduler::InitOnThread( | 53 void WebThreadImplForWorkerScheduler::InitOnThread( |
| 52 base::WaitableEvent* completion) { | 54 base::WaitableEvent* completion) { |
| 53 // TODO(alexclarke): Do we need to unify virtual time for workers and the | 55 // TODO(alexclarke): Do we need to unify virtual time for workers and the |
| 54 // main thread? | 56 // main thread? |
| 55 task_runner_delegate_ = SchedulerTqmDelegateImpl::Create( | 57 task_runner_delegate_ = SchedulerTqmDelegateImpl::Create( |
| 56 thread_->message_loop(), make_scoped_ptr(new base::DefaultTickClock())); | 58 thread_->message_loop(), make_scoped_ptr(new base::DefaultTickClock())); |
| 57 worker_scheduler_ = WorkerScheduler::Create(task_runner_delegate_); | 59 worker_scheduler_ = WorkerScheduler::Create(task_runner_delegate_); |
| 58 worker_scheduler_->Init(); | 60 worker_scheduler_->Init(); |
| 59 task_runner_ = worker_scheduler_->DefaultTaskRunner(); | 61 if (UseThreadTaskRunnerAsDefault()) |
| 62 task_runner_ = thread_task_runner_; |
| 63 else |
| 64 task_runner_ = worker_scheduler_->DefaultTaskRunner(); |
| 60 idle_task_runner_ = worker_scheduler_->IdleTaskRunner(); | 65 idle_task_runner_ = worker_scheduler_->IdleTaskRunner(); |
| 61 web_scheduler_.reset(new WebSchedulerImpl( | 66 web_scheduler_.reset(new WebSchedulerImpl( |
| 62 worker_scheduler_.get(), worker_scheduler_->IdleTaskRunner(), | 67 worker_scheduler_.get(), worker_scheduler_->IdleTaskRunner(), |
| 63 worker_scheduler_->DefaultTaskRunner(), | 68 worker_scheduler_->DefaultTaskRunner(), |
| 64 worker_scheduler_->DefaultTaskRunner())); | 69 worker_scheduler_->DefaultTaskRunner())); |
| 65 base::MessageLoop::current()->AddDestructionObserver(this); | 70 base::MessageLoop::current()->AddDestructionObserver(this); |
| 66 web_task_runner_ = make_scoped_ptr(new WebTaskRunnerImpl(task_runner_)); | 71 web_task_runner_ = make_scoped_ptr(new WebTaskRunnerImpl(task_runner_)); |
| 67 completion->Signal(); | 72 completion->Signal(); |
| 68 } | 73 } |
| 69 | 74 |
| 70 void WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread( | 75 void WebThreadImplForWorkerScheduler::RestoreTaskRunnerOnThread( |
| 71 base::WaitableEvent* completion) { | 76 base::WaitableEvent* completion) { |
| 72 task_runner_delegate_->RestoreDefaultTaskRunner(); | 77 task_runner_delegate_->RestoreDefaultTaskRunner(); |
| 73 completion->Signal(); | 78 completion->Signal(); |
| 74 } | 79 } |
| 75 | 80 |
| 76 void WebThreadImplForWorkerScheduler::WillDestroyCurrentMessageLoop() { | 81 void WebThreadImplForWorkerScheduler::WillDestroyCurrentMessageLoop() { |
| 77 task_runner_ = nullptr; | 82 task_runner_ = nullptr; |
| 78 idle_task_runner_ = nullptr; | 83 idle_task_runner_ = nullptr; |
| 79 web_scheduler_.reset(); | 84 web_scheduler_.reset(); |
| 80 worker_scheduler_.reset(); | 85 worker_scheduler_.reset(); |
| 81 } | 86 } |
| 82 | 87 |
| 88 bool WebThreadImplForWorkerScheduler::UseThreadTaskRunnerAsDefault() const { |
| 89 return false; |
| 90 } |
| 91 |
| 83 blink::PlatformThreadId WebThreadImplForWorkerScheduler::threadId() const { | 92 blink::PlatformThreadId WebThreadImplForWorkerScheduler::threadId() const { |
| 84 return thread_->GetThreadId(); | 93 return thread_->GetThreadId(); |
| 85 } | 94 } |
| 86 | 95 |
| 87 blink::WebScheduler* WebThreadImplForWorkerScheduler::scheduler() const { | 96 blink::WebScheduler* WebThreadImplForWorkerScheduler::scheduler() const { |
| 88 return web_scheduler_.get(); | 97 return web_scheduler_.get(); |
| 89 } | 98 } |
| 90 | 99 |
| 91 base::SingleThreadTaskRunner* WebThreadImplForWorkerScheduler::TaskRunner() | 100 base::SingleThreadTaskRunner* WebThreadImplForWorkerScheduler::TaskRunner() |
| 92 const { | 101 const { |
| (...skipping 13 matching lines...) Expand all Loading... |
| 106 base::MessageLoop::TaskObserver* observer) { | 115 base::MessageLoop::TaskObserver* observer) { |
| 107 worker_scheduler_->AddTaskObserver(observer); | 116 worker_scheduler_->AddTaskObserver(observer); |
| 108 } | 117 } |
| 109 | 118 |
| 110 void WebThreadImplForWorkerScheduler::RemoveTaskObserverInternal( | 119 void WebThreadImplForWorkerScheduler::RemoveTaskObserverInternal( |
| 111 base::MessageLoop::TaskObserver* observer) { | 120 base::MessageLoop::TaskObserver* observer) { |
| 112 worker_scheduler_->RemoveTaskObserver(observer); | 121 worker_scheduler_->RemoveTaskObserver(observer); |
| 113 } | 122 } |
| 114 | 123 |
| 115 } // namespace scheduler | 124 } // namespace scheduler |
| OLD | NEW |