Write thumbnail to the database on SetPageThumbnail....

chrome/browser/history/top_sites.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 "chrome/browser/history/top_sites.h"
 
 #include "base/logging.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/history/top_sites_database.h"
 #include "chrome/browser/history/page_usage_data.h"
@@ skipping 29 matching lines @@
   {
     AutoLock lock(lock_);
     MostVisitedURLList top_urls = db_->GetTopURLs();
     StoreMostVisited(&top_urls);
   }  // Lock is released here.
 
   for (size_t i = 0; i < top_sites_.size(); i++) {
     MostVisitedURL url = top_sites_[i];
     Images thumbnail;
     if (db_->GetPageThumbnail(url, &thumbnail)) {
-      SetPageThumbnail(url.url, thumbnail.thumbnail, thumbnail.thumbnail_score);
+      SetPageThumbnailNoDB(url.url, thumbnail.thumbnail,
+                           thumbnail.thumbnail_score);
     }
   }
 }
 
-// Public wrapper that encodes the bitmap into RefCountedBytes.
+// Public function that encodes the bitmap into RefCountedBytes and
+// updates the database.
 bool TopSites::SetPageThumbnail(const GURL& url,
                                 const SkBitmap& thumbnail,
                                 const ThumbnailScore& score) {
   scoped_refptr<RefCountedBytes> thumbnail_data = new RefCountedBytes;
   SkAutoLockPixels thumbnail_lock(thumbnail);
   bool encoded = gfx::JPEGCodec::Encode(
       reinterpret_cast<unsigned char*>(thumbnail.getAddr32(0, 0)),
       gfx::JPEGCodec::FORMAT_BGRA, thumbnail.width(),
       thumbnail.height(),
       static_cast<int>(thumbnail.rowBytes()), 90,
       &thumbnail_data->data);
   if (!encoded)
     return false;
-  return SetPageThumbnail(url, thumbnail_data, score);
+  if (!SetPageThumbnailNoDB(url, thumbnail_data, score))
+    return false;
+
+  // Update the database.
+  if (!db_.get())
+    return true;
+  std::map<GURL, size_t>::iterator found = canonical_urls_.find(url);
+  if (found == canonical_urls_.end())
+    return false;
+  size_t index = found->second;
+
+  MostVisitedURL& most_visited = top_sites_[index];
+  db_->SetPageThumbnail(most_visited, index, top_images_[most_visited.url]);
+  return true;
 }
 
 // private
-bool TopSites::SetPageThumbnail(const GURL& url,
-                                const RefCountedBytes* thumbnail_data,
-                                const ThumbnailScore& score) {
+bool TopSites::SetPageThumbnailNoDB(const GURL& url,
+                                    const RefCountedBytes* thumbnail_data,
+                                    const ThumbnailScore& score) {
   AutoLock lock(lock_);
 
   std::map<GURL, size_t>::iterator found = canonical_urls_.find(url);
   if (found == canonical_urls_.end())
     return false;  // This URL is not known to us.
   MostVisitedURL& most_visited = top_sites_[found->second];
   Images& image = top_images_[most_visited.url];
 
   // When comparing the thumbnail scores, we need to take into account the
   // redirect hops, which are not generated when the thumbnail is because the
@@ skipping 187 matching lines @@
     MostVisitedURLList pages) {
   AutoLock lock(lock_);
   UpdateMostVisited(&pages);
 }
 
 void TopSites::SetMockHistoryService(MockHistoryService* mhs) {
   mock_history_service_ = mhs;
 }
 
 }  // namespace history