Write thumbnail to the database on SetPageThumbnail....

chrome/browser/history/top_sites_database.cc

 // Copyright (c) 2009 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 "app/sql/transaction.h"
 #include "base/string_util.h"
 #include "chrome/browser/diagnostics/sqlite_diagnostics.h"
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/history/top_sites_database.h"
 
@@ skipping 16 matching lines @@
   return InitThumbnailTable();
 }
 
 bool TopSitesDatabaseImpl::InitThumbnailTable() {
   if (!db_.DoesTableExist("thumbnails")) {
     if (!db_.Execute("CREATE TABLE thumbnails ("
                      "url LONGVARCHAR PRIMARY KEY,"
                      "url_rank INTEGER ,"
                      "title LONGVARCHAR,"
                      "thumbnail BLOB,"
-                     "redirects LONGVARCHAR)")) {
+                     "redirects LONGVARCHAR,"
+                     "boring_score DOUBLE DEFAULT 1.0, "
+                     "good_clipping INTEGER DEFAULT 0, "
+                     "at_top INTEGER DEFAULT 0, "
+                     "last_updated INTEGER DEFAULT 0) ")) {
       LOG(WARNING) << db_.GetErrorMessage();
       return false;
     }
   }
   return true;
 }
 
 MostVisitedURLList TopSitesDatabaseImpl::GetTopURLs() {
   MostVisitedURLList result;
   sql::Statement select_statement(db_.GetCachedStatement(
@@ skipping 46 matching lines @@
                                             int new_rank,
                                             const TopSites::Images& thumbnail) {
   sql::Transaction transaction(&db_);
   transaction.Begin();
 
   int prev_rank = GetURLRank(url);
 
   sql::Statement statement(db_.GetCachedStatement(
       SQL_FROM_HERE,
       "INSERT OR REPLACE INTO thumbnails "
-      "(url, url_rank, title, thumbnail, redirects) "
-      "VALUES (?, ?, ?, ?, ?)"));  // TODO(Nik): add the rest of the schema.
+      "(url, url_rank, title, thumbnail, redirects, "
+      "boring_score, good_clipping, at_top, last_updated) "
+      "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)"));
   if (!statement)
     return;
 
   statement.BindString(0, url.url.spec());
   statement.BindInt(1, -1);  // Temporary value
   statement.BindString16(2, url.title);
   if (thumbnail.thumbnail.get()) {
     statement.BindBlob(3, &thumbnail.thumbnail->data.front(),
                        static_cast<int>(thumbnail.thumbnail->data.size()));
   }
   statement.BindString(4, GetRedirects(url));
+  const ThumbnailScore& score = thumbnail.thumbnail_score;
+  statement.BindDouble(5, score.boring_score);
+  statement.BindBool(6, score.good_clipping);
+  statement.BindBool(7, score.at_top);
+  statement.BindInt64(8, score.time_at_snapshot.ToInternalValue());
 
   if (!statement.Run())
     NOTREACHED() << db_.GetErrorMessage();
 
   // Shift the ranks.
   if (prev_rank == -1) {
     // Shift up
     sql::Statement shift_statement(db_.GetCachedStatement(
         SQL_FROM_HERE,
         "UPDATE thumbnails "
@@ skipping 32 matching lines @@
       "SET url_rank = ? "
       "WHERE url_rank == -1"));
   set_statement.BindInt(0, new_rank);
   if (set_statement)
     set_statement.Run();
   transaction.Commit();
 }
 
 bool TopSitesDatabaseImpl::GetPageThumbnail(const MostVisitedURL& url,
                                             TopSites::Images* thumbnail) {
-  sql::Statement select_statement(db_.GetCachedStatement(
+  sql::Statement statement(db_.GetCachedStatement(
       SQL_FROM_HERE,
-      "SELECT thumbnail "
+      "SELECT thumbnail, boring_score, good_clipping, at_top, last_updated "
       "FROM thumbnails WHERE url=?"));
 
-  if (!select_statement) {
+  if (!statement) {
     LOG(WARNING) << db_.GetErrorMessage();
     return false;
   }
 
-  select_statement.BindString(0, url.url.spec());
-  if (!select_statement.Step())
+  statement.BindString(0, url.url.spec());
+  if (!statement.Step())
     return false;
 
   std::vector<unsigned char> data;
-  select_statement.ColumnBlobAsVector(0, &data);
+  statement.ColumnBlobAsVector(0, &data);
   thumbnail->thumbnail = RefCountedBytes::TakeVector(&data);
+  thumbnail->thumbnail_score.boring_score = statement.ColumnDouble(1);
+  thumbnail->thumbnail_score.good_clipping = statement.ColumnBool(2);
+  thumbnail->thumbnail_score.at_top = statement.ColumnBool(3);
+  thumbnail->thumbnail_score.time_at_snapshot =
+      base::Time::FromInternalValue(statement.ColumnInt64(4));
   return true;
 }
 
 int TopSitesDatabaseImpl::GetURLRank(const MostVisitedURL& url) {
   int result = -1;
   sql::Statement select_statement(db_.GetCachedStatement(
       SQL_FROM_HERE,
       "SELECT url_rank "
       "FROM thumbnails WHERE url=?"));
   if (!select_statement) {
@@ skipping 29 matching lines @@
                              "DELETE FROM thumbnails WHERE url = ?"));
   if (!delete_statement)
     return false;
   delete_statement.BindString(0, url.url.spec());
   delete_statement.Run();
 
   return transaction.Commit();
 }
 
 }  // namespace history
-