Decouple Blimp transport output from BlimpConnections using MessagePort.

blimp/net/tcp_client_transport.cc

 // 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/tcp_client_transport.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/callback.h"
 #include "base/callback_helpers.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
+#include "blimp/net/message_port.h"
+#include "blimp/net/message_port_util.h"
 #include "blimp/net/stream_socket_connection.h"
 #include "net/socket/client_socket_factory.h"
 #include "net/socket/stream_socket.h"
 #include "net/socket/tcp_client_socket.h"
 
 namespace blimp {
 
 TCPClientTransport::TCPClientTransport(const net::IPEndPoint& ip_endpoint,
                                        net::NetLog* net_log)
     : ip_endpoint_(ip_endpoint),
@@ skipping 19 matching lines @@
       &TCPClientTransport::OnTCPConnectComplete, base::Unretained(this));
 
   int result = socket_->Connect(completion_callback);
   if (result == net::ERR_IO_PENDING) {
     return;
   }
 
   OnTCPConnectComplete(result);
 }
 
-std::unique_ptr<BlimpConnection> TCPClientTransport::TakeConnection() {
+std::unique_ptr<MessagePort> TCPClientTransport::TakeMessagePort() {
   DCHECK(connect_callback_.is_null());
   DCHECK(socket_);
-  return base::MakeUnique<StreamSocketConnection>(std::move(socket_));
+  return CreateMessagePortForStreamSocket(std::move(socket_));
 }
 
 const char* TCPClientTransport::GetName() const {
   return "TCP";
 }
 
 void TCPClientTransport::OnTCPConnectComplete(int result) {
   DCHECK_NE(net::ERR_IO_PENDING, result);
   OnConnectComplete(result);
 }
@@ skipping 12 matching lines @@
 void TCPClientTransport::SetSocket(std::unique_ptr<net::StreamSocket> socket) {
   DCHECK(socket);
   socket_ = std::move(socket);
 }
 
 net::ClientSocketFactory* TCPClientTransport::socket_factory() const {
   return socket_factory_;
 }
 
 }  // namespace blimp