GRPC Stream implementation of HeliumStream

blimp/net/engine_connection_manager.h

 // 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.
 
 #ifndef BLIMP_NET_ENGINE_CONNECTION_MANAGER_H_
 #define BLIMP_NET_ENGINE_CONNECTION_MANAGER_H_
 
 #include <memory>
 #include <vector>
 
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "blimp/net/blimp_net_export.h"
 #include "blimp/net/connection_handler.h"
 #include "net/base/ip_endpoint.h"
 #include "net/log/net_log.h"
 
 namespace blimp {
 
+struct AssignmentOptions;
 class BlimpConnection;
 class BlimpTransport;
 class BlimpEngineTransport;
 
 // Coordinates the channel creation and authentication workflows for
 // incoming (Engine) network connections.
 //
 // TODO(kmarshall): Add rate limiting and abuse handling logic.
 class BLIMP_NET_EXPORT EngineConnectionManager {
  public:
   enum class EngineTransportType { TCP, GRPC };
 
   // Caller is responsible for ensuring that |connection_handler| outlives
   // |this|.
   explicit EngineConnectionManager(ConnectionHandler* connection_handler,
                                    net::NetLog* net_log);
 
   ~EngineConnectionManager();
 
   // Adds a transport and accepts new BlimpConnections from it as fast as they
   // arrive. The |ip_endpoint| will receive the actual port-number if the
   // provided one is not available for listening.
-  void ConnectTransport(net::IPEndPoint* ip_endpoint,
+  void ConnectTransport(AssignmentOptions* assignment_options,
                         EngineTransportType transport_type);
 
  private:
   // Callback invoked by |transport| to indicate that it has a connection
   // ready to be authenticated.
   void OnConnectResult(BlimpTransport* transport, int result);
 
   ConnectionHandler* connection_handler_;
   net::NetLog* net_log_;
   std::unique_ptr<BlimpEngineTransport> transport_;
 
   DISALLOW_COPY_AND_ASSIGN(EngineConnectionManager);
 };
 
 }  // namespace blimp
 
 #endif  // BLIMP_NET_ENGINE_CONNECTION_MANAGER_H_