Mojo service implementation for HTTP server - part 2

mojo/services/network/http_connection_impl.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 MOJO_SERVICES_NETWORK_HTTP_CONNECTION_IMPL_H_
 #define MOJO_SERVICES_NETWORK_HTTP_CONNECTION_IMPL_H_
 
+#include <set>
 #include <string>
 
 #include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
 #include "mojo/services/network/public/interfaces/http_connection.mojom.h"
+#include "mojo/services/network/public/interfaces/url_loader.mojom.h"
 #include "third_party/mojo/src/mojo/public/cpp/bindings/binding.h"
 #include "third_party/mojo/src/mojo/public/cpp/bindings/error_handler.h"
 
 namespace net {
 class HttpServerRequestInfo;
 }
 
 namespace mojo {
 
 class HttpServerImpl;
 
 class HttpConnectionImpl : public HttpConnection,
                            public ErrorHandler {
  public:
   // |owner| must outlive this object.
   HttpConnectionImpl(int connection_id,
                      HttpServerImpl* owner,
                      HttpConnectionDelegatePtr delegate,
                      HttpConnectionPtr* connection);
 
   ~HttpConnectionImpl() override;
 
   void OnReceivedHttpRequest(const net::HttpServerRequestInfo& info);
   void OnReceivedWebSocketRequest(const net::HttpServerRequestInfo& info);
   void OnReceivedWebSocketMessage(const std::string& data);
 
  private:
+  class SimpleDataPipeReader;
+
   // HttpConnection implementation.
   void SetSendBufferSize(uint32_t size,
                          const SetSendBufferSizeCallback& callback) override;
   void SetReceiveBufferSize(
       uint32_t size,
       const SetReceiveBufferSizeCallback& callback) override;
 
   // ErrorHandler implementation.
   void OnConnectionError() override;
 
+  void OnFinishedReadingResponseBody(URLResponsePtr response_ptr,
+                                     SimpleDataPipeReader* reader,
+                                     scoped_ptr<std::string> body);
+
   const int connection_id_;
   HttpServerImpl* const owner_;
   HttpConnectionDelegatePtr delegate_;
   Binding<HttpConnection> binding_;
+  // Owns its elements.
+  std::set<SimpleDataPipeReader*> response_body_readers_;
+  bool encountered_connection_error_;
 
   DISALLOW_COPY_AND_ASSIGN(HttpConnectionImpl);
 };
 
 }  // namespace mojo
 
 #endif  // MOJO_SERVICES_NETWORK_HTTP_CONNECTION_IMPL_H_