1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "blimp/net/blimp_message_thread_pipe.h"
#include "base/bind_helpers.h"
#include "base/location.h"
#include "base/logging.h"
#include "base/sequenced_task_runner.h"
#include "base/threading/sequenced_task_runner_handle.h"
#include "blimp/common/proto/blimp_message.pb.h"
#include "blimp/net/blimp_message_processor.h"
namespace blimp {
namespace {
class BlimpMessageThreadProxy : public BlimpMessageProcessor {
public:
BlimpMessageThreadProxy(
const scoped_refptr<base::SequencedTaskRunner>& task_runner,
const base::WeakPtr<BlimpMessageThreadPipe>& pipe);
~BlimpMessageThreadProxy() override;
// BlimpMessageProcessor implementation.
void ProcessMessage(scoped_ptr<BlimpMessage> message,
const net::CompletionCallback& callback) override;
private:
// Thread & pipe instance to route messages to/through.
scoped_refptr<base::SequencedTaskRunner> task_runner_;
base::WeakPtr<BlimpMessageThreadPipe> pipe_;
// Used to correctly drop ProcessMessage callbacks if |this| is deleted.
base::WeakPtrFactory<BlimpMessageThreadProxy> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(BlimpMessageThreadProxy);
};
BlimpMessageThreadProxy::BlimpMessageThreadProxy(
const scoped_refptr<base::SequencedTaskRunner>& task_runner,
const base::WeakPtr<BlimpMessageThreadPipe>& pipe)
: task_runner_(task_runner), pipe_(pipe), weak_factory_(this) {}
BlimpMessageThreadProxy::~BlimpMessageThreadProxy() {}
void DispatchProcessMessage(const base::WeakPtr<BlimpMessageThreadPipe> pipe,
scoped_ptr<BlimpMessage> message,
const net::CompletionCallback& callback) {
// Process the message only if the pipe is still active.
if (pipe) {
pipe->target_processor()->ProcessMessage(std::move(message), callback);
}
}
void DispatchProcessMessageCallback(
const scoped_refptr<base::SequencedTaskRunner>& task_runner,
const base::WeakPtr<BlimpMessageThreadProxy> proxy,
const net::CompletionCallback& callback,
int result) {
if (!task_runner->RunsTasksOnCurrentThread()) {
// Bounce the completion to the thread from which ProcessMessage was called.
task_runner->PostTask(
FROM_HERE, base::Bind(&DispatchProcessMessageCallback, task_runner,
proxy, callback, result));
return;
}
// Only dispatch the completion callback if the |proxy| is still live.
if (proxy) {
callback.Run(result);
}
}
void BlimpMessageThreadProxy::ProcessMessage(
scoped_ptr<BlimpMessage> message,
const net::CompletionCallback& callback) {
// If |callback| is non-null then wrap it to be called on this thread, iff
// this proxy instance is still alive at the time.
net::CompletionCallback wrapped_callback;
if (!callback.is_null()) {
wrapped_callback = base::Bind(&DispatchProcessMessageCallback,
base::SequencedTaskRunnerHandle::Get(),
weak_factory_.GetWeakPtr(), callback);
}
// Post |message| to be processed via |pipe_| on |task_runner_|.
task_runner_->PostTask(FROM_HERE,
base::Bind(&DispatchProcessMessage, pipe_,
base::Passed(&message), wrapped_callback));
}
} // namespace
BlimpMessageThreadPipe::BlimpMessageThreadPipe(
const scoped_refptr<base::SequencedTaskRunner>& task_runner)
: target_task_runner_(task_runner), weak_factory_(this) {}
BlimpMessageThreadPipe::~BlimpMessageThreadPipe() {
DCHECK(target_task_runner_->RunsTasksOnCurrentThread());
}
scoped_ptr<BlimpMessageProcessor> BlimpMessageThreadPipe::CreateProxy() {
return make_scoped_ptr(new BlimpMessageThreadProxy(
target_task_runner_, weak_factory_.GetWeakPtr()));
}
void BlimpMessageThreadPipe::set_target_processor(
BlimpMessageProcessor* processor) {
DCHECK(target_task_runner_->RunsTasksOnCurrentThread());
target_processor_ = processor;
}
BlimpMessageProcessor* BlimpMessageThreadPipe::target_processor() const {
DCHECK(target_task_runner_->RunsTasksOnCurrentThread());
return target_processor_;
}
} // namespace blimp
|
