GRPC Stream implementation of HeliumStream

blimp/net/grpc_client_transport.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_CLIENT_TRANSPORT_H_
+#define BLIMP_NET_GRPC_CLIENT_TRANSPORT_H_
+
+#include <memory>
+#include <string>
+
+#include "base/callback.h"
+#include "base/macros.h"
+#include "blimp/common/assignment_options.h"
+#include "blimp/net/blimp_net_export.h"
+#include "blimp/net/blimp_transport.h"
+#include "net/base/completion_callback.h"
+
+namespace grpc {
+class Client;
+}
+
+namespace blimp {
+
+class GrpcClientStream;
+class GrpcConnection;
+
+// Class that interfaces a |BlimpTransport| to a |GrpcClientStream| through a
+// |GrpcConnection|.
+class BLIMP_NET_EXPORT GrpcClientTransport : public BlimpTransport {
+ public:
+  // Creates a new |GrpcClientStream| based on the given assignment options.
+  explicit GrpcClientTransport(const AssignmentOptions& assignment_options);
+
+  // Connects using the |GrpcClientStream| and calls |callback| on the IO
+  // thread on connection success.
+  void Connect(const net::CompletionCallback& callback) override;
+
+  // Creates a new |GrpcConnection| and transfers ownership to caller.
+  std::unique_ptr<BlimpConnection> MakeConnection() override;
+
+  const char* GetName() const override;
+  ~GrpcClientTransport() override;
+
+ private:
+  AssignmentOptions assignment_options_;
+  std::unique_ptr<GrpcConnection> grpc_connection_;
+};
+
+}  // namespace blimp
+
+#endif  // BLIMP_NET_GRPC_CLIENT_TRANSPORT_H_