commit 75fe34fbf030a27e7be3b1b1365ba395a894e0e0
author rajendrant <rajendrant@chromium.org> Tue Mar 28 05:53:00 2017
committer Commit bot <commit-bot@chromium.org> Tue Mar 28 05:53:00 2017
tree 29d3d5bb148f5da06f0e4b018427e0f67d43944d
parent 379b1ec8fc03311f140ac8ca7f54314416ddaadb

Track data use of WebSockets

BUG=684756

Review-Url: https://codereview.chromium.org/2762563003
Cr-Commit-Position: refs/heads/master@{#460021}

net/websockets/websocket_basic_stream.cc

diff --git a/net/websockets/websocket_basic_stream.cc b/net/websockets/websocket_basic_stream.cc
index e13e1eb..3e8c761 100644
--- a/net/websockets/websocket_basic_stream.cc
+++ b/net/websockets/websocket_basic_stream.cc
@@ -14,6 +14,7 @@
 
 #include "base/bind.h"
 #include "base/logging.h"
+#include "base/metrics/histogram_macros.h"
 #include "base/numerics/safe_conversions.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
@@ -211,6 +212,7 @@
                    buffer,
                    callback));
     if (result > 0) {
+      UMA_HISTOGRAM_COUNTS_100000("Net.WebSocket.DataUse.Upstream", result);
       buffer->DidConsume(result);
     } else {
       return result;
@@ -230,6 +232,8 @@
   }
 
   DCHECK_NE(0, result);
+  UMA_HISTOGRAM_COUNTS_100000("Net.WebSocket.DataUse.Upstream", result);
+
   buffer->DidConsume(result);
   result = WriteEverything(buffer, callback);
   if (result != ERR_IO_PENDING)
@@ -245,6 +249,9 @@
     return result;
   if (result == 0)
     return ERR_CONNECTION_CLOSED;
+
+  UMA_HISTOGRAM_COUNTS_100000("Net.WebSocket.DataUse.Downstream", result);
+
   std::vector<std::unique_ptr<WebSocketFrameChunk>> frame_chunks;
   if (!parser_.Decode(read_buffer_->data(), result, &frame_chunks))
     return WebSocketErrorToNetError(parser_.websocket_error());