Chromium Code Reviews| Index: blimp/net/stream_packet_writer.cc |
| diff --git a/blimp/net/stream_packet_writer.cc b/blimp/net/stream_packet_writer.cc |
| index 7fe00d2a1f7333b94a4cba5cac0e8427c91199bb..213dbe8cbb667da7ee89ebd2602ff7327db57002 100644 |
| --- a/blimp/net/stream_packet_writer.cc |
| +++ b/blimp/net/stream_packet_writer.cc |
| @@ -12,6 +12,7 @@ |
| #include "base/message_loop/message_loop.h" |
| #include "base/sys_byteorder.h" |
| #include "blimp/common/proto/blimp_message.pb.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" |
| @@ -35,12 +36,14 @@ std::ostream& operator<<(std::ostream& out, |
| return out; |
| } |
| -StreamPacketWriter::StreamPacketWriter(net::StreamSocket* socket) |
| +StreamPacketWriter::StreamPacketWriter(net::StreamSocket* socket, |
| + BlimpConnectionStatistics* statistics) |
| : write_state_(WriteState::IDLE), |
| socket_(socket), |
| header_buffer_( |
| new net::DrainableIOBuffer(new net::IOBuffer(kPacketHeaderSizeBytes), |
| kPacketHeaderSizeBytes)), |
| + statistics_(statistics), |
| weak_factory_(this) { |
| DCHECK(socket_); |
|
Kevin M
2016/05/24 22:56:40
DCHECK(statistics_)
|
| } |
| @@ -60,6 +63,8 @@ void StreamPacketWriter::WritePacket( |
| base::HostToNet32(data->BytesRemaining()); |
| payload_buffer_ = data; |
| + statistics_->Add(BlimpConnectionStatistics::BYTES_SENT, |
| + payload_buffer_->BytesRemaining()); |
| int result = DoWriteLoop(net::OK); |
| if (result != net::ERR_IO_PENDING) { |
| // Release the payload buffer, since the write operation has completed |