mojo/services/network/public/cpp/web_socket_read_queue.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef MOJO_SERVICES_NETWORK_PUBLIC_CPP_WEB_SOCKET_READ_QUEUE_H_
   6 #define MOJO_SERVICES_NETWORK_PUBLIC_CPP_WEB_SOCKET_READ_QUEUE_H_
   7
   8 #include "base/callback.h"
   9 #include "base/memory/scoped_vector.h"
  10 #include "base/memory/weak_ptr.h"
  11 #include "mojo/message_pump/handle_watcher.h"
  12 #include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h"
  13
  14 namespace mojo {
  15
  16 // This class simplifies the handling of multiple Reads on a DataPipe. It reads
  17 // the data in the expected chunk size, calling the callback once a full chunk
  18 // is ready. Callbacks are owned by this class, and are guaranteed not to be
  19 // called after this class is destroyed.
  20 // See also: WebSocketWriteQueue
  21 class WebSocketReadQueue {
  22  public:
  23   explicit WebSocketReadQueue(DataPipeConsumerHandle handle);
  24   ~WebSocketReadQueue();
  25
  26   void Read(uint32_t num_bytes,
  27             const base::Callback<void(const char*)>& callback);
  28
  29  private:
  30   struct Operation;
  31
  32   void TryToRead();
  33   void Wait();
  34   void OnHandleReady(MojoResult result);
  35
  36   DataPipeConsumerHandle handle_;
  37   common::HandleWatcher handle_watcher_;
  38   ScopedVector<Operation> queue_;
  39   bool is_busy_;
  40   base::WeakPtrFactory<WebSocketReadQueue> weak_factory_;
  41 };
  42
  43 }  // namespace mojo
  44
  45 #endif  // MOJO_SERVICES_NETWORK_PUBLIC_CPP_WEB_SOCKET_READ_QUEUE_H_