15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/predictors/predictor_database.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/metrics/histogram.h" 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/predictors/autocomplete_action_predictor_table.h" 14c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/browser/predictors/logged_in_predictor_table.h" 151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/predictors/resource_prefetch_predictor.h" 161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/predictors/resource_prefetch_predictor_tables.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prerender/prerender_field_trial.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sql/connection.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sql/statement.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using content::BrowserThread; 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(shishir): This should move to a more generic name. 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::FilePath::CharType kPredictorDatabaseName[] = 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("Network Action Predictor"); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace predictors { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Refcounted as it is created, initialized and destroyed on a different thread 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to the DB thread that is required for all methods performing database access. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class PredictorDatabaseInternal 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public base::RefCountedThreadSafe<PredictorDatabaseInternal> { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class base::RefCountedThreadSafe<PredictorDatabaseInternal>; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class PredictorDatabase; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit PredictorDatabaseInternal(Profile* profile); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~PredictorDatabaseInternal(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Opens the database file from the profile path. Separated from the 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // constructor to ease construction/destruction of this object on one thread 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // but database access on the DB thread. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Initialize(); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LogDatabaseStats(); // DB Thread. 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancels pending DB transactions. Should only be called on the UI thread. 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetCancelled(); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 551320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci bool is_resource_prefetch_predictor_enabled_; 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath db_path_; 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<sql::Connection> db_; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(shishir): These tables may not need to be refcounted. Maybe move them 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to using a WeakPtr instead. 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<AutocompleteActionPredictorTable> autocomplete_table_; 62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_refptr<LoggedInPredictorTable> logged_in_table_; 631320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_refptr<ResourcePrefetchPredictorTables> resource_prefetch_tables_; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(PredictorDatabaseInternal); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)PredictorDatabaseInternal::PredictorDatabaseInternal(Profile* profile) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : db_path_(profile->GetPath().Append(kPredictorDatabaseName)), 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) db_(new sql::Connection()), 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) autocomplete_table_(new AutocompleteActionPredictorTable()), 731320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci logged_in_table_(new LoggedInPredictorTable()), 741320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci resource_prefetch_tables_(new ResourcePrefetchPredictorTables()) { 75eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch db_->set_histogram_tag("Predictor"); 761320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ResourcePrefetchPredictorConfig config; 771320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci is_resource_prefetch_predictor_enabled_ = 781320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci IsSpeculativeResourcePrefetchingEnabled(profile, &config); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)PredictorDatabaseInternal::~PredictorDatabaseInternal() { 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The connection pointer needs to be deleted on the DB thread since there 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // might be a task in progress on the DB thread which uses this connection. 845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (BrowserThread::IsMessageLoopValid(BrowserThread::DB)) 855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BrowserThread::DeleteSoon(BrowserThread::DB, FROM_HERE, db_.release()); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void PredictorDatabaseInternal::Initialize() { 89a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB) || 90a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) !BrowserThread::IsMessageLoopValid(BrowserThread::DB)); 91c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // TODO(tburkard): figure out if we need this. 92c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // db_->set_exclusive_locking(); 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool success = db_->Open(db_path_); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!success) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) autocomplete_table_->Initialize(db_.get()); 99c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) logged_in_table_->Initialize(db_.get()); 1001320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci resource_prefetch_tables_->Initialize(db_.get()); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LogDatabaseStats(); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void PredictorDatabaseInternal::SetCancelled() { 106a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI) || 107a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) !BrowserThread::IsMessageLoopValid(BrowserThread::UI)); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) autocomplete_table_->SetCancelled(); 110c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) logged_in_table_->SetCancelled(); 1111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci resource_prefetch_tables_->SetCancelled(); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void PredictorDatabaseInternal::LogDatabaseStats() { 115a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB) || 116a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) !BrowserThread::IsMessageLoopValid(BrowserThread::DB)); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 db_size; 119a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) bool success = base::GetFileSize(db_path_, &db_size); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(success) << "Failed to get file size for " << db_path_.value(); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UMA_HISTOGRAM_MEMORY_KB("PredictorDatabase.DatabaseSizeKB", 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<int>(db_size / 1024)); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) autocomplete_table_->LogDatabaseStats(); 125c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) logged_in_table_->LogDatabaseStats(); 1261320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (is_resource_prefetch_predictor_enabled_) 1271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci resource_prefetch_tables_->LogDatabaseStats(); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)PredictorDatabase::PredictorDatabase(Profile* profile) 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : db_(new PredictorDatabaseInternal(profile)) { 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&PredictorDatabaseInternal::Initialize, db_)); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)PredictorDatabase::~PredictorDatabase() { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void PredictorDatabase::Shutdown() { 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_->SetCancelled(); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_refptr<AutocompleteActionPredictorTable> 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PredictorDatabase::autocomplete_table() { 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return db_->autocomplete_table_; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 148c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)scoped_refptr<LoggedInPredictorTable> 149c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) PredictorDatabase::logged_in_table() { 150c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return db_->logged_in_table_; 151c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 152c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucciscoped_refptr<ResourcePrefetchPredictorTables> 1541320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci PredictorDatabase::resource_prefetch_tables() { 1551320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci return db_->resource_prefetch_tables_; 1561320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 1571320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)sql::Connection* PredictorDatabase::GetDatabase() { 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return db_->db_.get(); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace predictors 163