Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(558)

Unified Diff: remoting/protocol/message_reader.cc

Issue 10919259: Fix MessageReader not to invoke message callback from done callback. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 8 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « remoting/protocol/message_reader.h ('k') | remoting/protocol/message_reader_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: remoting/protocol/message_reader.cc
diff --git a/remoting/protocol/message_reader.cc b/remoting/protocol/message_reader.cc
index 309f87dd060f577cfdf1dfbe5499d215babf6000..926b05870ed3ded50b67c9368c3041f0f5952b90 100644
--- a/remoting/protocol/message_reader.cc
+++ b/remoting/protocol/message_reader.cc
@@ -9,6 +9,7 @@
#include "base/compiler_specific.h"
#include "base/location.h"
#include "base/thread_task_runner_handle.h"
+#include "base/single_thread_task_runner.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
#include "net/socket/socket.h"
@@ -88,23 +89,23 @@ void MessageReader::OnDataReceived(net::IOBuffer* data, int data_size) {
// Get list of all new messages first, and then call the callback
// for all of them.
- std::vector<CompoundBuffer*> new_messages;
while (true) {
CompoundBuffer* buffer = message_decoder_.GetNextMessage();
if (!buffer)
break;
- new_messages.push_back(buffer);
+ pending_messages_++;
+ base::ThreadTaskRunnerHandle::Get()->PostTask(
+ FROM_HERE,
+ base::Bind(&MessageReader::RunCallback,
+ weak_factory_.GetWeakPtr(),
+ base::Passed(scoped_ptr<CompoundBuffer>(buffer))));
}
+}
- pending_messages_ += new_messages.size();
-
- for (std::vector<CompoundBuffer*>::iterator it = new_messages.begin();
- it != new_messages.end(); ++it) {
- message_received_callback_.Run(
- scoped_ptr<CompoundBuffer>(*it),
- base::Bind(&MessageReader::OnMessageDone,
- weak_factory_.GetWeakPtr()));
- }
+void MessageReader::RunCallback(scoped_ptr<CompoundBuffer> message) {
+ message_received_callback_.Run(
+ message.Pass(), base::Bind(&MessageReader::OnMessageDone,
+ weak_factory_.GetWeakPtr()));
}
void MessageReader::OnMessageDone() {
@@ -112,8 +113,8 @@ void MessageReader::OnMessageDone() {
pending_messages_--;
DCHECK_GE(pending_messages_, 0);
- if (!read_pending_)
- DoRead(); // Start next read if necessary.
+ // Start next read if necessary.
+ DoRead();
}
} // namespace protocol
« no previous file with comments | « remoting/protocol/message_reader.h ('k') | remoting/protocol/message_reader_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698