| OLD | NEW |
| (Empty) |
| 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 NET_QUIC_QUIC_WRITE_BLOCKED_LIST_H_ | |
| 6 #define NET_QUIC_QUIC_WRITE_BLOCKED_LIST_H_ | |
| 7 | |
| 8 #include <set> | |
| 9 | |
| 10 #include "net/base/net_export.h" | |
| 11 #include "net/quic/quic_protocol.h" | |
| 12 #include "net/spdy/write_blocked_list.h" | |
| 13 | |
| 14 namespace net { | |
| 15 | |
| 16 // Keeps tracks of the QUIC streams that have data to write, sorted by | |
| 17 // priority. QUIC stream priority order is: | |
| 18 // Crypto stream > Headers stream > Data streams by requested priority. | |
| 19 class NET_EXPORT_PRIVATE QuicWriteBlockedList { | |
| 20 private: | |
| 21 typedef WriteBlockedList<QuicStreamId> QuicWriteBlockedListBase; | |
| 22 | |
| 23 public: | |
| 24 static const QuicPriority kHighestPriority; | |
| 25 static const QuicPriority kLowestPriority; | |
| 26 | |
| 27 QuicWriteBlockedList(); | |
| 28 ~QuicWriteBlockedList(); | |
| 29 | |
| 30 bool HasWriteBlockedDataStreams() const { | |
| 31 return base_write_blocked_list_.HasWriteBlockedStreams(); | |
| 32 } | |
| 33 | |
| 34 bool HasWriteBlockedCryptoOrHeadersStream() const { | |
| 35 return crypto_stream_blocked_ || headers_stream_blocked_; | |
| 36 } | |
| 37 | |
| 38 size_t NumBlockedStreams() const { | |
| 39 size_t num_blocked = base_write_blocked_list_.NumBlockedStreams(); | |
| 40 if (crypto_stream_blocked_) { | |
| 41 ++num_blocked; | |
| 42 } | |
| 43 if (headers_stream_blocked_) { | |
| 44 ++num_blocked; | |
| 45 } | |
| 46 | |
| 47 return num_blocked; | |
| 48 } | |
| 49 | |
| 50 QuicStreamId PopFront() { | |
| 51 if (crypto_stream_blocked_) { | |
| 52 crypto_stream_blocked_ = false; | |
| 53 return kCryptoStreamId; | |
| 54 } | |
| 55 | |
| 56 if (headers_stream_blocked_) { | |
| 57 headers_stream_blocked_ = false; | |
| 58 return kHeadersStreamId; | |
| 59 } | |
| 60 | |
| 61 SpdyPriority priority = | |
| 62 base_write_blocked_list_.GetHighestPriorityWriteBlockedList(); | |
| 63 QuicStreamId id = base_write_blocked_list_.PopFront(priority); | |
| 64 blocked_streams_.erase(id); | |
| 65 return id; | |
| 66 } | |
| 67 | |
| 68 void PushBack(QuicStreamId stream_id, QuicPriority priority) { | |
| 69 if (stream_id == kCryptoStreamId) { | |
| 70 DCHECK_EQ(kHighestPriority, priority); | |
| 71 // TODO(avd) Add DCHECK(!crypto_stream_blocked_) | |
| 72 crypto_stream_blocked_ = true; | |
| 73 return; | |
| 74 } | |
| 75 | |
| 76 if (stream_id == kHeadersStreamId) { | |
| 77 DCHECK_EQ(kHighestPriority, priority); | |
| 78 // TODO(avd) Add DCHECK(!headers_stream_blocked_); | |
| 79 headers_stream_blocked_ = true; | |
| 80 return; | |
| 81 } | |
| 82 | |
| 83 if (blocked_streams_.find(stream_id) != blocked_streams_.end()) { | |
| 84 DVLOG(1) << "Stream " << stream_id << " already in write blocked list."; | |
| 85 return; | |
| 86 } | |
| 87 | |
| 88 base_write_blocked_list_.PushBack( | |
| 89 stream_id, static_cast<SpdyPriority>(priority)); | |
| 90 blocked_streams_.insert(stream_id); | |
| 91 return; | |
| 92 } | |
| 93 | |
| 94 bool crypto_stream_blocked() const { return crypto_stream_blocked_; } | |
| 95 bool headers_stream_blocked() const { return headers_stream_blocked_; } | |
| 96 | |
| 97 private: | |
| 98 QuicWriteBlockedListBase base_write_blocked_list_; | |
| 99 bool crypto_stream_blocked_; | |
| 100 bool headers_stream_blocked_; | |
| 101 | |
| 102 // Keep track of write blocked streams in a set for faster membership checking | |
| 103 // than iterating over the base_write_blocked_list_. The contents of this set | |
| 104 // should mirror the contents of base_write_blocked_list_. | |
| 105 std::set<QuicStreamId> blocked_streams_; | |
| 106 | |
| 107 DISALLOW_COPY_AND_ASSIGN(QuicWriteBlockedList); | |
| 108 }; | |
| 109 | |
| 110 } // namespace net | |
| 111 | |
| 112 | |
| 113 #endif // NET_QUIC_QUIC_WRITE_BLOCKED_LIST_H_ | |
| OLD | NEW |