1 // Copyright (c) 2012 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/predictors/predictor_database.h" 6 7 #include "base/bind.h" 8 #include "base/files/file_path.h" 9 #include "base/files/file_util.h" 10 #include "base/logging.h" 11 #include "base/metrics/histogram.h" 12 #include "base/strings/stringprintf.h" 13 #include "chrome/browser/predictors/autocomplete_action_predictor_table.h" 14 #include "chrome/browser/predictors/logged_in_predictor_table.h" 15 #include "chrome/browser/predictors/resource_prefetch_predictor.h" 16 #include "chrome/browser/predictors/resource_prefetch_predictor_tables.h" 17 #include "chrome/browser/prerender/prerender_field_trial.h" 18 #include "chrome/browser/profiles/profile.h" 19 #include "content/public/browser/browser_thread.h" 20 #include "sql/connection.h" 21 #include "sql/statement.h" 22 23 using content::BrowserThread; 24 25 namespace { 26 27 // TODO(shishir): This should move to a more generic name. 28 const base::FilePath::CharType kPredictorDatabaseName[] = 29 FILE_PATH_LITERAL("Network Action Predictor"); 30 31 } // namespace 32 33 namespace predictors { 34 35 // Refcounted as it is created, initialized and destroyed on a different thread 36 // to the DB thread that is required for all methods performing database access. 37 class PredictorDatabaseInternal 38 : public base::RefCountedThreadSafe<PredictorDatabaseInternal> { 39 private: 40 friend class base::RefCountedThreadSafe<PredictorDatabaseInternal>; 41 friend class PredictorDatabase; 42 43 explicit PredictorDatabaseInternal(Profile* profile); 44 virtual ~PredictorDatabaseInternal(); 45 46 // Opens the database file from the profile path. Separated from the 47 // constructor to ease construction/destruction of this object on one thread 48 // but database access on the DB thread. 49 void Initialize(); 50 void LogDatabaseStats(); // DB Thread. 51 52 // Cancels pending DB transactions. Should only be called on the UI thread. 53 void SetCancelled(); 54 55 bool is_resource_prefetch_predictor_enabled_; 56 base::FilePath db_path_; 57 scoped_ptr<sql::Connection> db_; 58 59 // TODO(shishir): These tables may not need to be refcounted. Maybe move them 60 // to using a WeakPtr instead. 61 scoped_refptr<AutocompleteActionPredictorTable> autocomplete_table_; 62 scoped_refptr<LoggedInPredictorTable> logged_in_table_; 63 scoped_refptr<ResourcePrefetchPredictorTables> resource_prefetch_tables_; 64 65 DISALLOW_COPY_AND_ASSIGN(PredictorDatabaseInternal); 66 }; 67 68 69 PredictorDatabaseInternal::PredictorDatabaseInternal(Profile* profile) 70 : db_path_(profile->GetPath().Append(kPredictorDatabaseName)), 71 db_(new sql::Connection()), 72 autocomplete_table_(new AutocompleteActionPredictorTable()), 73 logged_in_table_(new LoggedInPredictorTable()), 74 resource_prefetch_tables_(new ResourcePrefetchPredictorTables()) { 75 db_->set_histogram_tag("Predictor"); 76 ResourcePrefetchPredictorConfig config; 77 is_resource_prefetch_predictor_enabled_ = 78 IsSpeculativeResourcePrefetchingEnabled(profile, &config); 79 } 80 81 PredictorDatabaseInternal::~PredictorDatabaseInternal() { 82 // The connection pointer needs to be deleted on the DB thread since there 83 // might be a task in progress on the DB thread which uses this connection. 84 if (BrowserThread::IsMessageLoopValid(BrowserThread::DB)) 85 BrowserThread::DeleteSoon(BrowserThread::DB, FROM_HERE, db_.release()); 86 } 87 88 void PredictorDatabaseInternal::Initialize() { 89 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB) || 90 !BrowserThread::IsMessageLoopValid(BrowserThread::DB)); 91 // TODO(tburkard): figure out if we need this. 92 // db_->set_exclusive_locking(); 93 bool success = db_->Open(db_path_); 94 95 if (!success) 96 return; 97 98 autocomplete_table_->Initialize(db_.get()); 99 logged_in_table_->Initialize(db_.get()); 100 resource_prefetch_tables_->Initialize(db_.get()); 101 102 LogDatabaseStats(); 103 } 104 105 void PredictorDatabaseInternal::SetCancelled() { 106 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI) || 107 !BrowserThread::IsMessageLoopValid(BrowserThread::UI)); 108 109 autocomplete_table_->SetCancelled(); 110 logged_in_table_->SetCancelled(); 111 resource_prefetch_tables_->SetCancelled(); 112 } 113 114 void PredictorDatabaseInternal::LogDatabaseStats() { 115 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB) || 116 !BrowserThread::IsMessageLoopValid(BrowserThread::DB)); 117 118 int64 db_size; 119 bool success = base::GetFileSize(db_path_, &db_size); 120 DCHECK(success) << "Failed to get file size for " << db_path_.value(); 121 UMA_HISTOGRAM_MEMORY_KB("PredictorDatabase.DatabaseSizeKB", 122 static_cast<int>(db_size / 1024)); 123 124 autocomplete_table_->LogDatabaseStats(); 125 logged_in_table_->LogDatabaseStats(); 126 if (is_resource_prefetch_predictor_enabled_) 127 resource_prefetch_tables_->LogDatabaseStats(); 128 } 129 130 PredictorDatabase::PredictorDatabase(Profile* profile) 131 : db_(new PredictorDatabaseInternal(profile)) { 132 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, 133 base::Bind(&PredictorDatabaseInternal::Initialize, db_)); 134 } 135 136 PredictorDatabase::~PredictorDatabase() { 137 } 138 139 void PredictorDatabase::Shutdown() { 140 db_->SetCancelled(); 141 } 142 143 scoped_refptr<AutocompleteActionPredictorTable> 144 PredictorDatabase::autocomplete_table() { 145 return db_->autocomplete_table_; 146 } 147 148 scoped_refptr<LoggedInPredictorTable> 149 PredictorDatabase::logged_in_table() { 150 return db_->logged_in_table_; 151 } 152 153 scoped_refptr<ResourcePrefetchPredictorTables> 154 PredictorDatabase::resource_prefetch_tables() { 155 return db_->resource_prefetch_tables_; 156 } 157 158 sql::Connection* PredictorDatabase::GetDatabase() { 159 return db_->db_.get(); 160 } 161 162 } // namespace predictors 163