| Index: blimp/net/stream_packet_reader.cc
|
| diff --git a/blimp/net/stream_packet_reader.cc b/blimp/net/stream_packet_reader.cc
|
| index 8e675b32c8b3bfe511d07c84bb7fdfd9d8e85c52..6cd91642f8596cf85e3da1d0a6a94acc42c3cd30 100644
|
| --- a/blimp/net/stream_packet_reader.cc
|
| +++ b/blimp/net/stream_packet_reader.cc
|
| @@ -11,6 +11,7 @@
|
| #include "base/memory/weak_ptr.h"
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/sys_byteorder.h"
|
| +#include "blimp/net/blimp_connection_statistics.h"
|
| #include "blimp/net/common.h"
|
| #include "net/base/io_buffer.h"
|
| #include "net/base/net_errors.h"
|
| @@ -34,8 +35,12 @@ std::ostream& operator<<(std::ostream& out,
|
| return out;
|
| }
|
|
|
| -StreamPacketReader::StreamPacketReader(net::StreamSocket* socket)
|
| - : read_state_(ReadState::IDLE), socket_(socket), weak_factory_(this) {
|
| +StreamPacketReader::StreamPacketReader(net::StreamSocket* socket,
|
| + BlimpConnectionStatistics* statistics)
|
| + : read_state_(ReadState::IDLE),
|
| + socket_(socket),
|
| + statistics_(statistics),
|
| + weak_factory_(this) {
|
| DCHECK(socket_);
|
| header_buffer_ = new net::GrowableIOBuffer;
|
| header_buffer_->SetCapacity(kPacketHeaderSizeBytes);
|
| @@ -138,6 +143,7 @@ int StreamPacketReader::DoReadPayload(int result) {
|
| base::Bind(&StreamPacketReader::OnReadComplete,
|
| weak_factory_.GetWeakPtr()));
|
| }
|
| + statistics_->Add(BlimpConnectionStatistics::BYTES_RECEIVED, payload_size_);
|
|
|
| // Finished reading the payload.
|
| read_state_ = ReadState::IDLE;
|
|
|