Mojo: WebSocket interface now reuses the DataPipe for subsequent sends or

mojo/services/html_viewer/websockethandle_impl.h

Index: mojo/services/html_viewer/websockethandle_impl.h
diff --git a/mojo/services/html_viewer/websockethandle_impl.h b/mojo/services/html_viewer/websockethandle_impl.h
index a0408d3012bd4b28915e025286f0ffd5aaf8d49e..5ed8ff9a9ec49a3c14724a6140c0978c88865a98 100644
--- a/mojo/services/html_viewer/websockethandle_impl.h
+++ b/mojo/services/html_viewer/websockethandle_impl.h
@@ -6,7 +6,6 @@
 #define MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_
 
 #include "base/memory/scoped_ptr.h"
-#include "base/memory/weak_ptr.h"
 #include "mojo/common/handle_watcher.h"
 #include "mojo/services/public/interfaces/network/web_socket.mojom.h"
 #include "third_party/WebKit/public/platform/WebSocketHandle.h"
@@ -14,6 +13,7 @@
 namespace mojo {
 class NetworkService;
 class WebSocketClientImpl;
+class WebSocketWriteQueue;
 
 // Implements WebSocketHandle by talking to the mojo WebSocket interface.
 class WebSocketHandleImpl : public blink::WebSocketHandle {
@@ -38,11 +38,19 @@ class WebSocketHandleImpl : public blink::WebSocketHandle {
   virtual void close(unsigned short code,
                      const blink::WebString& reason) OVERRIDE;
 
+  // Called when we finished writing to |send_stream_|.
+  void DidWriteToSendStream(bool fin,
+                            WebSocketHandle::MessageType type,
+                            uint32_t num_bytes,
+                            const char* data);
+
   // Called when the socket is closed.
   void Disconnect();
 
   WebSocketPtr web_socket_;
   scoped_ptr<WebSocketClientImpl> client_;
+  ScopedDataPipeProducerHandle send_stream_;
+  scoped_ptr<WebSocketWriteQueue> write_queue_;
   // True if close() was called.
   bool did_close_;