webthread_impl.cc revision 1320f92c476a1ad9d19dba2a48c72b75566198e9 
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
16namespace content {
17
18WebThreadBase::WebThreadBase() {}
19WebThreadBase::~WebThreadBase() {}
20
21class WebThreadBase::TaskObserverAdapter
22    : public base::MessageLoop::TaskObserver {
23 public:
24  TaskObserverAdapter(WebThread::TaskObserver* observer)
25      : observer_(observer) {}
26
27  virtual void WillProcessTask(const base::PendingTask& pending_task) OVERRIDE {
28    observer_->willProcessTask();
29  }
30
31  virtual void DidProcessTask(const base::PendingTask& pending_task) OVERRIDE {
32    observer_->didProcessTask();
33  }
34
35private:
36  WebThread::TaskObserver* observer_;
37};
38
39void 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
48void 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
58WebThreadImpl::WebThreadImpl(const char* name)
59    : thread_(new base::Thread(name)) {
60  thread_->Start();
61}
62
63void WebThreadImpl::postTask(Task* task) {
64  thread_->message_loop()->PostTask(
65      FROM_HERE, base::Bind(&blink::WebThread::Task::run, base::Owned(task)));
66}
67
68void 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
75void WebThreadImpl::enterRunLoop() {
76  CHECK(isCurrentThread());
77  CHECK(!thread_->message_loop()->is_running());  // We don't support nesting.
78  thread_->message_loop()->Run();
79}
80
81void WebThreadImpl::exitRunLoop() {
82  CHECK(isCurrentThread());
83  CHECK(thread_->message_loop()->is_running());
84  thread_->message_loop()->Quit();
85}
86
87bool WebThreadImpl::isCurrentThread() const {
88  return thread_->thread_id() == base::PlatformThread::CurrentId();
89}
90
91blink::PlatformThreadId WebThreadImpl::threadId() const {
92  return thread_->thread_id();
93}
94
95WebThreadImpl::~WebThreadImpl() {
96  thread_->Stop();
97}
98
99WebThreadImplForMessageLoop::WebThreadImplForMessageLoop(
100    base::MessageLoopProxy* message_loop)
101    : message_loop_(message_loop),
102      thread_id_(base::PlatformThread::CurrentId()) {}
103
104void WebThreadImplForMessageLoop::postTask(Task* task) {
105  message_loop_->PostTask(
106      FROM_HERE, base::Bind(&blink::WebThread::Task::run, base::Owned(task)));
107}
108
109void 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
117void 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
124void WebThreadImplForMessageLoop::exitRunLoop() {
125  CHECK(isCurrentThread());
126  CHECK(base::MessageLoop::current()->is_running());
127  base::MessageLoop::current()->Quit();
128}
129
130bool WebThreadImplForMessageLoop::isCurrentThread() const {
131  return message_loop_->BelongsToCurrentThread();
132}
133
134blink::PlatformThreadId WebThreadImplForMessageLoop::threadId() const {
135  return thread_id_;
136}
137
138WebThreadImplForMessageLoop::~WebThreadImplForMessageLoop() {}
139
140}  // namespace content
141