predictors: Store protobufs in the database.

chrome/browser/predictors/resource_prefetch_predictor_tables.h

 // 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.
 
 #ifndef CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TABLES_H_
 #define CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TABLES_H_
 
 #include <stddef.h>
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/time/time.h"
 #include "chrome/browser/predictors/predictor_table_base.h"
 #include "chrome/browser/predictors/resource_prefetch_common.h"
+#include "chrome/browser/predictors/resource_prefetch_predictor.pb.h"
 #include "content/public/common/resource_type.h"
 #include "net/base/request_priority.h"
 #include "url/gurl.h"
 
 namespace sql {
 class Statement;
 }
 
 namespace predictors {
 
+using chrome_browser_predictors::ResourceData;
+
 // Interface for database tables used by the ResourcePrefetchPredictor.
 // All methods except the constructor and destructor need to be called on the DB
 // thread.
 //
 // Currently manages:
 //  - UrlResourceTable - resources per Urls.
 //  - UrlMetadataTable - misc data for Urls (like last visit time).
 //  - HostResourceTable - resources per host.
 //  - HostMetadataTable - misc data for hosts.
 class ResourcePrefetchPredictorTables : public PredictorTableBase {
  public:
   // Used in the UrlResourceTable and HostResourceTable to store resources
   // required for the page or host.
   struct ResourceRow {
     ResourceRow();
     ResourceRow(const ResourceRow& other);
     ResourceRow(const std::string& main_frame_url,
                 const std::string& resource_url,
                 content::ResourceType resource_type,
                 int number_of_hits,
                 int number_of_misses,
                 int consecutive_misses,
                 double average_position,
                 net::RequestPriority priority,
                 bool has_validators,
                 bool always_revalidate);
     void UpdateScore();
     bool operator==(const ResourceRow& rhs) const;
+    static void FromProto(const ResourceData& proto, ResourceRow* row);
+    void ToProto(ResourceData* resource_data) const;
 
     // Stores the host for host based data, main frame Url for the Url based
     // data. This field is cleared for efficiency reasons and the code outside
     // this class should not assume it is set.
     std::string primary_key;
 
     GURL resource_url;
     content::ResourceType resource_type;
     size_t number_of_hits;
     size_t number_of_misses;
     size_t consecutive_misses;
     double average_position;
     net::RequestPriority priority;
     bool has_validators;
     bool always_revalidate;
 
     // Not stored.
     float score;
   };
   typedef std::vector<ResourceRow> ResourceRows;
 
-  // Sorts the ResourceRows by score, descending.
-  struct ResourceRowSorter {
-    bool operator()(const ResourceRow& x, const ResourceRow& y) const;
-  };
+  // Sorts the resource rows by score, decreasing.
+  static void SortResourceRows(ResourceRows* rows);
 
   // Aggregated data for a Url or Host. Although the data differs slightly, we
   // store them in the same structure, because most of the fields are common and
   // it allows us to use the same functions.
   struct PrefetchData {
     PrefetchData(PrefetchKeyType key_type, const std::string& primary_key);
     PrefetchData(const PrefetchData& other);
     ~PrefetchData();
     bool operator==(const PrefetchData& rhs) const;
 
@@ skipping 71 matching lines @@
   sql::Statement* GetHostResourceUpdateStatement();
   sql::Statement* GetHostMetadataDeleteStatement();
   sql::Statement* GetHostMetadataUpdateStatement();
 
   DISALLOW_COPY_AND_ASSIGN(ResourcePrefetchPredictorTables);
 };
 
 }  // namespace predictors
 
 #endif  // CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TABLES_H_