Precache tracking database

components/precache/core/precache_url_table.cc

+// Copyright 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 "components/precache/core/precache_url_table.h"
+
+#include <string>
+
+#include "base/logging.h"
+#include "sql/connection.h"
+#include "sql/statement.h"
+
+using sql::Statement;
+
+namespace {
+
+// Returns the spec of the given URL.
+std::string GetKey(const GURL& url) {
+  return url.spec();
+}
+
+}  // namespace
+
+namespace precache {
+
+PrecacheURLTable::PrecacheURLTable() : db_(NULL) {}
+
+PrecacheURLTable::~PrecacheURLTable() {}
+
+bool PrecacheURLTable::Init(sql::Connection* db) {
+  DCHECK(!db_);  // Init must only be called once.
+  DCHECK(db);    // The database connection must be non-NULL.
+  db_ = db;
+  return CreateTableIfNonExistent();
+}
+
+void PrecacheURLTable::AddURL(const GURL& url,
+                              const base::Time& precache_time) {
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE,
+      "INSERT OR REPLACE INTO precache_urls (url, time) VALUES(?,?)"));
+
+  statement.BindString(0, GetKey(url));
+  statement.BindInt64(1, precache_time.ToInternalValue());
+  statement.Run();
+}
+
+bool PrecacheURLTable::HasURL(const GURL& url) {
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "SELECT time FROM precache_urls WHERE url=?"));
+
+  statement.BindString(0, GetKey(url));
+  return statement.Step();
+}
+
+void PrecacheURLTable::DeleteURL(const GURL& url) {
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "DELETE FROM precache_urls WHERE url=?"));
+
+  statement.BindString(0, GetKey(url));
+  statement.Run();
+}
+
+void PrecacheURLTable::DeleteAllPrecachedBefore(const base::Time& delete_end) {
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "DELETE FROM precache_urls WHERE time < ?"));
+
+  statement.BindInt64(0, delete_end.ToInternalValue());
+  statement.Run();
+}
+
+void PrecacheURLTable::GetAllDataForTesting(std::map<GURL, base::Time>* map) {
+  map->clear();
+
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "SELECT url, time FROM precache_urls"));
+
+  while (statement.Step()) {
+    GURL url = GURL(statement.ColumnString(0));
+    (*map)[url] = base::Time::FromInternalValue(statement.ColumnInt64(1));
+  }
+}
+
+bool PrecacheURLTable::CreateTableIfNonExistent() {
+  return db_->Execute(
+      "CREATE TABLE IF NOT EXISTS precache_urls (url TEXT PRIMARY KEY, time "
+      "INTEGER)");
+}
+
+}  // namespace precache