GRPC Stream implementation of HeliumStream

blimp/net/grpc_engine_stream.h

+// 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.
+
+#ifndef BLIMP_NET_GRPC_ENGINE_STREAM_H_
+#define BLIMP_NET_GRPC_ENGINE_STREAM_H_
+
+#include <memory>
+#include <string>
+
+#include "base/callback.h"
+#include "base/macros.h"
+#include "blimp/net/blimp_net_export.h"
+#include "blimp/net/blimp_transport.h"
+#include "net/base/completion_callback.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 "base/callback.h"
+#include "base/command_line.h"
+#include "base/memory/ptr_util.h"
+#include "base/synchronization/lock.h"
+#include "base/threading/thread.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_stream.h"
+
+namespace grpc {
+class Channel;
+class ServerCompletionQueue;
+}  // namespace grpc
+
+namespace blimp {
+
+class GrpcEngineStream : public GrpcStream {
+ public:
+  GrpcEngineStream(const std::string& ip_address,
+                   const net::CompletionCallback& connection_callback);
+
+  void SendMessage(std::unique_ptr<HeliumWrapper> helium_message,
+                   const Stream::SendMessageCallback& callback) override;
+
+  void ReceiveMessage(
+      const Stream::ReceiveMessageCallback& on_receive_cb) override;
+
+  ~GrpcEngineStream() override;
+
+ private:
+  std::string ip_address_;
+  HeliumService::AsyncService service_;
+  std::unique_ptr<grpc::ServerCompletionQueue> completion_queue_;
+  std::unique_ptr<grpc::Server> server_;
+  grpc::ServerContext context_;
+  std::unique_ptr<grpc::ServerAsyncReaderWriter<HeliumWrapper, HeliumWrapper>>
+      stream_;
+};
+
+}  // namespace blimp
+
+#endif  // BLIMP_NET_GRPC_ENGINE_STREAM_H_