Added a debug info UI for Blimp

blimp/net/stream_packet_writer.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "blimp/net/stream_packet_writer.h"
 
 #include <iostream>
 
 #include "base/callback_helpers.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #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"
 #include "net/socket/stream_socket.h"
 
 namespace blimp {
 
 std::ostream& operator<<(std::ostream& out,
                          const StreamPacketWriter::WriteState state) {
   switch (state) {
     case StreamPacketWriter::WriteState::IDLE:
       out << "IDLE";
       break;
     case StreamPacketWriter::WriteState::HEADER:
       out << "HEADER";
       break;
     case StreamPacketWriter::WriteState::PAYLOAD:
       out << "PAYLOAD";
       break;
   }
   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_)

 }
 
 StreamPacketWriter::~StreamPacketWriter() {}
 
 void StreamPacketWriter::WritePacket(
     const scoped_refptr<net::DrainableIOBuffer>& data,
     const net::CompletionCallback& callback) {
   DCHECK_EQ(WriteState::IDLE, write_state_);
   DCHECK(data);
   CHECK(data->BytesRemaining());
 
   write_state_ = WriteState::HEADER;
   header_buffer_->SetOffset(0);
   *reinterpret_cast<uint32_t*>(header_buffer_->data()) =
       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
     // synchronously.
     payload_buffer_ = nullptr;
 
     // Adapt synchronous completion to an asynchronous style.
     base::MessageLoop::current()->PostTask(FROM_HERE,
                                            base::Bind(callback, result));
   } else {
@@ skipping 69 matching lines @@
 
   // If the write finished, either successfully or by error, inform the
   // caller.
   if (result != net::ERR_IO_PENDING) {
     payload_buffer_ = nullptr;
     base::ResetAndReturn(&callback_).Run(result);
   }
 }
 
 }  // namespace blimp