Record data use of user traffic by different core page transition types

components/data_use_measurement/core/data_use_measurement.cc

Index: components/data_use_measurement/core/data_use_measurement.cc
diff --git a/components/data_use_measurement/core/data_use_measurement.cc b/components/data_use_measurement/core/data_use_measurement.cc
index 2c7b0b3f0e66b83d3de1b752519916a8a4d74f45..0e2794cf1902ff550dfec6b2a861e4027719be8f 100644
--- a/components/data_use_measurement/core/data_use_measurement.cc
+++ b/components/data_use_measurement/core/data_use_measurement.cc
@@ -169,6 +169,7 @@ void DataUseMeasurement::OnCompleted(const net::URLRequest& request,
   // of redirected requests.
   UpdateDataUsePrefs(request);
   ReportServicesMessageSizeUMA(request);
+  RecordPageTransitionUMA(request);
 #if defined(OS_ANDROID)
   MaybeRecordNetworkBytesOS();
 #endif
@@ -392,7 +393,7 @@ void DataUseMeasurement::RecordTabStateHistogram(
     TrafficDirection dir,
     DataUseUserData::AppState app_state,
     bool is_tab_visible,
-    int64_t bytes) {
+    int64_t bytes) const {
   if (app_state == DataUseUserData::UNKNOWN)
     return;
 
@@ -444,4 +445,16 @@ void DataUseMeasurement::RecordContentTypeHistogram(
   }
 }
 
+void DataUseMeasurement::RecordPageTransitionUMA(
+    const net::URLRequest& request) const {
+  if (!url_request_classifier_->IsUserRequest(request))
+    return;
+
+  const DataUseRecorder* recorder = ascriber_->GetDataUseRecorder(request);
+  if (recorder) {
+    url_request_classifier_->RecordPageTransitionUMA(
+        recorder->page_transition(), request.GetTotalReceivedBytes());
+  }
+}
+
 }  // namespace data_use_measurement