GRPC Stream implementation of HeliumStream

blimp/net/grpc_engine_transport.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/net/grpc_engine_transport.h"
+
+#include <grpc++/security/server_credentials.h>
+#include <grpc++/server.h>
+#include <grpc++/server_builder.h>
+#include <grpc++/server_context.h>
+#include <grpc/grpc.h>
+
+#include <queue>
+#include <string>
+#include <utility>
+
+#include "base/command_line.h"
+#include "base/memory/ptr_util.h"
+#include "base/synchronization/lock.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "blimp/common/logging.h"
+#include "blimp/common/proto/blimp_message.pb.h"
+#include "blimp/common/proto/helium_service.grpc.pb.h"
+#include "blimp/net/blimp_message_processor.h"
+#include "blimp/net/blimp_message_pump.h"
+#include "blimp/net/grpc_connection.h"
+#include "blimp/net/grpc_engine_stream.h"
+#include "net/base/net_errors.h"
+
+namespace blimp {
+
+GrpcEngineTransport::GrpcEngineTransport(
+    const AssignmentOptions& assignment_options)
+    : BlimpEngineTransport(), assignment_options_(assignment_options) {
+  DVLOG(1) << "gRPC Engine Initialized @ "
+           << assignment_options.engine_endpoint.ToString();
+}
+
+void GrpcEngineTransport::Connect(const net::CompletionCallback& callback) {
+  std::unique_ptr<GrpcEngineStream> engine_stream =
+      base::MakeUnique<GrpcEngineStream>(assignment_options_, callback);
+  assignment_options_ = engine_stream->GetAssignmentOptions();
+  grpc_connection_ = base::MakeUnique<GrpcConnection>(std::move(engine_stream));
+}
+
+std::unique_ptr<BlimpConnection> GrpcEngineTransport::MakeConnection() {
+  DVLOG(1) << "Grpc Engine finished setup";
+  return std::move(grpc_connection_);
+}
+
+const char* GrpcEngineTransport::GetName() const {
+  return "GRPC";
+}
+
+void GrpcEngineTransport::GetAssignmentOptions(
+    AssignmentOptions* assignment_options) const {
+  *assignment_options = assignment_options_;
+}
+
+GrpcEngineTransport::~GrpcEngineTransport() {}
+
+}  // namespace blimp