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
PredictorDatabaseInternal(Profile * profile)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
~PredictorDatabaseInternal()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
Initialize()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
SetCancelled()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
LogDatabaseStats()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
PredictorDatabase(Profile * profile)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
~PredictorDatabase()136 PredictorDatabase::~PredictorDatabase() {
137 }
138
Shutdown()139 void PredictorDatabase::Shutdown() {
140 db_->SetCancelled();
141 }
142
143 scoped_refptr<AutocompleteActionPredictorTable>
autocomplete_table()144 PredictorDatabase::autocomplete_table() {
145 return db_->autocomplete_table_;
146 }
147
148 scoped_refptr<LoggedInPredictorTable>
logged_in_table()149 PredictorDatabase::logged_in_table() {
150 return db_->logged_in_table_;
151 }
152
153 scoped_refptr<ResourcePrefetchPredictorTables>
resource_prefetch_tables()154 PredictorDatabase::resource_prefetch_tables() {
155 return db_->resource_prefetch_tables_;
156 }
157
GetDatabase()158 sql::Connection* PredictorDatabase::GetDatabase() {
159 return db_->db_.get();
160 }
161
162 } // namespace predictors
163