1 // Copyright 2014 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 // An implementation of WebThread in terms of base::MessageLoop and
6 // base::Thread
7
8 #include "content/child/webthread_impl.h"
9
10 #include "base/bind.h"
11 #include "base/bind_helpers.h"
12 #include "base/message_loop/message_loop.h"
13 #include "base/pending_task.h"
14 #include "base/threading/platform_thread.h"
15
16 namespace content {
17
WebThreadBase()18 WebThreadBase::WebThreadBase() {}
~WebThreadBase()19 WebThreadBase::~WebThreadBase() {}
20
21 class WebThreadBase::TaskObserverAdapter
22 : public base::MessageLoop::TaskObserver {
23 public:
TaskObserverAdapter(WebThread::TaskObserver * observer)24 TaskObserverAdapter(WebThread::TaskObserver* observer)
25 : observer_(observer) {}
26
WillProcessTask(const base::PendingTask & pending_task)27 virtual void WillProcessTask(const base::PendingTask& pending_task) OVERRIDE {
28 observer_->willProcessTask();
29 }
30
DidProcessTask(const base::PendingTask & pending_task)31 virtual void DidProcessTask(const base::PendingTask& pending_task) OVERRIDE {
32 observer_->didProcessTask();
33 }
34
35 private:
36 WebThread::TaskObserver* observer_;
37 };
38
addTaskObserver(TaskObserver * observer)39 void WebThreadBase::addTaskObserver(TaskObserver* observer) {
40 CHECK(isCurrentThread());
41 std::pair<TaskObserverMap::iterator, bool> result = task_observer_map_.insert(
42 std::make_pair(observer, static_cast<TaskObserverAdapter*>(NULL)));
43 if (result.second)
44 result.first->second = new TaskObserverAdapter(observer);
45 base::MessageLoop::current()->AddTaskObserver(result.first->second);
46 }
47
removeTaskObserver(TaskObserver * observer)48 void WebThreadBase::removeTaskObserver(TaskObserver* observer) {
49 CHECK(isCurrentThread());
50 TaskObserverMap::iterator iter = task_observer_map_.find(observer);
51 if (iter == task_observer_map_.end())
52 return;
53 base::MessageLoop::current()->RemoveTaskObserver(iter->second);
54 delete iter->second;
55 task_observer_map_.erase(iter);
56 }
57
WebThreadImpl(const char * name)58 WebThreadImpl::WebThreadImpl(const char* name)
59 : thread_(new base::Thread(name)) {
60 thread_->Start();
61 }
62
postTask(Task * task)63 void WebThreadImpl::postTask(Task* task) {
64 thread_->message_loop()->PostTask(
65 FROM_HERE, base::Bind(&blink::WebThread::Task::run, base::Owned(task)));
66 }
67
postDelayedTask(Task * task,long long delay_ms)68 void WebThreadImpl::postDelayedTask(Task* task, long long delay_ms) {
69 thread_->message_loop()->PostDelayedTask(
70 FROM_HERE,
71 base::Bind(&blink::WebThread::Task::run, base::Owned(task)),
72 base::TimeDelta::FromMilliseconds(delay_ms));
73 }
74
enterRunLoop()75 void WebThreadImpl::enterRunLoop() {
76 CHECK(isCurrentThread());
77 CHECK(!thread_->message_loop()->is_running()); // We don't support nesting.
78 thread_->message_loop()->Run();
79 }
80
exitRunLoop()81 void WebThreadImpl::exitRunLoop() {
82 CHECK(isCurrentThread());
83 CHECK(thread_->message_loop()->is_running());
84 thread_->message_loop()->Quit();
85 }
86
isCurrentThread() const87 bool WebThreadImpl::isCurrentThread() const {
88 return thread_->thread_id() == base::PlatformThread::CurrentId();
89 }
90
threadId() const91 blink::PlatformThreadId WebThreadImpl::threadId() const {
92 return thread_->thread_id();
93 }
94
~WebThreadImpl()95 WebThreadImpl::~WebThreadImpl() {
96 thread_->Stop();
97 }
98
WebThreadImplForMessageLoop(base::MessageLoopProxy * message_loop)99 WebThreadImplForMessageLoop::WebThreadImplForMessageLoop(
100 base::MessageLoopProxy* message_loop)
101 : message_loop_(message_loop),
102 thread_id_(base::PlatformThread::CurrentId()) {}
103
postTask(Task * task)104 void WebThreadImplForMessageLoop::postTask(Task* task) {
105 message_loop_->PostTask(
106 FROM_HERE, base::Bind(&blink::WebThread::Task::run, base::Owned(task)));
107 }
108
postDelayedTask(Task * task,long long delay_ms)109 void WebThreadImplForMessageLoop::postDelayedTask(Task* task,
110 long long delay_ms) {
111 message_loop_->PostDelayedTask(
112 FROM_HERE,
113 base::Bind(&blink::WebThread::Task::run, base::Owned(task)),
114 base::TimeDelta::FromMilliseconds(delay_ms));
115 }
116
enterRunLoop()117 void WebThreadImplForMessageLoop::enterRunLoop() {
118 CHECK(isCurrentThread());
119 // We don't support nesting.
120 CHECK(!base::MessageLoop::current()->is_running());
121 base::MessageLoop::current()->Run();
122 }
123
exitRunLoop()124 void WebThreadImplForMessageLoop::exitRunLoop() {
125 CHECK(isCurrentThread());
126 CHECK(base::MessageLoop::current()->is_running());
127 base::MessageLoop::current()->Quit();
128 }
129
isCurrentThread() const130 bool WebThreadImplForMessageLoop::isCurrentThread() const {
131 return message_loop_->BelongsToCurrentThread();
132 }
133
threadId() const134 blink::PlatformThreadId WebThreadImplForMessageLoop::threadId() const {
135 return thread_id_;
136 }
137
~WebThreadImplForMessageLoop()138 WebThreadImplForMessageLoop::~WebThreadImplForMessageLoop() {}
139
140 } // namespace content
141