Precache tracking database

components/precache/core/precache_statistics_table.cc

Index: components/precache/core/precache_statistics_table.cc
diff --git a/components/precache/core/precache_statistics_table.cc b/components/precache/core/precache_statistics_table.cc
new file mode 100644
index 0000000000000000000000000000000000000000..1bfa4721bc8fc61921ab4cdfbadeef23caf36d20
--- /dev/null
+++ b/components/precache/core/precache_statistics_table.cc
@@ -0,0 +1,162 @@
+// 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_statistics_table.h"
+
+#include "base/logging.h"
+#include "base/strings/stringprintf.h"
+#include "base/time/time.h"
+#include "sql/connection.h"
+#include "sql/statement.h"
+
+using sql::Statement;
+
+namespace {
+
+const char kStatisticsTableName[] = "precache_statistics";
+
+}  // namespace
+
+namespace precache {
+
+PrecacheStatisticsTable::PrecacheStatistics::PrecacheStatistics()
+    : precached_bytes(0),
+      downloaded_bytes(0),
+      downloaded_bytes_cellular(0),
+      saved_bytes(0),
+      saved_bytes_cellular(0) {}
+
+PrecacheStatisticsTable::PrecacheStatistics::PrecacheStatistics(
+    int64 precached_bytes, int64 downloaded_bytes,
+    int64 downloaded_bytes_cellular, int64 saved_bytes,
+    int64 saved_bytes_cellular)
+    : precached_bytes(precached_bytes),
+      downloaded_bytes(downloaded_bytes),
+      downloaded_bytes_cellular(downloaded_bytes_cellular),
+      saved_bytes(saved_bytes),
+      saved_bytes_cellular(saved_bytes_cellular) {}
+
+bool PrecacheStatisticsTable::PrecacheStatistics::operator==(
+    const PrecacheStatistics& other) const {
+  return precached_bytes == other.precached_bytes &&
+         downloaded_bytes == other.downloaded_bytes &&
+         downloaded_bytes_cellular == other.downloaded_bytes_cellular &&
+         saved_bytes == other.saved_bytes &&
+         saved_bytes_cellular == other.saved_bytes_cellular;
+}
+
+PrecacheStatisticsTable::PrecacheStatisticsTable() : db_(NULL) {}
+
+PrecacheStatisticsTable::~PrecacheStatisticsTable() {}
+
+void PrecacheStatisticsTable::Init(sql::Connection* db) {
+  DCHECK(!db_);  // Init must only be called once.
+  DCHECK(db);    // The database connection must be non-NULL.
+  db_ = db;
+  CreateTableIfNonExistent();
+}
+
+void PrecacheStatisticsTable::IncreaseDailyStats(
+    const base::Time& time_on_day, const PrecacheStatistics& stats_change) {
+  DCHECK(db_);
+
+  // Create a row for that day if there is no existing one.
+  Statement insert_statement(db_->GetCachedStatement(
+      SQL_FROM_HERE,
+      base::StringPrintf("INSERT OR IGNORE INTO %s (date) VALUES(?)",
+                         kStatisticsTableName).c_str()));
+
+  insert_statement.BindInt64(0, GetKey(time_on_day));
+  insert_statement.Run();
+
+  // Increase the recorded statistics for that day.
+  Statement update_statement(db_->GetCachedStatement(
+      SQL_FROM_HERE,
+      base::StringPrintf(
+          "UPDATE %s SET precached_bytes = precached_bytes + ?, "
+              "downloaded_bytes = downloaded_bytes + ?, "
+              "downloaded_bytes_cellular = downloaded_bytes_cellular + ?, "
+              "saved_bytes = saved_bytes + ?, "
+              "saved_bytes_cellular = saved_bytes_cellular + ? "
+          "WHERE date = ?", kStatisticsTableName).c_str()));
+
+  update_statement.BindInt64(0, stats_change.precached_bytes);
+  update_statement.BindInt64(1, stats_change.downloaded_bytes);
+  update_statement.BindInt64(2, stats_change.downloaded_bytes_cellular);
+  update_statement.BindInt64(3, stats_change.saved_bytes);
+  update_statement.BindInt64(4, stats_change.saved_bytes_cellular);
+  update_statement.BindInt64(5, GetKey(time_on_day));
+
+  update_statement.Run();
+}
+
+void PrecacheStatisticsTable::GetAllStatsBetween(const base::Time& start_time,
+                                                 const base::Time& end_time,
+                                                 PrecacheStatisticsMap* map) {
+  DCHECK(db_);
+
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE,
+      base::StringPrintf(
+          "SELECT date, precached_bytes, downloaded_bytes, "
+          "downloaded_bytes_cellular, saved_bytes, saved_bytes_cellular FROM "
+          "%s WHERE date >= ? and date <= ?",
+          kStatisticsTableName).c_str()));
+
+  statement.BindInt64(0, GetKey(start_time));
+  statement.BindInt64(1, GetKey(end_time));
+
+  while (statement.Step()) {
+    base::Time date = base::Time::FromInternalValue(statement.ColumnInt64(0));
+    PrecacheStatistics stats;
+
+    stats.precached_bytes = statement.ColumnInt64(1);
+    stats.downloaded_bytes = statement.ColumnInt64(2);
+    stats.downloaded_bytes_cellular = statement.ColumnInt64(3);
+    stats.saved_bytes = statement.ColumnInt64(4);
+    stats.saved_bytes_cellular = statement.ColumnInt64(5);
+
+    (*map)[date] = stats;
+  }
+}
+
+void PrecacheStatisticsTable::DeleteAllStatsBetween(
+    const base::Time& delete_begin, const base::Time& delete_end) {
+  DCHECK(db_);
+
+  Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE,
+      base::StringPrintf("DELETE FROM %s WHERE date >= ? AND date <= ?",
+                         kStatisticsTableName).c_str()));
+
+  statement.BindInt64(0, GetKey(delete_begin));
+  statement.BindInt64(1, GetKey(delete_end));
+
+  statement.Run();
+}
+
+// static
+int64 PrecacheStatisticsTable::GetKey(const base::Time& time_on_day) {
+  return time_on_day.LocalMidnight().ToInternalValue();
+}
+
+void PrecacheStatisticsTable::CreateTableIfNonExistent() {
+  DCHECK(db_);
+  if (db_->DoesTableExist(kStatisticsTableName))
+    return;
+
+  if (!db_->Execute(base::StringPrintf(
+           "CREATE TABLE %s (date INTEGER, "
+           "precached_bytes INTEGER DEFAULT 0, "
+           "downloaded_bytes INTEGER DEFAULT 0, "
+           "downloaded_bytes_cellular INTEGER DEFAULT 0, "
+           "saved_bytes INTEGER DEFAULT 0, "
+           "saved_bytes_cellular INTEGER DEFAULT 0, "
+           "PRIMARY KEY(date))",
+           kStatisticsTableName).c_str())) {
+    DLOG(WARNING) << "Could not create precache statistics table in database.";
+  }
+}
+
+}  // namespace precache