Index: blimp/net/stream_packet_reader.h |
diff --git a/blimp/net/stream_packet_reader.h b/blimp/net/stream_packet_reader.h |
index dd67351337e15a02b68955148d86bb31091c2bd6..2516ca078ed06cf06e1c9a9639b081b7546eb303 100644 |
--- a/blimp/net/stream_packet_reader.h |
+++ b/blimp/net/stream_packet_reader.h |
@@ -21,6 +21,7 @@ class StreamSocket; |
} // namespace net |
namespace blimp { |
+class BlimpConnectionStatistics; |
// Reads opaque length-prefixed packets of bytes from a StreamSocket. |
// The header segment is 32-bit, encoded in network byte order. |
@@ -30,7 +31,10 @@ class BLIMP_NET_EXPORT StreamPacketReader : public PacketReader { |
public: |
// |socket|: The socket to read packets from. The caller must ensure |socket| |
// is valid while the reader is in-use (see ReadPacket below). |
- explicit StreamPacketReader(net::StreamSocket* socket); |
+ // |statistics|: Statistics collector to keep track of number of bytes read. |
+ // |statistics| is expected to outlive |this|. |
+ StreamPacketReader(net::StreamSocket* socket, |
+ BlimpConnectionStatistics* statistics); |
~StreamPacketReader() override; |
@@ -72,6 +76,7 @@ class BLIMP_NET_EXPORT StreamPacketReader : public PacketReader { |
scoped_refptr<net::GrowableIOBuffer> header_buffer_; |
scoped_refptr<net::GrowableIOBuffer> payload_buffer_; |
net::CompletionCallback callback_; |
+ BlimpConnectionStatistics* statistics_; |
base::WeakPtrFactory<StreamPacketReader> weak_factory_; |