chrome/browser/predictors componentization part-1

chrome/browser/predictors/predictor_database.cc

 // 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/predictors/predictor_database.h"
 
 #include "base/bind.h"
-#include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/stringprintf.h"
 #include "base/metrics/histogram.h"
 #include "chrome/browser/predictors/autocomplete_action_predictor_table.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor_tables.h"
 #include "chrome/browser/prerender/prerender_field_trial.h"
-#include "chrome/browser/profiles/profile.h"
 #include "content/public/browser/browser_thread.h"
 #include "sql/connection.h"
 #include "sql/statement.h"
 
 using content::BrowserThread;
 
 namespace {
 
 // TODO(shishir): This should move to a more generic name.
 const FilePath::CharType kPredictorDatabaseName[] =
     FILE_PATH_LITERAL("Network Action Predictor");
 
 }  // namespace
 
 namespace predictors {
 
-// Refcounted as it is created, initialized and destroyed on a different thread
-// to the DB thread that is required for all methods performing database access.
-class PredictorDatabaseInternal
-    : public base::RefCountedThreadSafe<PredictorDatabaseInternal> {
- private:
-  friend class base::RefCountedThreadSafe<PredictorDatabaseInternal>;
-  friend class PredictorDatabase;
-
-  explicit PredictorDatabaseInternal(Profile* profile);
-  virtual ~PredictorDatabaseInternal();
-
-  // Opens the database file from the profile path. Separated from the
-  // constructor to ease construction/destruction of this object on one thread
-  // but database access on the DB thread.
-  void Initialize();
-  void LogDatabaseStats();  //  DB Thread.
-
-  // Cancels pending DB transactions. Should only be called on the UI thread.
-  void SetCancelled();
-
-  bool is_resource_prefetch_predictor_enabled_;
-  FilePath db_path_;
-  scoped_ptr<sql::Connection> db_;
-
-  // TODO(shishir): These tables may not need to be refcounted. Maybe move them
-  // to using a WeakPtr instead.
-  scoped_refptr<AutocompleteActionPredictorTable> autocomplete_table_;
-  scoped_refptr<ResourcePrefetchPredictorTables> resource_prefetch_tables_;
-
-  DISALLOW_COPY_AND_ASSIGN(PredictorDatabaseInternal);
-};
-
-
-PredictorDatabaseInternal::PredictorDatabaseInternal(Profile* profile)
-    : db_path_(profile->GetPath().Append(kPredictorDatabaseName)),
+PredictorDatabase::PredictorDatabase(
+    const FilePath& path,
+    bool is_resource_prefetch_predictor_enabled)
+    : db_path_(path.Append(kPredictorDatabaseName)),
+      is_resource_prefetch_predictor_enabled_(
+          is_resource_prefetch_predictor_enabled),
       db_(new sql::Connection()),
       autocomplete_table_(new AutocompleteActionPredictorTable()),
       resource_prefetch_tables_(new ResourcePrefetchPredictorTables()) {
-  ResourcePrefetchPredictorConfig config;
-  is_resource_prefetch_predictor_enabled_ =
-      IsSpeculativeResourcePrefetchingEnabled(profile, &config);
+  BrowserThread::PostTask(BrowserThread::DB, FROM_HERE,
+      base::Bind(&PredictorDatabase::Initialize, this));
 }
 
-PredictorDatabaseInternal::~PredictorDatabaseInternal() {
+PredictorDatabase::~PredictorDatabase() {
   // The connection pointer needs to be deleted on the DB thread since there
   // might be a task in progress on the DB thread which uses this connection.
   BrowserThread::DeleteSoon(BrowserThread::DB, FROM_HERE, db_.release());
 }
 
-void PredictorDatabaseInternal::Initialize() {
+scoped_refptr<AutocompleteActionPredictorTable>
+    PredictorDatabase::autocomplete_table() {
+  return autocomplete_table_;
+}
+
+scoped_refptr<ResourcePrefetchPredictorTables>
+    PredictorDatabase::resource_prefetch_tables() {
+  return resource_prefetch_tables_;
+}
+
+sql::Connection* PredictorDatabase::GetDatabase() {
+  return db_.get();
+}
+
+void PredictorDatabase::Initialize() {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
   db_->set_exclusive_locking();
   bool success = db_->Open(db_path_);
 
   if (!success)
     return;
 
   autocomplete_table_->Initialize(db_.get());
   resource_prefetch_tables_->Initialize(db_.get());
 
   LogDatabaseStats();
 }
 
-void PredictorDatabaseInternal::SetCancelled() {
+void PredictorDatabase::SetCancelled() {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   autocomplete_table_->SetCancelled();
   resource_prefetch_tables_->SetCancelled();
 }
 
-void PredictorDatabaseInternal::LogDatabaseStats() {
+void PredictorDatabase::LogDatabaseStats() {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
 
   int64 db_size;
   bool success = file_util::GetFileSize(db_path_, &db_size);
   DCHECK(success) << "Failed to get file size for " << db_path_.value();
   UMA_HISTOGRAM_MEMORY_KB("PredictorDatabase.DatabaseSizeKB",
                           static_cast<int>(db_size / 1024));
 
   autocomplete_table_->LogDatabaseStats();
   if (is_resource_prefetch_predictor_enabled_)
     resource_prefetch_tables_->LogDatabaseStats();
 }
 
-PredictorDatabase::PredictorDatabase(Profile* profile)
-    : db_(new PredictorDatabaseInternal(profile)) {
-  BrowserThread::PostTask(BrowserThread::DB, FROM_HERE,
-      base::Bind(&PredictorDatabaseInternal::Initialize, db_));
-}
-
-PredictorDatabase::~PredictorDatabase() {
-}
-
-void PredictorDatabase::Shutdown() {
-  db_->SetCancelled();
-}
-
-scoped_refptr<AutocompleteActionPredictorTable>
-    PredictorDatabase::autocomplete_table() {
-  return db_->autocomplete_table_;
-}
-
-scoped_refptr<ResourcePrefetchPredictorTables>
-    PredictorDatabase::resource_prefetch_tables() {
-  return db_->resource_prefetch_tables_;
-}
-
-sql::Connection* PredictorDatabase::GetDatabase() {
-  return db_->db_.get();
-}
-
 }  // namespace predictors