Index: chrome/browser/predictors/predictor_database.cc |
=================================================================== |
--- chrome/browser/predictors/predictor_database.cc (revision 194439) |
+++ chrome/browser/predictors/predictor_database.cc (working copy) |
@@ -11,6 +11,7 @@ |
#include "base/metrics/histogram.h" |
#include "base/stringprintf.h" |
#include "chrome/browser/predictors/autocomplete_action_predictor_table.h" |
+#include "chrome/browser/predictors/logged_in_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" |
@@ -58,6 +59,7 @@ |
// 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<LoggedInPredictorTable> logged_in_table_; |
scoped_refptr<ResourcePrefetchPredictorTables> resource_prefetch_tables_; |
DISALLOW_COPY_AND_ASSIGN(PredictorDatabaseInternal); |
@@ -68,6 +70,7 @@ |
: db_path_(profile->GetPath().Append(kPredictorDatabaseName)), |
db_(new sql::Connection()), |
autocomplete_table_(new AutocompleteActionPredictorTable()), |
+ logged_in_table_(new LoggedInPredictorTable()), |
resource_prefetch_tables_(new ResourcePrefetchPredictorTables()) { |
ResourcePrefetchPredictorConfig config; |
is_resource_prefetch_predictor_enabled_ = |
@@ -89,6 +92,7 @@ |
return; |
autocomplete_table_->Initialize(db_.get()); |
+ logged_in_table_->Initialize(db_.get()); |
resource_prefetch_tables_->Initialize(db_.get()); |
LogDatabaseStats(); |
@@ -98,6 +102,7 @@ |
CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
autocomplete_table_->SetCancelled(); |
+ logged_in_table_->SetCancelled(); |
resource_prefetch_tables_->SetCancelled(); |
} |
@@ -111,6 +116,7 @@ |
static_cast<int>(db_size / 1024)); |
autocomplete_table_->LogDatabaseStats(); |
+ logged_in_table_->LogDatabaseStats(); |
if (is_resource_prefetch_predictor_enabled_) |
resource_prefetch_tables_->LogDatabaseStats(); |
} |
@@ -133,6 +139,11 @@ |
return db_->autocomplete_table_; |
} |
+scoped_refptr<LoggedInPredictorTable> |
+ PredictorDatabase::logged_in_table() { |
+ return db_->logged_in_table_; |
+} |
+ |
scoped_refptr<ResourcePrefetchPredictorTables> |
PredictorDatabase::resource_prefetch_tables() { |
return db_->resource_prefetch_tables_; |