GRPC Stream implementation of HeliumStream

blimp/net/engine_connection_manager.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/engine_connection_manager.h"
 
 #include <utility>
 
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "blimp/net/blimp_connection.h"
 #include "blimp/net/blimp_transport.h"
 #include "blimp/net/message_port.h"
+#include "blimp/net/grpc_engine_transport.h"
 #include "blimp/net/tcp_engine_transport.h"
 #include "net/base/ip_address.h"
 #include "net/base/net_errors.h"
 
 namespace blimp {
 
 EngineConnectionManager::EngineConnectionManager(
     ConnectionHandler* connection_handler,
     net::NetLog* net_log)
     : connection_handler_(connection_handler), net_log_(net_log) {
   DCHECK(connection_handler_);
 }
 
 EngineConnectionManager::~EngineConnectionManager() {}
 
 void EngineConnectionManager::ConnectTransport(
     net::IPEndPoint* ip_endpoint,
     EngineTransportType transport_type) {
   switch (transport_type) {
     case EngineTransportType::TCP: {
       transport_ = base::MakeUnique<TCPEngineTransport>(*ip_endpoint, net_log_);
       break;
     }
 
     case EngineTransportType::GRPC: {
-      NOTIMPLEMENTED();
-      // TODO(perumaal): Unimplemented as yet.
-      // transport_ =
-      //      base::MakeUnique<GrpcEngineTransport>(ip_endpoint.ToString());
+      transport_ =
+          base::MakeUnique<GrpcEngineTransport>(ip_endpoint->ToString());
       break;
     }
   }
 
   transport_->Connect(base::Bind(&EngineConnectionManager::OnConnectResult,
                                  base::Unretained(this),
                                  base::Unretained(transport_.get())));
   transport_->GetLocalAddress(ip_endpoint);
 }
 
 void EngineConnectionManager::OnConnectResult(BlimpTransport* transport,
                                               int result) {
   CHECK_EQ(net::OK, result) << "Transport failure:" << transport->GetName();
   connection_handler_->HandleConnection(transport->MakeConnection());
 }
 
 }  // namespace blimp