Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(568)

Unified Diff: chrome/browser/history/top_sites_database.cc

Issue 870063002: Componentize TopSites, TopSitesBackend, TopSitesDatabase (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@815983002
Patch Set: Fix typo Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome/browser/history/top_sites_database.h ('k') | chrome/browser/history/top_sites_database_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: chrome/browser/history/top_sites_database.cc
diff --git a/chrome/browser/history/top_sites_database.cc b/chrome/browser/history/top_sites_database.cc
deleted file mode 100644
index 1fe4c25bdd1c9e4ac9353a012bbd6989a904d7d8..0000000000000000000000000000000000000000
--- a/chrome/browser/history/top_sites_database.cc
+++ /dev/null
@@ -1,736 +0,0 @@
-// Copyright (c) 2012 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_database.h"
-
-#include "base/files/file_util.h"
-#include "base/memory/ref_counted.h"
-#include "base/metrics/histogram.h"
-#include "base/strings/string_split.h"
-#include "base/strings/string_util.h"
-#include "chrome/browser/history/top_sites.h"
-#include "components/history/core/browser/history_types.h"
-#include "components/history/core/common/thumbnail_score.h"
-#include "sql/connection.h"
-#include "sql/recovery.h"
-#include "sql/statement.h"
-#include "sql/transaction.h"
-#include "third_party/sqlite/sqlite3.h"
-
-// Description of database table:
-//
-// thumbnails
-// url URL of the sites for which we have a thumbnail.
-// url_rank Index of the URL in that thumbnail, 0-based. The thumbnail
-// with the highest rank will be the next one evicted. Forced
-// thumbnails have a rank of -1.
-// title The title to display under that thumbnail.
-// redirects A space separated list of URLs that are known to redirect
-// to this url.
-// boring_score How "boring" that thumbnail is. See ThumbnailScore.
-// good_clipping True if the thumbnail was clipped from the bottom, keeping
-// the entire width of the window. See ThumbnailScore.
-// at_top True if the thumbnail was captured at the top of the
-// website.
-// last_updated The time at which this thumbnail was last updated.
-// load_completed True if the thumbnail was captured after the page load was
-// completed.
-// last_forced If this is a forced thumbnail, records the last time it
-// was forced. If it's not a forced thumbnail, 0.
-
-namespace {
-
-// For this database, schema migrations are deprecated after two
-// years. This means that the oldest non-deprecated version should be
-// two years old or greater (thus the migrations to get there are
-// older). Databases containing deprecated versions will be cleared
-// at startup. Since this database is a cache, losing old data is not
-// fatal (in fact, very old data may be expired immediately at startup
-// anyhow).
-
-// Version 3: b6d6a783/r231648 by beaudoin@chromium.org on 2013-10-29
-// Version 2: eb0b24e6/r87284 by satorux@chromium.org on 2011-05-31
-// Version 1: 809cc4d8/r64072 by sky@chromium.org on 2010-10-27 (deprecated)
-
-// NOTE(shess): When changing the version, add a new golden file for
-// the new version and a test to verify that Init() works with it.
-// NOTE(shess): RecoverDatabaseOrRaze() depends on the specific
-// version number. The code is subtle and in development, contact me
-// if the necessary changes are not obvious.
-static const int kVersionNumber = 3;
-static const int kDeprecatedVersionNumber = 1; // and earlier.
-
-bool InitTables(sql::Connection* db) {
- const char kThumbnailsSql[] =
- "CREATE TABLE IF NOT EXISTS thumbnails ("
- "url LONGVARCHAR PRIMARY KEY,"
- "url_rank INTEGER,"
- "title LONGVARCHAR,"
- "thumbnail BLOB,"
- "redirects LONGVARCHAR,"
- "boring_score DOUBLE DEFAULT 1.0,"
- "good_clipping INTEGER DEFAULT 0,"
- "at_top INTEGER DEFAULT 0,"
- "last_updated INTEGER DEFAULT 0,"
- "load_completed INTEGER DEFAULT 0,"
- "last_forced INTEGER DEFAULT 0)";
- return db->Execute(kThumbnailsSql);
-}
-
-// Encodes redirects into a string.
-std::string GetRedirects(const history::MostVisitedURL& url) {
- std::vector<std::string> redirects;
- for (size_t i = 0; i < url.redirects.size(); i++)
- redirects.push_back(url.redirects[i].spec());
- return JoinString(redirects, ' ');
-}
-
-// Decodes redirects from a string and sets them for the url.
-void SetRedirects(const std::string& redirects, history::MostVisitedURL* url) {
- std::vector<std::string> redirects_vector;
- base::SplitStringAlongWhitespace(redirects, &redirects_vector);
- for (size_t i = 0; i < redirects_vector.size(); ++i) {
- GURL redirects_url(redirects_vector[i]);
- if (redirects_url.is_valid())
- url->redirects.push_back(redirects_url);
- }
-}
-
-// Track various failure (and success) cases in recovery code.
-//
-// TODO(shess): The recovery code is complete, but by nature runs in challenging
-// circumstances, so initially the default error response is to leave the
-// existing database in place. This histogram is intended to expose the
-// failures seen in the fleet. Frequent failure cases can be explored more
-// deeply to see if the complexity to fix them is warranted. Infrequent failure
-// cases can be resolved by marking the database unrecoverable (which will
-// delete the data).
-//
-// Based on the thumbnail_database.cc recovery code, FAILED_SCOPER should
-// dominate, followed distantly by FAILED_META, with few or no other failures.
-enum RecoveryEventType {
- // Database successfully recovered.
- RECOVERY_EVENT_RECOVERED = 0,
-
- // Database successfully deprecated.
- RECOVERY_EVENT_DEPRECATED,
-
- // Sqlite.RecoveryEvent can usually be used to get more detail about the
- // specific failure (see sql/recovery.cc).
- RECOVERY_EVENT_FAILED_SCOPER,
- RECOVERY_EVENT_FAILED_META_VERSION,
- RECOVERY_EVENT_FAILED_META_WRONG_VERSION,
- RECOVERY_EVENT_FAILED_META_INIT,
- RECOVERY_EVENT_FAILED_SCHEMA_INIT,
- RECOVERY_EVENT_FAILED_AUTORECOVER_THUMBNAILS,
- RECOVERY_EVENT_FAILED_COMMIT,
-
- // Track invariants resolved by FixThumbnailsTable().
- RECOVERY_EVENT_INVARIANT_RANK,
- RECOVERY_EVENT_INVARIANT_REDIRECT,
- RECOVERY_EVENT_INVARIANT_CONTIGUOUS,
-
- // Always keep this at the end.
- RECOVERY_EVENT_MAX,
-};
-
-void RecordRecoveryEvent(RecoveryEventType recovery_event) {
- UMA_HISTOGRAM_ENUMERATION("History.TopSitesRecovery",
- recovery_event, RECOVERY_EVENT_MAX);
-}
-
-// Most corruption comes down to atomic updates between pages being broken
-// somehow. This can result in either missing data, or overlapping data,
-// depending on the operation broken. This table has large rows, which will use
-// overflow pages, so it is possible (though unlikely) that a chain could fit
-// together and yield a row with errors.
-void FixThumbnailsTable(sql::Connection* db) {
- // Enforce invariant separating forced and non-forced thumbnails.
- const char kFixRankSql[] =
- "DELETE FROM thumbnails "
- "WHERE (url_rank = -1 AND last_forced = 0) "
- "OR (url_rank <> -1 AND last_forced <> 0)";
- ignore_result(db->Execute(kFixRankSql));
- if (db->GetLastChangeCount() > 0)
- RecordRecoveryEvent(RECOVERY_EVENT_INVARIANT_RANK);
-
- // Enforce invariant that url is in its own redirects.
- const char kFixRedirectsSql[] =
- "DELETE FROM thumbnails "
- "WHERE url <> substr(redirects, -length(url), length(url))";
- ignore_result(db->Execute(kFixRedirectsSql));
- if (db->GetLastChangeCount() > 0)
- RecordRecoveryEvent(RECOVERY_EVENT_INVARIANT_REDIRECT);
-
- // Enforce invariant that url_rank>=0 forms a contiguous series.
- // TODO(shess): I have not found an UPDATE+SUBSELECT method of managing this.
- // It can be done with a temporary table and a subselect, but doing it
- // manually is easier to follow. Another option would be to somehow integrate
- // the renumbering into the table recovery code.
- const char kByRankSql[] =
- "SELECT url_rank, rowid FROM thumbnails WHERE url_rank <> -1 "
- "ORDER BY url_rank";
- sql::Statement select_statement(db->GetUniqueStatement(kByRankSql));
-
- const char kAdjustRankSql[] =
- "UPDATE thumbnails SET url_rank = ? WHERE rowid = ?";
- sql::Statement update_statement(db->GetUniqueStatement(kAdjustRankSql));
-
- // Update any rows where |next_rank| doesn't match |url_rank|.
- int next_rank = 0;
- bool adjusted = false;
- while (select_statement.Step()) {
- const int url_rank = select_statement.ColumnInt(0);
- if (url_rank != next_rank) {
- adjusted = true;
- update_statement.Reset(true);
- update_statement.BindInt(0, next_rank);
- update_statement.BindInt64(1, select_statement.ColumnInt64(1));
- update_statement.Run();
- }
- ++next_rank;
- }
- if (adjusted)
- RecordRecoveryEvent(RECOVERY_EVENT_INVARIANT_CONTIGUOUS);
-}
-
-// Recover the database to the extent possible, razing it if recovery is not
-// possible.
-void RecoverDatabaseOrRaze(sql::Connection* db, const base::FilePath& db_path) {
- // NOTE(shess): If the version changes, review this code.
- DCHECK_EQ(3, kVersionNumber);
-
- // It is almost certain that some operation against |db| will fail, prevent
- // reentry.
- db->reset_error_callback();
-
- // For generating histogram stats.
- size_t thumbnails_recovered = 0;
- int64 original_size = 0;
- base::GetFileSize(db_path, &original_size);
-
- scoped_ptr<sql::Recovery> recovery = sql::Recovery::Begin(db, db_path);
- if (!recovery) {
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_SCOPER);
- return;
- }
-
- // Setup the meta recovery table and fetch the version number from the corrupt
- // database.
- int version = 0;
- if (!recovery->SetupMeta() || !recovery->GetMetaVersionNumber(&version)) {
- // TODO(shess): Prior histograms indicate all failures are in creating the
- // recover virtual table for corrupt.meta. The table may not exist, or the
- // database may be too far gone. Either way, unclear how to resolve.
- sql::Recovery::Rollback(recovery.Pass());
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_META_VERSION);
- return;
- }
-
- // This code runs in a context which may be able to read version information
- // that the regular deprecation path cannot. The effect of this code will be
- // to raze the database.
- if (version <= kDeprecatedVersionNumber) {
- sql::Recovery::Unrecoverable(recovery.Pass());
- RecordRecoveryEvent(RECOVERY_EVENT_DEPRECATED);
- return;
- }
-
- // TODO(shess): Earlier versions have been deprecated, later versions should
- // be impossible. Unrecoverable() seems like a feasible response if this is
- // infrequent enough.
- if (version != 2 && version != 3) {
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_META_WRONG_VERSION);
- sql::Recovery::Rollback(recovery.Pass());
- return;
- }
-
- // Both v2 and v3 recover to current schema version.
- sql::MetaTable recover_meta_table;
- if (!recover_meta_table.Init(recovery->db(), kVersionNumber,
- kVersionNumber)) {
- sql::Recovery::Rollback(recovery.Pass());
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_META_INIT);
- return;
- }
-
- // Create a fresh version of the schema. The recovery code uses
- // conflict-resolution to handle duplicates, so any indices are necessary.
- if (!InitTables(recovery->db())) {
- // TODO(shess): Unable to create the new schema in the new database. The
- // new database should be a temporary file, so being unable to work with it
- // is pretty unclear.
- //
- // What are the potential responses, even? The recovery database could be
- // opened as in-memory. If the temp database had a filesystem problem and
- // the temp filesystem differs from the main database, then that could fix
- // it.
- sql::Recovery::Rollback(recovery.Pass());
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_SCHEMA_INIT);
- return;
- }
-
- // The |1| is because v2 [thumbnails] has one less column than v3 did. In the
- // v2 case the column will get default values.
- if (!recovery->AutoRecoverTable("thumbnails", 1, &thumbnails_recovered)) {
- sql::Recovery::Rollback(recovery.Pass());
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_AUTORECOVER_THUMBNAILS);
- return;
- }
-
- // TODO(shess): Inline this?
- FixThumbnailsTable(recovery->db());
-
- if (!sql::Recovery::Recovered(recovery.Pass())) {
- // TODO(shess): Very unclear what this failure would actually mean, and what
- // should be done. Add histograms to Recovered() implementation to get some
- // insight.
- RecordRecoveryEvent(RECOVERY_EVENT_FAILED_COMMIT);
- return;
- }
-
- // Track the size of the recovered database relative to the size of the input
- // database. The size should almost always be smaller, unless the input
- // database was empty to start with. If the percentage results are very low,
- // something is awry.
- int64 final_size = 0;
- if (original_size > 0 &&
- base::GetFileSize(db_path, &final_size) &&
- final_size > 0) {
- UMA_HISTOGRAM_PERCENTAGE("History.TopSitesRecoveredPercentage",
- final_size * 100 / original_size);
- }
-
- // Using 10,000 because these cases mostly care about "none recovered" and
- // "lots recovered". More than 10,000 rows recovered probably means there's
- // something wrong with the profile.
- UMA_HISTOGRAM_COUNTS_10000("History.TopSitesRecoveredRowsThumbnails",
- thumbnails_recovered);
-
- RecordRecoveryEvent(RECOVERY_EVENT_RECOVERED);
-}
-
-void DatabaseErrorCallback(sql::Connection* db,
- const base::FilePath& db_path,
- int extended_error,
- sql::Statement* stmt) {
- // TODO(shess): Assert that this is running on a safe thread. AFAICT, should
- // be the history thread, but at this level I can't see how to reach that.
-
- // Attempt to recover corrupt databases.
- int error = (extended_error & 0xFF);
- if (error == SQLITE_CORRUPT ||
- error == SQLITE_CANTOPEN ||
- error == SQLITE_NOTADB) {
- RecoverDatabaseOrRaze(db, db_path);
- }
-
- // TODO(shess): This database's error histograms look like:
- // 84% SQLITE_CORRUPT, SQLITE_CANTOPEN, SQLITE_NOTADB
- // 7% SQLITE_ERROR
- // 6% SQLITE_IOERR variants
- // 2% SQLITE_READONLY
- // .4% SQLITE_FULL
- // nominal SQLITE_TOBIG, SQLITE_AUTH, and SQLITE_BUSY. In the case of
- // thumbnail_database.cc, as soon as the recovery code landed, SQLITE_IOERR
- // shot to leadership. If the I/O error is system-level, there is probably no
- // hope, but if it is restricted to something about the database file, it is
- // possible that the recovery code could be brought to bear. In fact, it is
- // possible that running recovery would be a reasonable default when errors
- // are seen.
-
- // The default handling is to assert on debug and to ignore on release.
- if (!sql::Connection::ShouldIgnoreSqliteError(extended_error))
- DLOG(FATAL) << db->GetErrorMessage();
-}
-
-} // namespace
-
-namespace history {
-
-// static
-const int TopSitesDatabase::kRankOfForcedURL = -1;
-
-// static
-const int TopSitesDatabase::kRankOfNonExistingURL = -2;
-
-TopSitesDatabase::TopSitesDatabase() {
-}
-
-TopSitesDatabase::~TopSitesDatabase() {
-}
-
-bool TopSitesDatabase::Init(const base::FilePath& db_name) {
- // Retry failed InitImpl() in case the recovery system fixed things.
- // TODO(shess): Instrument to figure out if there are any persistent failure
- // cases which do not resolve themselves.
- const size_t kAttempts = 2;
-
- for (size_t i = 0; i < kAttempts; ++i) {
- if (InitImpl(db_name))
- return true;
-
- meta_table_.Reset();
- db_.reset();
- }
- return false;
-}
-
-bool TopSitesDatabase::InitImpl(const base::FilePath& db_name) {
- const bool file_existed = base::PathExists(db_name);
-
- db_.reset(CreateDB(db_name));
- if (!db_)
- return false;
-
- // An older version had data with no meta table. Deprecate by razing.
- // TODO(shess): Just have RazeIfDeprecated() handle this case.
- const bool does_meta_exist = sql::MetaTable::DoesTableExist(db_.get());
- if (!does_meta_exist && file_existed) {
- if (!db_->Raze())
- return false;
- }
-
- // Clear databases which are too old to process.
- DCHECK_LT(kDeprecatedVersionNumber, kVersionNumber);
- sql::MetaTable::RazeIfDeprecated(db_.get(), kDeprecatedVersionNumber);
-
- // Scope initialization in a transaction so we can't be partially
- // initialized.
- sql::Transaction transaction(db_.get());
- // TODO(shess): Failure to open transaction is bad, address it.
- if (!transaction.Begin())
- return false;
-
- if (!meta_table_.Init(db_.get(), kVersionNumber, kVersionNumber))
- return false;
-
- if (!InitTables(db_.get()))
- return false;
-
- if (meta_table_.GetVersionNumber() == 2) {
- if (!UpgradeToVersion3()) {
- LOG(WARNING) << "Unable to upgrade top sites database to version 3.";
- return false;
- }
- }
-
- // Version check.
- if (meta_table_.GetVersionNumber() != kVersionNumber)
- return false;
-
- // Initialization is complete.
- if (!transaction.Commit())
- return false;
-
- return true;
-}
-
-bool TopSitesDatabase::UpgradeToVersion3() {
- // Add 'last_forced' column.
- if (!db_->Execute(
- "ALTER TABLE thumbnails ADD last_forced INTEGER DEFAULT 0")) {
- NOTREACHED();
- return false;
- }
- meta_table_.SetVersionNumber(3);
- return true;
-}
-
-void TopSitesDatabase::GetPageThumbnails(MostVisitedURLList* urls,
- URLToImagesMap* thumbnails) {
- sql::Statement statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "SELECT url, url_rank, title, thumbnail, redirects, "
- "boring_score, good_clipping, at_top, last_updated, load_completed, "
- "last_forced FROM thumbnails ORDER BY url_rank, last_forced"));
-
- if (!statement.is_valid()) {
- LOG(WARNING) << db_->GetErrorMessage();
- return;
- }
-
- urls->clear();
- thumbnails->clear();
-
- while (statement.Step()) {
- // Results are sorted by url_rank. For forced thumbnails with url_rank = -1,
- // thumbnails are sorted by last_forced.
- MostVisitedURL url;
- GURL gurl(statement.ColumnString(0));
- url.url = gurl;
- url.title = statement.ColumnString16(2);
- url.last_forced_time =
- base::Time::FromInternalValue(statement.ColumnInt64(10));
- std::string redirects = statement.ColumnString(4);
- SetRedirects(redirects, &url);
- urls->push_back(url);
-
- std::vector<unsigned char> data;
- statement.ColumnBlobAsVector(3, &data);
- Images thumbnail;
- if (!data.empty())
- thumbnail.thumbnail = base::RefCountedBytes::TakeVector(&data);
- thumbnail.thumbnail_score.boring_score = statement.ColumnDouble(5);
- thumbnail.thumbnail_score.good_clipping = statement.ColumnBool(6);
- thumbnail.thumbnail_score.at_top = statement.ColumnBool(7);
- thumbnail.thumbnail_score.time_at_snapshot =
- base::Time::FromInternalValue(statement.ColumnInt64(8));
- thumbnail.thumbnail_score.load_completed = statement.ColumnBool(9);
- (*thumbnails)[gurl] = thumbnail;
- }
-}
-
-void TopSitesDatabase::SetPageThumbnail(const MostVisitedURL& url,
- int new_rank,
- const Images& thumbnail) {
- sql::Transaction transaction(db_.get());
- transaction.Begin();
-
- int rank = GetURLRank(url);
- if (rank == kRankOfNonExistingURL) {
- AddPageThumbnail(url, new_rank, thumbnail);
- } else {
- UpdatePageRankNoTransaction(url, new_rank);
- UpdatePageThumbnail(url, thumbnail);
- }
-
- transaction.Commit();
-}
-
-bool TopSitesDatabase::UpdatePageThumbnail(
- const MostVisitedURL& url, const Images& thumbnail) {
- sql::Statement statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails SET "
- "title = ?, thumbnail = ?, redirects = ?, "
- "boring_score = ?, good_clipping = ?, at_top = ?, last_updated = ?, "
- "load_completed = ?, last_forced = ?"
- "WHERE url = ? "));
- statement.BindString16(0, url.title);
- if (thumbnail.thumbnail.get() && thumbnail.thumbnail->front()) {
- statement.BindBlob(1, thumbnail.thumbnail->front(),
- static_cast<int>(thumbnail.thumbnail->size()));
- }
- statement.BindString(2, GetRedirects(url));
- const ThumbnailScore& score = thumbnail.thumbnail_score;
- statement.BindDouble(3, score.boring_score);
- statement.BindBool(4, score.good_clipping);
- statement.BindBool(5, score.at_top);
- statement.BindInt64(6, score.time_at_snapshot.ToInternalValue());
- statement.BindBool(7, score.load_completed);
- statement.BindInt64(8, url.last_forced_time.ToInternalValue());
- statement.BindString(9, url.url.spec());
-
- return statement.Run();
-}
-
-void TopSitesDatabase::AddPageThumbnail(const MostVisitedURL& url,
- int new_rank,
- const Images& thumbnail) {
- sql::Statement statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "INSERT OR REPLACE INTO thumbnails "
- "(url, url_rank, title, thumbnail, redirects, "
- "boring_score, good_clipping, at_top, last_updated, load_completed, "
- "last_forced) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"));
- statement.BindString(0, url.url.spec());
- statement.BindInt(1, kRankOfForcedURL); // Fist make it a forced thumbnail.
- statement.BindString16(2, url.title);
- if (thumbnail.thumbnail.get() && thumbnail.thumbnail->front()) {
- statement.BindBlob(3, thumbnail.thumbnail->front(),
- static_cast<int>(thumbnail.thumbnail->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());
- statement.BindBool(9, score.load_completed);
- int64 last_forced = url.last_forced_time.ToInternalValue();
- DCHECK((last_forced == 0) == (new_rank != kRankOfForcedURL))
- << "Thumbnail without a forced time stamp has a forced rank, or the "
- << "opposite.";
- statement.BindInt64(10, last_forced);
- if (!statement.Run())
- return;
-
- // Update rank if this is not a forced thumbnail.
- if (new_rank != kRankOfForcedURL)
- UpdatePageRankNoTransaction(url, new_rank);
-}
-
-void TopSitesDatabase::UpdatePageRank(const MostVisitedURL& url,
- int new_rank) {
- DCHECK((url.last_forced_time.ToInternalValue() == 0) ==
- (new_rank != kRankOfForcedURL))
- << "Thumbnail without a forced time stamp has a forced rank, or the "
- << "opposite.";
- sql::Transaction transaction(db_.get());
- transaction.Begin();
- UpdatePageRankNoTransaction(url, new_rank);
- transaction.Commit();
-}
-
-// Caller should have a transaction open.
-void TopSitesDatabase::UpdatePageRankNoTransaction(
- const MostVisitedURL& url, int new_rank) {
- DCHECK_GT(db_->transaction_nesting(), 0);
- DCHECK((url.last_forced_time.is_null()) == (new_rank != kRankOfForcedURL))
- << "Thumbnail without a forced time stamp has a forced rank, or the "
- << "opposite.";
-
- int prev_rank = GetURLRank(url);
- if (prev_rank == kRankOfNonExistingURL) {
- LOG(WARNING) << "Updating rank of an unknown URL: " << url.url.spec();
- return;
- }
-
- // Shift the ranks.
- if (prev_rank > new_rank) {
- if (new_rank == kRankOfForcedURL) {
- // From non-forced to forced, shift down.
- // Example: 2 -> -1
- // -1, -1, -1, 0, 1, [2 -> -1], [3 -> 2], [4 -> 3]
- sql::Statement shift_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = url_rank - 1 "
- "WHERE url_rank > ?"));
- shift_statement.BindInt(0, prev_rank);
- shift_statement.Run();
- } else {
- // From non-forced to non-forced, shift up.
- // Example: 3 -> 1
- // -1, -1, -1, 0, [1 -> 2], [2 -> 3], [3 -> 1], 4
- sql::Statement shift_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = url_rank + 1 "
- "WHERE url_rank >= ? AND url_rank < ?"));
- shift_statement.BindInt(0, new_rank);
- shift_statement.BindInt(1, prev_rank);
- shift_statement.Run();
- }
- } else if (prev_rank < new_rank) {
- if (prev_rank == kRankOfForcedURL) {
- // From non-forced to forced, shift up.
- // Example: -1 -> 2
- // -1, [-1 -> 2], -1, 0, 1, [2 -> 3], [3 -> 4], [4 -> 5]
- sql::Statement shift_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = url_rank + 1 "
- "WHERE url_rank >= ?"));
- shift_statement.BindInt(0, new_rank);
- shift_statement.Run();
- } else {
- // From non-forced to non-forced, shift down.
- // Example: 1 -> 3.
- // -1, -1, -1, 0, [1 -> 3], [2 -> 1], [3 -> 2], 4
- sql::Statement shift_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = url_rank - 1 "
- "WHERE url_rank > ? AND url_rank <= ?"));
- shift_statement.BindInt(0, prev_rank);
- shift_statement.BindInt(1, new_rank);
- shift_statement.Run();
- }
- }
-
- // Set the url's rank and last_forced, since the latter changes when a URL
- // goes from forced to non-forced and vice-versa.
- sql::Statement set_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = ?, last_forced = ? "
- "WHERE url == ?"));
- set_statement.BindInt(0, new_rank);
- set_statement.BindInt64(1, url.last_forced_time.ToInternalValue());
- set_statement.BindString(2, url.url.spec());
- set_statement.Run();
-}
-
-bool TopSitesDatabase::GetPageThumbnail(const GURL& url,
- Images* thumbnail) {
- sql::Statement statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "SELECT thumbnail, boring_score, good_clipping, at_top, last_updated "
- "FROM thumbnails WHERE url=?"));
- statement.BindString(0, url.spec());
- if (!statement.Step())
- return false;
-
- std::vector<unsigned char> data;
- statement.ColumnBlobAsVector(0, &data);
- thumbnail->thumbnail = base::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 TopSitesDatabase::GetURLRank(const MostVisitedURL& url) {
- sql::Statement select_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "SELECT url_rank "
- "FROM thumbnails WHERE url=?"));
- select_statement.BindString(0, url.url.spec());
- if (select_statement.Step())
- return select_statement.ColumnInt(0);
-
- return kRankOfNonExistingURL;
-}
-
-// Remove the record for this URL. Returns true iff removed successfully.
-bool TopSitesDatabase::RemoveURL(const MostVisitedURL& url) {
- int old_rank = GetURLRank(url);
- if (old_rank == kRankOfNonExistingURL)
- return false;
-
- sql::Transaction transaction(db_.get());
- transaction.Begin();
- if (old_rank != kRankOfForcedURL) {
- // Decrement all following ranks.
- sql::Statement shift_statement(db_->GetCachedStatement(
- SQL_FROM_HERE,
- "UPDATE thumbnails "
- "SET url_rank = url_rank - 1 "
- "WHERE url_rank > ?"));
- shift_statement.BindInt(0, old_rank);
-
- if (!shift_statement.Run())
- return false;
- }
-
- sql::Statement delete_statement(
- db_->GetCachedStatement(SQL_FROM_HERE,
- "DELETE FROM thumbnails WHERE url = ?"));
- delete_statement.BindString(0, url.url.spec());
-
- if (!delete_statement.Run())
- return false;
-
- return transaction.Commit();
-}
-
-sql::Connection* TopSitesDatabase::CreateDB(const base::FilePath& db_name) {
- scoped_ptr<sql::Connection> db(new sql::Connection());
- // Settings copied from ThumbnailDatabase.
- db->set_histogram_tag("TopSites");
- db->set_error_callback(base::Bind(&DatabaseErrorCallback,
- db.get(), db_name));
- db->set_page_size(4096);
- db->set_cache_size(32);
-
- if (!db->Open(db_name))
- return NULL;
- return db.release();
-}
-
-} // namespace history
« no previous file with comments | « chrome/browser/history/top_sites_database.h ('k') | chrome/browser/history/top_sites_database_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698