Index: blimp/net/stream_packet_writer.h |
diff --git a/blimp/net/stream_packet_writer.h b/blimp/net/stream_packet_writer.h |
index a13711b84009ea85e57c9207a927e0a474acd842..98a7143f15067ff548eb3233b4de1c77e836121a 100644 |
--- a/blimp/net/stream_packet_writer.h |
+++ b/blimp/net/stream_packet_writer.h |
@@ -21,6 +21,7 @@ class StreamSocket; |
} // namespace net |
namespace blimp { |
+class BlimpConnectionStatistics; |
// Writes opaque length-prefixed packets to a StreamSocket. |
// The header segment is 32-bit, encoded in network byte order. |
@@ -30,7 +31,10 @@ class BLIMP_NET_EXPORT StreamPacketWriter : public PacketWriter { |
public: |
// |socket|: The socket to write packets to. The caller must ensure |socket| |
// is valid while the reader is in-use (see ReadPacket below). |
- explicit StreamPacketWriter(net::StreamSocket* socket); |
+ // |statistics|: Statistics collector which keeps track of number of bytes |
+ // written. |statistics| is expected to outlive |this|. |
+ StreamPacketWriter(net::StreamSocket* socket, |
+ BlimpConnectionStatistics* statistics); |
~StreamPacketWriter() override; |
@@ -67,6 +71,7 @@ class BLIMP_NET_EXPORT StreamPacketWriter : public PacketWriter { |
scoped_refptr<net::DrainableIOBuffer> payload_buffer_; |
scoped_refptr<net::DrainableIOBuffer> header_buffer_; |
net::CompletionCallback callback_; |
+ BlimpConnectionStatistics* statistics_; |
base::WeakPtrFactory<StreamPacketWriter> weak_factory_; |