Implementing URL prefix match for history thumbnail cache.

chrome/browser/history/top_sites_impl.cc

 // Copyright (c) 2013 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_impl.h"
 
 #include <algorithm>
 #include <set>
 
 #include "base/bind.h"
@@ skipping 194 matching lines @@
                      base::MessageLoopProxy::current(),
                      callback));
       return;
     }
     filtered_urls = thread_safe_cache_->top_sites();
   }
   callback.Run(filtered_urls);
 }
 
 bool TopSitesImpl::GetPageThumbnail(
-    const GURL& url, scoped_refptr<base::RefCountedMemory>* bytes) {
+    const GURL& url,
+    bool prefix_match,
+    scoped_refptr<base::RefCountedMemory>* bytes) {
   // WARNING: this may be invoked on any thread.
+  // Perform exact match.
   {
     base::AutoLock lock(lock_);
     if (thread_safe_cache_->GetPageThumbnail(url, bytes))
       return true;
   }
 
   // Resource bundle is thread safe.
   for (size_t i = 0; i < arraysize(kPrepopulatedPages); i++) {
     if (url == prepopulated_page_urls_[i]) {
       *bytes = ResourceBundle::GetSharedInstance().
           LoadDataResourceBytesForScale(
               kPrepopulatedPages[i].thumbnail_id,
               ui::SCALE_FACTOR_100P);
       return true;
     }
   }
 
+  if (prefix_match) {
+    // Still not found, so strip "?query#ref", and perform prefix match.
+    GURL::Replacements replacements;
+    replacements.ClearQuery();
+    replacements.ClearRef();
+    GURL url_stripped(url.ReplaceComponents(replacements));
+    base::AutoLock lock(lock_);
+    GURL canonical_url(
+        thread_safe_cache_->GetCanonicalURLForPrefix(url_stripped));
+    if (thread_safe_cache_->GetPageThumbnail(canonical_url, bytes))
+      return true;
+  }
   return false;
 }
 
 bool TopSitesImpl::GetPageThumbnailScore(const GURL& url,
                                          ThumbnailScore* score) {
   // WARNING: this may be invoked on any thread.
   base::AutoLock lock(lock_);
   return thread_safe_cache_->GetPageThumbnailScore(url, score);
 }
 
@@ skipping 526 matching lines @@
   SetTopSites(pages);
 
   // Used only in testing.
   content::NotificationService::current()->Notify(
       chrome::NOTIFICATION_TOP_SITES_UPDATED,
       content::Source<TopSitesImpl>(this),
       content::Details<CancelableRequestProvider::Handle>(&handle));
 }
 
 }  // namespace history