Home | History | Annotate | Download | only in history 
      1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/history/top_sites_backend.h"
      6 
      7 #include "base/bind.h"
      8 #include "base/bind_helpers.h"
      9 #include "base/files/file_path.h"
     10 #include "base/memory/ref_counted.h"
     11 #include "base/task/cancelable_task_tracker.h"
     12 #include "chrome/browser/history/top_sites_database.h"
     13 #include "content/public/browser/browser_thread.h"
     14 
     15 using content::BrowserThread;
     16 
     17 namespace history {
     18 
     19 TopSitesBackend::TopSitesBackend()
     20     : db_(new TopSitesDatabase()) {
     21 }
     22 
     23 void TopSitesBackend::Init(const base::FilePath& path) {
     24   db_path_ = path;
     25   BrowserThread::PostTask(
     26       BrowserThread::DB, FROM_HERE,
     27       base::Bind(&TopSitesBackend::InitDBOnDBThread, this, path));
     28 }
     29 
     30 void TopSitesBackend::Shutdown() {
     31   BrowserThread::PostTask(
     32       BrowserThread::DB, FROM_HERE,
     33       base::Bind(&TopSitesBackend::ShutdownDBOnDBThread, this));
     34 }
     35 
     36 void TopSitesBackend::GetMostVisitedThumbnails(
     37     const GetMostVisitedThumbnailsCallback& callback,
     38     base::CancelableTaskTracker* tracker) {
     39   scoped_refptr<MostVisitedThumbnails> thumbnails = new MostVisitedThumbnails();
     40 
     41   tracker->PostTaskAndReply(
     42       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(),
     43       FROM_HERE,
     44       base::Bind(&TopSitesBackend::GetMostVisitedThumbnailsOnDBThread,
     45                  this,
     46                  thumbnails),
     47       base::Bind(callback, thumbnails));
     48 }
     49 
     50 void TopSitesBackend::UpdateTopSites(const TopSitesDelta& delta) {
     51   BrowserThread::PostTask(
     52       BrowserThread::DB, FROM_HERE,
     53       base::Bind(&TopSitesBackend::UpdateTopSitesOnDBThread, this, delta));
     54 }
     55 
     56 void TopSitesBackend::SetPageThumbnail(const MostVisitedURL& url,
     57                                        int url_rank,
     58                                        const Images& thumbnail) {
     59   BrowserThread::PostTask(
     60       BrowserThread::DB, FROM_HERE,
     61       base::Bind(&TopSitesBackend::SetPageThumbnailOnDBThread, this, url,
     62                  url_rank, thumbnail));
     63 }
     64 
     65 void TopSitesBackend::ResetDatabase() {
     66   BrowserThread::PostTask(
     67       BrowserThread::DB, FROM_HERE,
     68       base::Bind(&TopSitesBackend::ResetDatabaseOnDBThread, this, db_path_));
     69 }
     70 
     71 void TopSitesBackend::DoEmptyRequest(const base::Closure& reply,
     72                                      base::CancelableTaskTracker* tracker) {
     73   tracker->PostTaskAndReply(
     74       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(),
     75       FROM_HERE,
     76       base::Bind(&base::DoNothing),
     77       reply);
     78 }
     79 
     80 TopSitesBackend::~TopSitesBackend() {
     81   DCHECK(!db_);  // Shutdown should have happened first (which results in
     82                  // nulling out db).
     83 }
     84 
     85 void TopSitesBackend::InitDBOnDBThread(const base::FilePath& path) {
     86   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
     87   if (!db_->Init(path)) {
     88     LOG(ERROR) << "Failed to initialize database.";
     89     db_.reset();
     90   }
     91 }
     92 
     93 void TopSitesBackend::ShutdownDBOnDBThread() {
     94   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
     95   db_.reset();
     96 }
     97 
     98 void TopSitesBackend::GetMostVisitedThumbnailsOnDBThread(
     99     scoped_refptr<MostVisitedThumbnails> thumbnails) {
    100   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
    101 
    102   if (db_) {
    103     db_->GetPageThumbnails(&(thumbnails->most_visited),
    104                            &(thumbnails->url_to_images_map));
    105   }
    106 }
    107 
    108 void TopSitesBackend::UpdateTopSitesOnDBThread(const TopSitesDelta& delta) {
    109   if (!db_)
    110     return;
    111 
    112   for (size_t i = 0; i < delta.deleted.size(); ++i)
    113     db_->RemoveURL(delta.deleted[i]);
    114 
    115   for (size_t i = 0; i < delta.added.size(); ++i)
    116     db_->SetPageThumbnail(delta.added[i].url, delta.added[i].rank, Images());
    117 
    118   for (size_t i = 0; i < delta.moved.size(); ++i)
    119     db_->UpdatePageRank(delta.moved[i].url, delta.moved[i].rank);
    120 }
    121 
    122 void TopSitesBackend::SetPageThumbnailOnDBThread(const MostVisitedURL& url,
    123                                                  int url_rank,
    124                                                  const Images& thumbnail) {
    125   if (!db_)
    126     return;
    127 
    128   db_->SetPageThumbnail(url, url_rank, thumbnail);
    129 }
    130 
    131 void TopSitesBackend::ResetDatabaseOnDBThread(const base::FilePath& file_path) {
    132   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
    133   db_.reset(NULL);
    134   sql::Connection::Delete(db_path_);
    135   db_.reset(new TopSitesDatabase());
    136   InitDBOnDBThread(db_path_);
    137 }
    138 
    139 }  // namespace history
    140