Refactor BlimpConnection to TCPConnection

blimp/client/core/session/assignment_source.cc

 // Copyright 2016 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/client/core/session/assignment_source.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/bind.h"
@@ skipping 33 matching lines @@
 // Assignment request JSON keys.
 const char kProtocolVersionKey[] = "protocol_version";
 
 // Assignment response JSON keys.
 const char kClientAuthTokenKey[] = "clientToken";
 const char kHostKey[] = "host";
 const char kPortKey[] = "port";
 const char kCertificateKey[] = "certificate";
 
 // Possible arguments for the "--engine-transport" command line parameter.
+const char kGrpcTransportValue[] = "grpc";
 const char kSSLTransportValue[] = "ssl";
 const char kTCPTransportValue[] = "tcp";
 
 class SimpleURLRequestContextGetter : public net::URLRequestContextGetter {
  public:
   SimpleURLRequestContextGetter(
       const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner,
       const scoped_refptr<base::SingleThreadTaskRunner>& file_task_runner)
       : io_task_runner_(io_task_runner), file_task_runner_(file_task_runner) {}
 
@@ skipping 67 matching lines @@
   }
   assignment.engine_endpoint =
       net::IPEndPoint(ip_address, base::checked_cast<uint16_t>(port_parsed));
 
   std::string transport_str =
       cmd_line->GetSwitchValueASCII(switches::kEngineTransport);
   if (transport_str == kSSLTransportValue) {
     assignment.transport_protocol = Assignment::TransportProtocol::SSL;
   } else if (transport_str == kTCPTransportValue) {
     assignment.transport_protocol = Assignment::TransportProtocol::TCP;
+  } else if (transport_str == kGrpcTransportValue) {
+    assignment.transport_protocol = Assignment::TransportProtocol::GRPC;
   } else {
     DLOG(FATAL) << "Invalid engine transport " << transport_str;
     return Assignment();
   }
 
   scoped_refptr<net::X509Certificate> cert;
   if (assignment.transport_protocol == Assignment::TransportProtocol::SSL) {
     base::FilePath cert_path =
         cmd_line->GetSwitchValuePath(switches::kEngineCertPath);
     if (cert_path.empty()) {
@@ skipping 220 matching lines @@
 }
 
 void AssignmentSource::OnJsonParseError(const std::string& error) {
   DLOG(ERROR) << "Error while parsing assigner JSON: " << error;
   base::ResetAndReturn(&callback_)
       .Run(ASSIGNMENT_REQUEST_RESULT_BAD_RESPONSE, Assignment());
 }
 
 }  // namespace client
 }  // namespace blimp