Index: blimp/net/tcp_engine_transport.cc |
diff --git a/blimp/net/tcp_engine_transport.cc b/blimp/net/tcp_engine_transport.cc |
index c694a43c1508e8991d1ec05473fe760a8aa93704..70a8b82b11320711c1844a21b3c13b4e04b21c37 100644 |
--- a/blimp/net/tcp_engine_transport.cc |
+++ b/blimp/net/tcp_engine_transport.cc |
@@ -11,9 +11,9 @@ |
#include "base/callback_helpers.h" |
#include "base/location.h" |
#include "base/memory/ptr_util.h" |
-#include "base/single_thread_task_runner.h" |
#include "base/threading/thread_task_runner_handle.h" |
-#include "blimp/net/stream_socket_connection.h" |
+#include "blimp/net/message_port.h" |
+#include "blimp/net/message_port_util.h" |
#include "net/socket/stream_socket.h" |
#include "net/socket/tcp_server_socket.h" |
@@ -51,7 +51,6 @@ void TCPEngineTransport::Connect(const net::CompletionCallback& callback) { |
} |
if (result != net::OK) { |
- // TODO(haibinlu): investigate when we can keep using this server socket. |
server_socket_.reset(); |
} |
@@ -59,11 +58,10 @@ void TCPEngineTransport::Connect(const net::CompletionCallback& callback) { |
base::Bind(callback, result)); |
} |
-std::unique_ptr<BlimpConnection> TCPEngineTransport::TakeConnection() { |
+std::unique_ptr<MessagePort> TCPEngineTransport::TakeMessagePort() { |
DCHECK(connect_callback_.is_null()); |
DCHECK(accepted_socket_); |
- return base::WrapUnique( |
- new StreamSocketConnection(std::move(accepted_socket_))); |
+ return CreateMessagePortForStreamSocket(std::move(accepted_socket_)); |
} |
const char* TCPEngineTransport::GetName() const { |