Record the tab state during data use

chrome/browser/data_use_measurement/chrome_data_use_ascriber.cc

 // Copyright 2016 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 "chrome/browser/data_use_measurement/chrome_data_use_ascriber.h"
 
 #include <string>
 
 #include "base/memory/ptr_util.h"
 #include "chrome/browser/data_use_measurement/chrome_data_use_recorder.h"
@@ skipping 26 matching lines @@
 ChromeDataUseAscriber::ChromeDataUseAscriber() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 }
 
 ChromeDataUseAscriber::~ChromeDataUseAscriber() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
   DCHECK(subframe_to_mainframe_map_.empty());
   DCHECK(data_use_recorders_.empty());
 }
 
-ChromeDataUseRecorder* ChromeDataUseAscriber::GetDataUseRecorder(
-    net::URLRequest* request,
-    bool can_create_new) {
-  DataUseRecorderEntry entry = GetDataUseRecorderEntry(request, can_create_new);
+ChromeDataUseRecorder* ChromeDataUseAscriber::GetOrCreateDataUseRecorder(
+    net::URLRequest* request) {
+  DataUseRecorderEntry entry = GetOrCreateDataUseRecorderEntry(request);
   return entry == data_use_recorders_.end() ? nullptr : &(*entry);
 }
 
-ChromeDataUseAscriber::DataUseRecorderEntry
-ChromeDataUseAscriber::GetDataUseRecorderEntry(net::URLRequest* request,
-                                               bool can_create_new) {
+ChromeDataUseRecorder* ChromeDataUseAscriber::GetDataUseRecorder(
+    const net::URLRequest& request) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 
   // TODO(ryansturm): Handle PlzNavigate (http://crbug/664233).
+  if (content::IsBrowserSideNavigationEnabled())
+    return nullptr;
+
+  // If a DataUseRecorder has already been set as user data, then return that.
+  auto user_data = static_cast<DataUseRecorderEntryAsUserData*>(
+      request.GetUserData(DataUseRecorderEntryAsUserData::kUserDataKey));
+  return user_data ? &(*user_data->recorder_entry()) : nullptr;
+}
+
+ChromeDataUseAscriber::DataUseRecorderEntry
+ChromeDataUseAscriber::GetOrCreateDataUseRecorderEntry(
+    net::URLRequest* request) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
+
+  // TODO(ryansturm): Handle PlzNavigate (http://crbug/664233).
   if (content::IsBrowserSideNavigationEnabled())
     return data_use_recorders_.end();
 
   // If a DataUseRecorder has already been set as user data, then return that.
   auto user_data = static_cast<DataUseRecorderEntryAsUserData*>(
       request->GetUserData(DataUseRecorderEntryAsUserData::kUserDataKey));
   if (user_data)
     return user_data->recorder_entry();
 
-  if (!can_create_new)
-    return data_use_recorders_.end();
-
   // If request is associated with a ChromeService, create a new
   // DataUseRecorder for it. There is no reason to aggregate URLRequests
   // from ChromeServices into the same DataUseRecorder instance.
   DataUseUserData* service = static_cast<DataUseUserData*>(
       request->GetUserData(DataUseUserData::kUserDataKey));
   if (service) {
     DataUseRecorderEntry entry = CreateNewDataUseRecorder(request);
 
     entry->data_use().set_description(
         DataUseUserData::GetServiceNameAsString(service->service_name()));
@@ skipping 54 matching lines @@
   // Create a new DataUseRecorder for all other requests.
   DataUseRecorderEntry entry = CreateNewDataUseRecorder(request);
   DataUse& data_use = entry->data_use();
   data_use.set_url(request->url());
   return entry;
 }
 
 void ChromeDataUseAscriber::OnUrlRequestDestroyed(net::URLRequest* request) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 
-  DataUseRecorderEntry entry = GetDataUseRecorderEntry(request, true);
+  DataUseRecorderEntry entry = GetOrCreateDataUseRecorderEntry(request);
 
   if (entry == data_use_recorders_.end())
     return;
 
   DataUseRecorder* recorder = &(*entry);
 
   RenderFrameHostID frame_key = entry->main_frame_id();
   auto frame_iter = main_render_frame_data_use_map_.find(frame_key);
   bool is_in_render_frame_map =
       frame_iter != main_render_frame_data_use_map_.end() &&
@@ skipping 235 matching lines @@
                                                    int new_render_process_id,
                                                    int new_render_frame_id) {
   if (visible_main_render_frames_.find(
           RenderFrameHostID(old_render_process_id, old_render_frame_id)) !=
       visible_main_render_frames_.end()) {
     WasShownOrHidden(new_render_process_id, new_render_frame_id, true);
   }
 }
 
 }  // namespace data_use_measurement