predictors: Refactor resource_prefetch_predictor_tables.

chrome/browser/predictors/resource_prefetch_predictor.cc

 // Copyright 2014 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/predictors/resource_prefetch_predictor.h"
 
 #include <map>
 #include <set>
 #include <utility>
 
@@ skipping 509 matching lines @@
       PopulatePrefetcherRequest(iterator->second, urls);
     }
   }
 
   return !urls->empty();
 }
 
 void ResourcePrefetchPredictor::PopulatePrefetcherRequest(
     const PrefetchData& data,
     std::vector<GURL>* urls) {
-  for (const ResourceData& resource : data.resources) {
+  for (const ResourceData& resource : data.resources()) {
     float confidence =
         static_cast<float>(resource.number_of_hits()) /
         (resource.number_of_hits() + resource.number_of_misses());
     if (confidence < config_.min_resource_confidence_to_trigger_prefetch ||
         resource.number_of_hits() <
             config_.min_resource_hits_to_trigger_prefetch) {
       continue;
     }
 
     urls->push_back(GURL(resource.resource_url()));
@@ skipping 167 matching lines @@
                    tables_, url_redirects_to_delete, host_redirects_to_delete));
   }
 }
 
 void ResourcePrefetchPredictor::RemoveOldestEntryInPrefetchDataMap(
     PrefetchKeyType key_type,
     PrefetchDataMap* data_map) {
   if (data_map->empty())
     return;
 
-  base::Time oldest_time;
+  uint64_t oldest_time = UINT64_MAX;
   std::string key_to_delete;
-  for (PrefetchDataMap::iterator it = data_map->begin();
-       it != data_map->end(); ++it) {
-    if (key_to_delete.empty() || it->second.last_visit < oldest_time) {
-      key_to_delete = it->first;
-      oldest_time = it->second.last_visit;
+  for (const auto& kv : *data_map) {
+    const PrefetchData& data = kv.second;
+    if (key_to_delete.empty() || data.last_visit_time() < oldest_time) {
+      key_to_delete = data.primary_key();
+      oldest_time = data.last_visit_time();
     }
   }
 
   data_map->erase(key_to_delete);
   BrowserThread::PostTask(
       BrowserThread::DB, FROM_HERE,
       base::Bind(
           &ResourcePrefetchPredictorTables::DeleteSingleResourceDataPoint,
           tables_, key_to_delete, key_type));
 }
@@ skipping 71 matching lines @@
   // If the primary key is too long reject it.
   if (key.length() > ResourcePrefetchPredictorTables::kMaxStringLength)
     return;
 
   PrefetchDataMap::iterator cache_entry = data_map->find(key);
   if (cache_entry == data_map->end()) {
     // If the table is full, delete an entry.
     if (data_map->size() >= max_data_map_size)
       RemoveOldestEntryInPrefetchDataMap(key_type, data_map);
 
-    cache_entry = data_map->insert(std::make_pair(
-        key, PrefetchData(key_type, key))).first;
-    cache_entry->second.last_visit = base::Time::Now();
+    cache_entry = data_map->insert(std::make_pair(key, PrefetchData())).first;
+    cache_entry->second.set_primary_key(key);
+    cache_entry->second.set_last_visit_time(
+        base::Time::Now().ToInternalValue());
     size_t new_resources_size = new_resources.size();
     std::set<GURL> resources_seen;
     for (size_t i = 0; i < new_resources_size; ++i) {
-      if (resources_seen.find(new_resources[i].resource_url) !=
-          resources_seen.end()) {
+      const URLRequestSummary& summary = new_resources[i];
+      if (resources_seen.find(summary.resource_url) != resources_seen.end())
         continue;
-      }
-      ResourceData resource_to_add;
-      resource_to_add.set_resource_url(new_resources[i].resource_url.spec());
-      resource_to_add.set_resource_type(static_cast<ResourceData::ResourceType>(
-          new_resources[i].resource_type));
-      resource_to_add.set_number_of_hits(1);
-      resource_to_add.set_average_position(i + 1);
-      resource_to_add.set_priority(
-          static_cast<ResourceData::Priority>(new_resources[i].priority));
-      resource_to_add.set_has_validators(new_resources[i].has_validators);
-      resource_to_add.set_always_revalidate(new_resources[i].always_revalidate);
-      cache_entry->second.resources.push_back(resource_to_add);
-      resources_seen.insert(new_resources[i].resource_url);
+
+      ResourceData* resource_to_add = cache_entry->second.add_resources();
+      resource_to_add->set_resource_url(summary.resource_url.spec());
+      resource_to_add->set_resource_type(
+          static_cast<ResourceData::ResourceType>(summary.resource_type));
+      resource_to_add->set_number_of_hits(1);
+      resource_to_add->set_average_position(i + 1);
+      resource_to_add->set_priority(
+          static_cast<ResourceData::Priority>(summary.priority));
+      resource_to_add->set_has_validators(summary.has_validators);
+      resource_to_add->set_always_revalidate(summary.always_revalidate);
+
+      resources_seen.insert(summary.resource_url);
     }
   } else {
-    std::vector<ResourceData>& old_resources = cache_entry->second.resources;
-    cache_entry->second.last_visit = base::Time::Now();
+    cache_entry->second.set_last_visit_time(
+        base::Time::Now().ToInternalValue());
 
     // Build indices over the data.
     std::map<GURL, int> new_index, old_index;
     int new_resources_size = static_cast<int>(new_resources.size());
     for (int i = 0; i < new_resources_size; ++i) {
       const URLRequestSummary& summary = new_resources[i];
       // Take the first occurence of every url.
       if (new_index.find(summary.resource_url) == new_index.end())
         new_index[summary.resource_url] = i;
     }
-    int old_resources_size = static_cast<int>(old_resources.size());
+    int old_resources_size =
+        static_cast<int>(cache_entry->second.resources_size());
     for (int i = 0; i < old_resources_size; ++i) {
-      const ResourceData& resource = old_resources[i];
+      const ResourceData& resource = cache_entry->second.resources(i);
       GURL resource_url(resource.resource_url());
       DCHECK(old_index.find(resource_url) == old_index.end());
       old_index[resource_url] = i;
     }
 
     // Go through the old urls and update their hit/miss counts.
     for (int i = 0; i < old_resources_size; ++i) {
-      ResourceData& old_resource = old_resources[i];
-      GURL resource_url(old_resource.resource_url());
+      ResourceData* old_resource = cache_entry->second.mutable_resources(i);
+      GURL resource_url(old_resource->resource_url());
       if (new_index.find(resource_url) == new_index.end()) {
-        old_resource.set_number_of_misses(old_resource.number_of_misses() + 1);
-        old_resource.set_consecutive_misses(old_resource.consecutive_misses() +
-                                            1);
+        old_resource->set_number_of_misses(old_resource->number_of_misses() +
+                                           1);
+        old_resource->set_consecutive_misses(
+            old_resource->consecutive_misses() + 1);
       } else {
         const URLRequestSummary& new_summary =
             new_resources[new_index[resource_url]];
 
         // Update the resource type since it could have changed.
-        if (new_summary.resource_type != content::RESOURCE_TYPE_LAST_TYPE)
-          old_resource.set_resource_type(
+        if (new_summary.resource_type != content::RESOURCE_TYPE_LAST_TYPE) {
+          old_resource->set_resource_type(
               static_cast<ResourceData::ResourceType>(
                   new_summary.resource_type));
+        }
 
-        old_resource.set_priority(
+        old_resource->set_priority(
             static_cast<ResourceData::Priority>(new_summary.priority));
 
         int position = new_index[resource_url] + 1;
         int total =
-            old_resource.number_of_hits() + old_resource.number_of_misses();
-        old_resource.set_average_position(
-            ((old_resource.average_position() * total) + position) /
+            old_resource->number_of_hits() + old_resource->number_of_misses();
+        old_resource->set_average_position(
+            ((old_resource->average_position() * total) + position) /
             (total + 1));
-        old_resource.set_number_of_hits(old_resource.number_of_hits() + 1);
-        old_resource.set_consecutive_misses(0);
+        old_resource->set_number_of_hits(old_resource->number_of_hits() + 1);
+        old_resource->set_consecutive_misses(0);
       }
     }
 
     // Add the new ones that we have not seen before.
     for (int i = 0; i < new_resources_size; ++i) {
       const URLRequestSummary& summary = new_resources[i];
       if (old_index.find(summary.resource_url) != old_index.end())
         continue;
 
       // Only need to add new stuff.
-      ResourceData resource_to_add;
-      resource_to_add.set_resource_url(summary.resource_url.spec());
-      resource_to_add.set_resource_type(
+      ResourceData* resource_to_add = cache_entry->second.add_resources();
+      resource_to_add->set_resource_url(summary.resource_url.spec());
+      resource_to_add->set_resource_type(
           static_cast<ResourceData::ResourceType>(summary.resource_type));
-      resource_to_add.set_number_of_hits(1);
-      resource_to_add.set_average_position(i + 1);
-      resource_to_add.set_priority(
+      resource_to_add->set_number_of_hits(1);
+      resource_to_add->set_average_position(i + 1);
+      resource_to_add->set_priority(
           static_cast<ResourceData::Priority>(summary.priority));
-      resource_to_add.set_has_validators(new_resources[i].has_validators);
-      resource_to_add.set_always_revalidate(new_resources[i].always_revalidate);
-      old_resources.push_back(resource_to_add);
+      resource_to_add->set_has_validators(new_resources[i].has_validators);
+      resource_to_add->set_always_revalidate(
+          new_resources[i].always_revalidate);
 
       // To ensure we dont add the same url twice.
       old_index[summary.resource_url] = 0;
     }
   }
 
   // Trim and sort the resources after the update.
-  std::vector<ResourceData>& resources = cache_entry->second.resources;
-  for (auto it = resources.begin(); it != resources.end();) {
-    if (it->consecutive_misses() >= config_.max_consecutive_misses)
-      it = resources.erase(it);
-    else
-      ++it;
+  std::vector<ResourceData> resources;
+  resources.reserve(cache_entry->second.resources_size());
+  for (const ResourceData& resource : cache_entry->second.resources()) {
+    if (resource.consecutive_misses() < config_.max_consecutive_misses)
+      resources.push_back(resource);
   }
   ResourcePrefetchPredictorTables::SortResources(&resources);
   if (resources.size() > config_.max_resources_per_entry)
     resources.resize(config_.max_resources_per_entry);
 
+  cache_entry->second.clear_resources();
+  for (const ResourceData& resource : resources)
+    cache_entry->second.add_resources()->CopyFrom(resource);
+
   // If the row has no resources, remove it from the cache and delete the
   // entry in the database. Else update the database.
   if (resources.empty()) {
     data_map->erase(key);
     BrowserThread::PostTask(
         BrowserThread::DB, FROM_HERE,
         base::Bind(
             &ResourcePrefetchPredictorTables::DeleteSingleResourceDataPoint,
             tables_, key, key_type));
   } else {
+    PrefetchData empty_data;
+    RedirectData empty_redirect_data;
     bool is_host = key_type == PREFETCH_KEY_TYPE_HOST;
-    PrefetchData empty_data(
-        !is_host ? PREFETCH_KEY_TYPE_HOST : PREFETCH_KEY_TYPE_URL,
-        std::string());
-    RedirectData empty_redirect_data;
     const PrefetchData& host_data = is_host ? cache_entry->second : empty_data;
     const PrefetchData& url_data = is_host ? empty_data : cache_entry->second;
     BrowserThread::PostTask(
         BrowserThread::DB, FROM_HERE,
         base::Bind(&ResourcePrefetchPredictorTables::UpdateData, tables_,
                    url_data, host_data, empty_redirect_data,
                    empty_redirect_data));
   }
 
   if (key != key_before_redirects) {
@@ skipping 63 matching lines @@
     cache_entry->second.add_redirect_endpoints()->CopyFrom(redirect);
 
   if (redirects.empty()) {
     redirect_map->erase(cache_entry);
     BrowserThread::PostTask(
         BrowserThread::DB, FROM_HERE,
         base::Bind(
             &ResourcePrefetchPredictorTables::DeleteSingleRedirectDataPoint,
             tables_, key, key_type));
   } else {
+    RedirectData empty_redirect_data;
+    PrefetchData empty_data;
     bool is_host = key_type == PREFETCH_KEY_TYPE_HOST;
-    RedirectData empty_redirect_data;
-    PrefetchData empty_url_data(PREFETCH_KEY_TYPE_URL, std::string());
-    PrefetchData empty_host_data(PREFETCH_KEY_TYPE_HOST, std::string());
     const RedirectData& host_redirect_data =
         is_host ? cache_entry->second : empty_redirect_data;
     const RedirectData& url_redirect_data =
         is_host ? empty_redirect_data : cache_entry->second;
     BrowserThread::PostTask(
         BrowserThread::DB, FROM_HERE,
         base::Bind(&ResourcePrefetchPredictorTables::UpdateData, tables_,
-                   empty_url_data, empty_host_data, url_redirect_data,
+                   empty_data, empty_data, url_redirect_data,
                    host_redirect_data));
   }
 }
 
 void ResourcePrefetchPredictor::OnURLsDeleted(
     history::HistoryService* history_service,
     bool all_history,
     bool expired,
     const history::URLRows& deleted_rows,
     const std::set<GURL>& favicon_urls) {
@@ skipping 31 matching lines @@
     // HistoryService is already loaded. Continue with Initialization.
     OnHistoryAndCacheLoaded();
     return;
   }
   DCHECK(!history_service_observer_.IsObserving(history_service));
   history_service_observer_.Add(history_service);
   return;
 }
 
 }  // namespace predictors