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

chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc

Index: chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc
diff --git a/chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc b/chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc
index 0bb0c0f49952993cc9b84fd96b1d81530c40d250..8ef1560841b1f00779ed8b10beb25e8b913baebd 100644
--- a/chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc
+++ b/chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc
@@ -358,6 +358,16 @@ void ChromeDataUseAscriber::ReadyToCommitMainFrameNavigation(
   }
 }
 
+void ChromeDataUseAscriber::DidFinishNavigation(int render_process_id,
+                                                int render_frame_id,
+                                                uint32_t page_transition) {
+  auto frame_it = main_render_frame_data_use_map_.find(
+      RenderFrameHostID(render_process_id, render_frame_id));
+  if (frame_it != main_render_frame_data_use_map_.end()) {
+    frame_it->second->set_page_transition(page_transition);
+  }
+}
+
 void ChromeDataUseAscriber::OnDataUseCompleted(DataUseRecorderEntry entry) {
   // TODO(ryansturm): Notify observers that data use is complete.
 }