net/spdy/spdy_io_buffer.h

   1 // Copyright (c) 2011 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 NET_SPDY_SPDY_IO_BUFFER_H_
   6 #define NET_SPDY_SPDY_IO_BUFFER_H_
   7 #pragma once
   8
   9 #include "base/memory/ref_counted.h"
  10 #include "net/base/io_buffer.h"
  11 #include "net/base/net_export.h"
  12 #include "net/spdy/spdy_stream.h"
  13
  14 namespace net {
  15
  16 // A class for managing SPDY IO buffers.  These buffers need to be prioritized
  17 // so that the SpdySession sends them in the right order.  Further, they need
  18 // to track the SpdyStream which they are associated with so that incremental
  19 // completion of the IO can notify the appropriate stream of completion.
  20 class NET_EXPORT_PRIVATE SpdyIOBuffer {
  21  public:
  22   // Constructor
  23   // |buffer| is the actual data buffer.
  24   // |size| is the size of the data buffer.
  25   // |priority| is the priority of this buffer.  Lower numbers are higher
  26   //            priority.
  27   // |stream| is a pointer to the stream which is managing this buffer.
  28   SpdyIOBuffer(IOBuffer* buffer, int size, int priority, SpdyStream* stream);
  29   SpdyIOBuffer();
  30   ~SpdyIOBuffer();
  31
  32   // Accessors.
  33   DrainableIOBuffer* buffer() const { return buffer_; }
  34   size_t size() const { return buffer_->size(); }
  35   void release();
  36   int priority() const { return priority_; }
  37   const scoped_refptr<SpdyStream>& stream() const { return stream_; }
  38
  39   // Comparison operator to support sorting.
  40   bool operator<(const SpdyIOBuffer& other) const {
  41     if (priority_ != other.priority_)
  42       return priority_ > other.priority_;
  43     return position_ > other.position_;
  44   }
  45
  46  private:
  47   scoped_refptr<DrainableIOBuffer> buffer_;
  48   int priority_;
  49   uint64 position_;
  50   scoped_refptr<SpdyStream> stream_;
  51   static uint64 order_;  // Maintains a FIFO order for equal priorities.
  52 };
  53
  54 }  // namespace net
  55
  56 #endif  // NET_SPDY_SPDY_IO_BUFFER_H_