| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/autocomplete/network_action_predictor.h" | 5 #include "chrome/browser/predictors/autocomplete_action_predictor.h" |
| 6 | 6 |
| 7 #include <math.h> | 7 #include <math.h> |
| 8 | 8 |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/i18n/case_conversion.h" | 12 #include "base/i18n/case_conversion.h" |
| 13 #include "base/metrics/histogram.h" | 13 #include "base/metrics/histogram.h" |
| 14 #include "base/string_util.h" | 14 #include "base/string_util.h" |
| 15 #include "base/stringprintf.h" | 15 #include "base/stringprintf.h" |
| 16 #include "base/utf_string_conversions.h" | 16 #include "base/utf_string_conversions.h" |
| 17 #include "chrome/browser/autocomplete/autocomplete.h" | 17 #include "chrome/browser/autocomplete/autocomplete.h" |
| 18 #include "chrome/browser/autocomplete/autocomplete_match.h" | 18 #include "chrome/browser/autocomplete/autocomplete_match.h" |
| 19 #include "chrome/browser/autocomplete/network_action_predictor_database.h" | |
| 20 #include "chrome/browser/history/history.h" | 19 #include "chrome/browser/history/history.h" |
| 21 #include "chrome/browser/history/history_notifications.h" | 20 #include "chrome/browser/history/history_notifications.h" |
| 22 #include "chrome/browser/history/in_memory_database.h" | 21 #include "chrome/browser/history/in_memory_database.h" |
| 22 #include "chrome/browser/predictors/predictor_database.h" |
| 23 #include "chrome/browser/predictors/predictor_database_factory.h" |
| 23 #include "chrome/browser/prerender/prerender_field_trial.h" | 24 #include "chrome/browser/prerender/prerender_field_trial.h" |
| 24 #include "chrome/browser/prerender/prerender_manager.h" | 25 #include "chrome/browser/prerender/prerender_manager.h" |
| 25 #include "chrome/browser/prerender/prerender_manager_factory.h" | 26 #include "chrome/browser/prerender/prerender_manager_factory.h" |
| 26 #include "chrome/browser/profiles/profile.h" | 27 #include "chrome/browser/profiles/profile.h" |
| 27 #include "chrome/common/chrome_notification_types.h" | 28 #include "chrome/common/chrome_notification_types.h" |
| 28 #include "chrome/common/guid.h" | 29 #include "chrome/common/guid.h" |
| 29 #include "content/public/browser/browser_thread.h" | 30 #include "content/public/browser/browser_thread.h" |
| 30 #include "content/public/browser/notification_details.h" | 31 #include "content/public/browser/notification_details.h" |
| 31 #include "content/public/browser/notification_service.h" | 32 #include "content/public/browser/notification_service.h" |
| 32 #include "content/public/browser/notification_source.h" | 33 #include "content/public/browser/notification_source.h" |
| 33 | 34 |
| 34 namespace { | 35 namespace { |
| 35 | 36 |
| 36 const float kConfidenceCutoff[] = { | 37 const float kConfidenceCutoff[] = { |
| 37 0.8f, | 38 0.8f, |
| 38 0.5f | 39 0.5f |
| 39 }; | 40 }; |
| 40 | 41 |
| 41 const size_t kMinimumUserTextLength = 1; | 42 const size_t kMinimumUserTextLength = 1; |
| 42 const int kMinimumNumberOfHits = 3; | 43 const int kMinimumNumberOfHits = 3; |
| 43 | 44 |
| 44 COMPILE_ASSERT(arraysize(kConfidenceCutoff) == | 45 COMPILE_ASSERT(arraysize(kConfidenceCutoff) == |
| 45 NetworkActionPredictor::LAST_PREDICT_ACTION, | 46 predictors::AutocompleteActionPredictor::LAST_PREDICT_ACTION, |
| 46 ConfidenceCutoff_count_mismatch); | 47 ConfidenceCutoff_count_mismatch); |
| 47 | 48 |
| 48 enum DatabaseAction { | 49 enum DatabaseAction { |
| 49 DATABASE_ACTION_ADD, | 50 DATABASE_ACTION_ADD, |
| 50 DATABASE_ACTION_UPDATE, | 51 DATABASE_ACTION_UPDATE, |
| 51 DATABASE_ACTION_DELETE_SOME, | 52 DATABASE_ACTION_DELETE_SOME, |
| 52 DATABASE_ACTION_DELETE_ALL, | 53 DATABASE_ACTION_DELETE_ALL, |
| 53 DATABASE_ACTION_COUNT | 54 DATABASE_ACTION_COUNT |
| 54 }; | 55 }; |
| 55 | 56 |
| 56 bool IsAutocompleteMatchSearchType(const AutocompleteMatch& match) { | 57 bool IsAutocompleteMatchSearchType(const AutocompleteMatch& match) { |
| 57 switch (match.type) { | 58 switch (match.type) { |
| 58 // Matches using the user's default search engine. | 59 // Matches using the user's default search engine. |
| 59 case AutocompleteMatch::SEARCH_WHAT_YOU_TYPED: | 60 case AutocompleteMatch::SEARCH_WHAT_YOU_TYPED: |
| 60 case AutocompleteMatch::SEARCH_HISTORY: | 61 case AutocompleteMatch::SEARCH_HISTORY: |
| 61 case AutocompleteMatch::SEARCH_SUGGEST: | 62 case AutocompleteMatch::SEARCH_SUGGEST: |
| 62 // A match that uses a non-default search engine (e.g. for tab-to-search). | 63 // A match that uses a non-default search engine (e.g. for tab-to-search). |
| 63 case AutocompleteMatch::SEARCH_OTHER_ENGINE: | 64 case AutocompleteMatch::SEARCH_OTHER_ENGINE: |
| 64 return true; | 65 return true; |
| 65 | 66 |
| 66 default: | 67 default: |
| 67 return false; | 68 return false; |
| 68 } | 69 } |
| 69 } | 70 } |
| 70 | 71 |
| 71 } | 72 } // namespace |
| 72 | 73 |
| 73 const int NetworkActionPredictor::kMaximumDaysToKeepEntry = 14; | 74 namespace predictors { |
| 74 | 75 |
| 75 double NetworkActionPredictor::hit_weight_ = 1.0; | 76 const int AutocompleteActionPredictor::kMaximumDaysToKeepEntry = 14; |
| 76 | 77 |
| 77 NetworkActionPredictor::NetworkActionPredictor(Profile* profile) | 78 double AutocompleteActionPredictor::hit_weight_ = 1.0; |
| 79 |
| 80 AutocompleteActionPredictor::AutocompleteActionPredictor(Profile* profile) |
| 78 : profile_(profile), | 81 : profile_(profile), |
| 79 db_(new NetworkActionPredictorDatabase(profile)), | 82 db_(PredictorDatabaseFactory::GetForProfile( |
| 83 profile)->autocomplete_table()), |
| 80 initialized_(false) { | 84 initialized_(false) { |
| 81 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | |
| 82 base::Bind(&NetworkActionPredictorDatabase::Initialize, db_)); | |
| 83 | |
| 84 // Request the in-memory database from the history to force it to load so it's | 85 // Request the in-memory database from the history to force it to load so it's |
| 85 // available as soon as possible. | 86 // available as soon as possible. |
| 86 HistoryService* history_service = | 87 HistoryService* history_service = |
| 87 profile_->GetHistoryService(Profile::EXPLICIT_ACCESS); | 88 profile_->GetHistoryService(Profile::EXPLICIT_ACCESS); |
| 88 if (history_service) | 89 if (history_service) |
| 89 history_service->InMemoryDatabase(); | 90 history_service->InMemoryDatabase(); |
| 90 | 91 |
| 91 // Create local caches using the database as loaded. We will garbage collect | 92 // Create local caches using the database as loaded. We will garbage collect |
| 92 // rows from the caches and the database once the history service is | 93 // rows from the caches and the database once the history service is |
| 93 // available. | 94 // available. |
| 94 std::vector<NetworkActionPredictorDatabase::Row>* rows = | 95 std::vector<AutocompleteActionPredictorTable::Row>* rows = |
| 95 new std::vector<NetworkActionPredictorDatabase::Row>(); | 96 new std::vector<AutocompleteActionPredictorTable::Row>(); |
| 96 content::BrowserThread::PostTaskAndReply( | 97 content::BrowserThread::PostTaskAndReply( |
| 97 content::BrowserThread::DB, FROM_HERE, | 98 content::BrowserThread::DB, FROM_HERE, |
| 98 base::Bind(&NetworkActionPredictorDatabase::GetAllRows, db_, rows), | 99 base::Bind(&AutocompleteActionPredictorTable::GetAllRows, |
| 99 base::Bind(&NetworkActionPredictor::CreateCaches, AsWeakPtr(), | 100 db_, |
| 101 rows), |
| 102 base::Bind(&AutocompleteActionPredictor::CreateCaches, AsWeakPtr(), |
| 100 base::Owned(rows))); | 103 base::Owned(rows))); |
| 101 | |
| 102 } | 104 } |
| 103 | 105 |
| 104 NetworkActionPredictor::~NetworkActionPredictor() { | 106 AutocompleteActionPredictor::~AutocompleteActionPredictor() { |
| 105 } | 107 } |
| 106 | 108 |
| 107 void NetworkActionPredictor::RegisterTransitionalMatches( | 109 void AutocompleteActionPredictor::RegisterTransitionalMatches( |
| 108 const string16& user_text, | 110 const string16& user_text, |
| 109 const AutocompleteResult& result) { | 111 const AutocompleteResult& result) { |
| 110 if (user_text.length() < kMinimumUserTextLength) | 112 if (user_text.length() < kMinimumUserTextLength) |
| 111 return; | 113 return; |
| 112 const string16 lower_user_text(base::i18n::ToLower(user_text)); | 114 const string16 lower_user_text(base::i18n::ToLower(user_text)); |
| 113 | 115 |
| 114 // Merge this in to an existing match if we already saw |user_text| | 116 // Merge this in to an existing match if we already saw |user_text| |
| 115 std::vector<TransitionalMatch>::iterator match_it = | 117 std::vector<TransitionalMatch>::iterator match_it = |
| 116 std::find(transitional_matches_.begin(), transitional_matches_.end(), | 118 std::find(transitional_matches_.begin(), transitional_matches_.end(), |
| 117 lower_user_text); | 119 lower_user_text); |
| 118 | 120 |
| 119 if (match_it == transitional_matches_.end()) { | 121 if (match_it == transitional_matches_.end()) { |
| 120 TransitionalMatch transitional_match; | 122 TransitionalMatch transitional_match; |
| 121 transitional_match.user_text = lower_user_text; | 123 transitional_match.user_text = lower_user_text; |
| 122 match_it = transitional_matches_.insert(transitional_matches_.end(), | 124 match_it = transitional_matches_.insert(transitional_matches_.end(), |
| 123 transitional_match); | 125 transitional_match); |
| 124 } | 126 } |
| 125 | 127 |
| 126 for (AutocompleteResult::const_iterator it = result.begin(); | 128 for (AutocompleteResult::const_iterator it = result.begin(); |
| 127 it != result.end(); ++it) { | 129 it != result.end(); ++it) { |
| 128 if (std::find(match_it->urls.begin(), match_it->urls.end(), | 130 if (std::find(match_it->urls.begin(), match_it->urls.end(), |
| 129 it->destination_url) == match_it->urls.end()) { | 131 it->destination_url) == match_it->urls.end()) { |
| 130 match_it->urls.push_back(it->destination_url); | 132 match_it->urls.push_back(it->destination_url); |
| 131 } | 133 } |
| 132 } | 134 } |
| 133 } | 135 } |
| 134 | 136 |
| 135 void NetworkActionPredictor::ClearTransitionalMatches() { | 137 void AutocompleteActionPredictor::ClearTransitionalMatches() { |
| 136 transitional_matches_.clear(); | 138 transitional_matches_.clear(); |
| 137 } | 139 } |
| 138 | 140 |
| 139 // Given a match, return a recommended action. | 141 // Given a match, return a recommended action. |
| 140 NetworkActionPredictor::Action NetworkActionPredictor::RecommendAction( | 142 AutocompleteActionPredictor::Action |
| 141 const string16& user_text, | 143 AutocompleteActionPredictor::RecommendAction( |
| 142 const AutocompleteMatch& match) const { | 144 const string16& user_text, |
| 145 const AutocompleteMatch& match) const { |
| 143 bool is_in_db = false; | 146 bool is_in_db = false; |
| 144 const double confidence = CalculateConfidence(user_text, match, &is_in_db); | 147 const double confidence = CalculateConfidence(user_text, match, &is_in_db); |
| 145 DCHECK(confidence >= 0.0 && confidence <= 1.0); | 148 DCHECK(confidence >= 0.0 && confidence <= 1.0); |
| 146 | 149 |
| 147 UMA_HISTOGRAM_BOOLEAN("NetworkActionPredictor.MatchIsInDb", is_in_db); | 150 UMA_HISTOGRAM_BOOLEAN("AutocompleteActionPredictor.MatchIsInDb", is_in_db); |
| 148 | 151 |
| 149 if (is_in_db) { | 152 if (is_in_db) { |
| 150 // Multiple enties with the same URL are fine as the confidence may be | 153 // Multiple enties with the same URL are fine as the confidence may be |
| 151 // different. | 154 // different. |
| 152 tracked_urls_.push_back(std::make_pair(match.destination_url, confidence)); | 155 tracked_urls_.push_back(std::make_pair(match.destination_url, confidence)); |
| 153 UMA_HISTOGRAM_COUNTS_100("NetworkActionPredictor.Confidence", | 156 UMA_HISTOGRAM_COUNTS_100("AutocompleteActionPredictor.Confidence", |
| 154 confidence * 100); | 157 confidence * 100); |
| 155 } | 158 } |
| 156 | 159 |
| 157 // Map the confidence to an action. | 160 // Map the confidence to an action. |
| 158 Action action = ACTION_NONE; | 161 Action action = ACTION_NONE; |
| 159 for (int i = 0; i < LAST_PREDICT_ACTION; ++i) { | 162 for (int i = 0; i < LAST_PREDICT_ACTION; ++i) { |
| 160 if (confidence >= kConfidenceCutoff[i]) { | 163 if (confidence >= kConfidenceCutoff[i]) { |
| 161 action = static_cast<Action>(i); | 164 action = static_cast<Action>(i); |
| 162 break; | 165 break; |
| 163 } | 166 } |
| (...skipping 10 matching lines...) Expand all Loading... |
| 174 !prerender::IsOmniboxEnabled(profile_))) { | 177 !prerender::IsOmniboxEnabled(profile_))) { |
| 175 action = ACTION_PRECONNECT; | 178 action = ACTION_PRECONNECT; |
| 176 } | 179 } |
| 177 | 180 |
| 178 return action; | 181 return action; |
| 179 } | 182 } |
| 180 | 183 |
| 181 // Return true if the suggestion type warrants a TCP/IP preconnection. | 184 // Return true if the suggestion type warrants a TCP/IP preconnection. |
| 182 // i.e., it is now quite likely that the user will select the related domain. | 185 // i.e., it is now quite likely that the user will select the related domain. |
| 183 // static | 186 // static |
| 184 bool NetworkActionPredictor::IsPreconnectable(const AutocompleteMatch& match) { | 187 bool AutocompleteActionPredictor::IsPreconnectable( |
| 188 const AutocompleteMatch& match) { |
| 185 return IsAutocompleteMatchSearchType(match); | 189 return IsAutocompleteMatchSearchType(match); |
| 186 } | 190 } |
| 187 | 191 |
| 188 void NetworkActionPredictor::Shutdown() { | 192 void AutocompleteActionPredictor::Shutdown() { |
| 189 db_->OnPredictorDestroyed(); | |
| 190 } | 193 } |
| 191 | 194 |
| 192 void NetworkActionPredictor::Observe( | 195 void AutocompleteActionPredictor::Observe( |
| 193 int type, | 196 int type, |
| 194 const content::NotificationSource& source, | 197 const content::NotificationSource& source, |
| 195 const content::NotificationDetails& details) { | 198 const content::NotificationDetails& details) { |
| 196 switch (type) { | 199 switch (type) { |
| 197 case chrome::NOTIFICATION_HISTORY_URLS_DELETED: { | 200 case chrome::NOTIFICATION_HISTORY_URLS_DELETED: { |
| 198 DCHECK(initialized_); | 201 DCHECK(initialized_); |
| 199 const content::Details<const history::URLsDeletedDetails> | 202 const content::Details<const history::URLsDeletedDetails> |
| 200 urls_deleted_details = | 203 urls_deleted_details = |
| 201 content::Details<const history::URLsDeletedDetails>(details); | 204 content::Details<const history::URLsDeletedDetails>(details); |
| 202 if (urls_deleted_details->all_history) | 205 if (urls_deleted_details->all_history) |
| (...skipping 24 matching lines...) Expand all Loading... |
| 227 content::Source<Profile>(profile_)); | 230 content::Source<Profile>(profile_)); |
| 228 break; | 231 break; |
| 229 } | 232 } |
| 230 | 233 |
| 231 default: | 234 default: |
| 232 NOTREACHED() << "Unexpected notification observed."; | 235 NOTREACHED() << "Unexpected notification observed."; |
| 233 break; | 236 break; |
| 234 } | 237 } |
| 235 } | 238 } |
| 236 | 239 |
| 237 void NetworkActionPredictor::OnOmniboxOpenedUrl(const AutocompleteLog& log) { | 240 void AutocompleteActionPredictor::OnOmniboxOpenedUrl( |
| 241 const AutocompleteLog& log) { |
| 238 if (log.text.length() < kMinimumUserTextLength) | 242 if (log.text.length() < kMinimumUserTextLength) |
| 239 return; | 243 return; |
| 240 | 244 |
| 241 const AutocompleteMatch& match = log.result.match_at(log.selected_index); | 245 const AutocompleteMatch& match = log.result.match_at(log.selected_index); |
| 242 | 246 |
| 243 UMA_HISTOGRAM_BOOLEAN( | 247 UMA_HISTOGRAM_BOOLEAN( |
| 244 StringPrintf("Prerender.OmniboxNavigationsCouldPrerender_%.1f%s", | 248 StringPrintf("Prerender.OmniboxNavigationsCouldPrerender_%.1f%s", |
| 245 get_hit_weight(), | 249 get_hit_weight(), |
| 246 prerender::PrerenderManager::GetModeString()).c_str(), | 250 prerender::PrerenderManager::GetModeString()).c_str(), |
| 247 prerender::IsOmniboxEnabled(profile_)); | 251 prerender::IsOmniboxEnabled(profile_)); |
| 248 | 252 |
| 249 const GURL& opened_url = match.destination_url; | 253 const GURL& opened_url = match.destination_url; |
| 250 | 254 |
| 251 // If the Omnibox triggered a prerender but the URL doesn't match the one the | 255 // If the Omnibox triggered a prerender but the URL doesn't match the one the |
| 252 // user is navigating to, cancel the prerender. | 256 // user is navigating to, cancel the prerender. |
| 253 prerender::PrerenderManager* prerender_manager = | 257 prerender::PrerenderManager* prerender_manager = |
| 254 prerender::PrerenderManagerFactory::GetForProfile(profile_); | 258 prerender::PrerenderManagerFactory::GetForProfile(profile_); |
| 255 // |prerender_manager| can be NULL in incognito mode or if prerendering is | 259 // |prerender_manager| can be NULL in incognito mode or if prerendering is |
| 256 // otherwise disabled. | 260 // otherwise disabled. |
| 257 if (prerender_manager && !prerender_manager->IsPrerendering(opened_url)) | 261 if (prerender_manager && !prerender_manager->IsPrerendering(opened_url)) |
| 258 prerender_manager->CancelOmniboxPrerenders(); | 262 prerender_manager->CancelOmniboxPrerenders(); |
| 259 | 263 |
| 260 const string16 lower_user_text(base::i18n::ToLower(log.text)); | 264 const string16 lower_user_text(base::i18n::ToLower(log.text)); |
| 261 | 265 |
| 262 BeginTransaction(); | |
| 263 // Traverse transitional matches for those that have a user_text that is a | 266 // Traverse transitional matches for those that have a user_text that is a |
| 264 // prefix of |lower_user_text|. | 267 // prefix of |lower_user_text|. |
| 268 std::vector<AutocompleteActionPredictorTable::Row> rows_to_add; |
| 269 std::vector<AutocompleteActionPredictorTable::Row> rows_to_update; |
| 270 |
| 265 for (std::vector<TransitionalMatch>::const_iterator it = | 271 for (std::vector<TransitionalMatch>::const_iterator it = |
| 266 transitional_matches_.begin(); it != transitional_matches_.end(); | 272 transitional_matches_.begin(); it != transitional_matches_.end(); |
| 267 ++it) { | 273 ++it) { |
| 268 if (!StartsWith(lower_user_text, it->user_text, true)) | 274 if (!StartsWith(lower_user_text, it->user_text, true)) |
| 269 continue; | 275 continue; |
| 270 | 276 |
| 271 // Add entries to the database for those matches. | 277 // Add entries to the database for those matches. |
| 272 for (std::vector<GURL>::const_iterator url_it = it->urls.begin(); | 278 for (std::vector<GURL>::const_iterator url_it = it->urls.begin(); |
| 273 url_it != it->urls.end(); ++url_it) { | 279 url_it != it->urls.end(); ++url_it) { |
| 274 DCHECK(it->user_text.length() >= kMinimumUserTextLength); | 280 DCHECK(it->user_text.length() >= kMinimumUserTextLength); |
| 275 const DBCacheKey key = { it->user_text, *url_it }; | 281 const DBCacheKey key = { it->user_text, *url_it }; |
| 276 const bool is_hit = (*url_it == opened_url); | 282 const bool is_hit = (*url_it == opened_url); |
| 277 | 283 |
| 278 NetworkActionPredictorDatabase::Row row; | 284 AutocompleteActionPredictorTable::Row row; |
| 279 row.user_text = key.user_text; | 285 row.user_text = key.user_text; |
| 280 row.url = key.url; | 286 row.url = key.url; |
| 281 | 287 |
| 282 DBCacheMap::iterator it = db_cache_.find(key); | 288 DBCacheMap::iterator it = db_cache_.find(key); |
| 283 if (it == db_cache_.end()) { | 289 if (it == db_cache_.end()) { |
| 284 row.id = guid::GenerateGUID(); | 290 row.id = guid::GenerateGUID(); |
| 285 row.number_of_hits = is_hit ? 1 : 0; | 291 row.number_of_hits = is_hit ? 1 : 0; |
| 286 row.number_of_misses = is_hit ? 0 : 1; | 292 row.number_of_misses = is_hit ? 0 : 1; |
| 287 | 293 |
| 288 AddRow(key, row); | 294 rows_to_add.push_back(row); |
| 289 } else { | 295 } else { |
| 290 DCHECK(db_id_cache_.find(key) != db_id_cache_.end()); | 296 DCHECK(db_id_cache_.find(key) != db_id_cache_.end()); |
| 291 row.id = db_id_cache_.find(key)->second; | 297 row.id = db_id_cache_.find(key)->second; |
| 292 row.number_of_hits = it->second.number_of_hits + (is_hit ? 1 : 0); | 298 row.number_of_hits = it->second.number_of_hits + (is_hit ? 1 : 0); |
| 293 row.number_of_misses = it->second.number_of_misses + (is_hit ? 0 : 1); | 299 row.number_of_misses = it->second.number_of_misses + (is_hit ? 0 : 1); |
| 294 | 300 |
| 295 UpdateRow(it, row); | 301 rows_to_update.push_back(row); |
| 296 } | 302 } |
| 297 } | 303 } |
| 298 } | 304 } |
| 299 CommitTransaction(); | 305 if (rows_to_add.size() > 0 || rows_to_update.size() > 0) |
| 306 AddAndUpdateRows(rows_to_add, rows_to_update); |
| 300 | 307 |
| 301 ClearTransitionalMatches(); | 308 ClearTransitionalMatches(); |
| 302 | 309 |
| 303 // Check against tracked urls and log accuracy for the confidence we | 310 // Check against tracked urls and log accuracy for the confidence we |
| 304 // predicted. | 311 // predicted. |
| 305 for (std::vector<std::pair<GURL, double> >::const_iterator it = | 312 for (std::vector<std::pair<GURL, double> >::const_iterator it = |
| 306 tracked_urls_.begin(); it != tracked_urls_.end(); | 313 tracked_urls_.begin(); it != tracked_urls_.end(); |
| 307 ++it) { | 314 ++it) { |
| 308 if (opened_url == it->first) { | 315 if (opened_url == it->first) { |
| 309 UMA_HISTOGRAM_COUNTS_100("NetworkActionPredictor.AccurateCount", | 316 UMA_HISTOGRAM_COUNTS_100("AutocompleteActionPredictor.AccurateCount", |
| 310 it->second * 100); | 317 it->second * 100); |
| 311 } | 318 } |
| 312 } | 319 } |
| 313 tracked_urls_.clear(); | 320 tracked_urls_.clear(); |
| 314 } | 321 } |
| 315 | 322 |
| 316 void NetworkActionPredictor::DeleteOldIdsFromCaches( | 323 void AutocompleteActionPredictor::DeleteOldIdsFromCaches( |
| 317 history::URLDatabase* url_db, | 324 history::URLDatabase* url_db, |
| 318 std::vector<NetworkActionPredictorDatabase::Row::Id>* id_list) { | 325 std::vector<AutocompleteActionPredictorTable::Row::Id>* id_list) { |
| 319 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | 326 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); |
| 320 DCHECK(url_db); | 327 DCHECK(url_db); |
| 321 DCHECK(id_list); | 328 DCHECK(id_list); |
| 322 id_list->clear(); | 329 id_list->clear(); |
| 323 for (DBCacheMap::iterator it = db_cache_.begin(); it != db_cache_.end();) { | 330 for (DBCacheMap::iterator it = db_cache_.begin(); it != db_cache_.end();) { |
| 324 history::URLRow url_row; | 331 history::URLRow url_row; |
| 325 | 332 |
| 326 if ((url_db->GetRowForURL(it->first.url, &url_row) == 0) || | 333 if ((url_db->GetRowForURL(it->first.url, &url_row) == 0) || |
| 327 ((base::Time::Now() - url_row.last_visit()).InDays() > | 334 ((base::Time::Now() - url_row.last_visit()).InDays() > |
| 328 kMaximumDaysToKeepEntry)) { | 335 kMaximumDaysToKeepEntry)) { |
| 329 const DBIdCacheMap::iterator id_it = db_id_cache_.find(it->first); | 336 const DBIdCacheMap::iterator id_it = db_id_cache_.find(it->first); |
| 330 DCHECK(id_it != db_id_cache_.end()); | 337 DCHECK(id_it != db_id_cache_.end()); |
| 331 id_list->push_back(id_it->second); | 338 id_list->push_back(id_it->second); |
| 332 db_id_cache_.erase(id_it); | 339 db_id_cache_.erase(id_it); |
| 333 db_cache_.erase(it++); | 340 db_cache_.erase(it++); |
| 334 } else { | 341 } else { |
| 335 ++it; | 342 ++it; |
| 336 } | 343 } |
| 337 } | 344 } |
| 338 } | 345 } |
| 339 | 346 |
| 340 void NetworkActionPredictor::DeleteOldEntries(history::URLDatabase* url_db) { | 347 void AutocompleteActionPredictor::DeleteOldEntries( |
| 348 history::URLDatabase* url_db) { |
| 341 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | 349 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); |
| 342 DCHECK(!initialized_); | 350 DCHECK(!initialized_); |
| 343 | 351 |
| 344 std::vector<NetworkActionPredictorDatabase::Row::Id> ids_to_delete; | 352 std::vector<AutocompleteActionPredictorTable::Row::Id> ids_to_delete; |
| 345 DeleteOldIdsFromCaches(url_db, &ids_to_delete); | 353 DeleteOldIdsFromCaches(url_db, &ids_to_delete); |
| 346 | 354 |
| 347 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | 355 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, |
| 348 base::Bind(&NetworkActionPredictorDatabase::DeleteRows, db_, | 356 base::Bind(&AutocompleteActionPredictorTable::DeleteRows, |
| 357 db_, |
| 349 ids_to_delete)); | 358 ids_to_delete)); |
| 350 | 359 |
| 351 // Register for notifications and set the |initialized_| flag. | 360 // Register for notifications and set the |initialized_| flag. |
| 352 notification_registrar_.Add(this, chrome::NOTIFICATION_OMNIBOX_OPENED_URL, | 361 notification_registrar_.Add(this, chrome::NOTIFICATION_OMNIBOX_OPENED_URL, |
| 353 content::Source<Profile>(profile_)); | 362 content::Source<Profile>(profile_)); |
| 354 notification_registrar_.Add(this, chrome::NOTIFICATION_HISTORY_URLS_DELETED, | 363 notification_registrar_.Add(this, chrome::NOTIFICATION_HISTORY_URLS_DELETED, |
| 355 content::Source<Profile>(profile_)); | 364 content::Source<Profile>(profile_)); |
| 356 initialized_ = true; | 365 initialized_ = true; |
| 357 } | 366 } |
| 358 | 367 |
| 359 void NetworkActionPredictor::CreateCaches( | 368 void AutocompleteActionPredictor::CreateCaches( |
| 360 std::vector<NetworkActionPredictorDatabase::Row>* rows) { | 369 std::vector<AutocompleteActionPredictorTable::Row>* rows) { |
| 361 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | 370 CHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); |
| 362 DCHECK(!initialized_); | 371 DCHECK(!initialized_); |
| 363 DCHECK(db_cache_.empty()); | 372 DCHECK(db_cache_.empty()); |
| 364 DCHECK(db_id_cache_.empty()); | 373 DCHECK(db_id_cache_.empty()); |
| 365 | 374 |
| 366 for (std::vector<NetworkActionPredictorDatabase::Row>::const_iterator it = | 375 for (std::vector<AutocompleteActionPredictorTable::Row>::const_iterator it = |
| 367 rows->begin(); it != rows->end(); ++it) { | 376 rows->begin(); it != rows->end(); ++it) { |
| 368 const DBCacheKey key = { it->user_text, it->url }; | 377 const DBCacheKey key = { it->user_text, it->url }; |
| 369 const DBCacheValue value = { it->number_of_hits, it->number_of_misses }; | 378 const DBCacheValue value = { it->number_of_hits, it->number_of_misses }; |
| 370 db_cache_[key] = value; | 379 db_cache_[key] = value; |
| 371 db_id_cache_[key] = it->id; | 380 db_id_cache_[key] = it->id; |
| 372 } | 381 } |
| 373 | 382 |
| 374 // If the history service is ready, delete any old or invalid entries. | 383 // If the history service is ready, delete any old or invalid entries. |
| 375 HistoryService* history_service = | 384 HistoryService* history_service = |
| 376 profile_->GetHistoryService(Profile::EXPLICIT_ACCESS); | 385 profile_->GetHistoryService(Profile::EXPLICIT_ACCESS); |
| 377 if (!TryDeleteOldEntries(history_service)) { | 386 if (!TryDeleteOldEntries(history_service)) { |
| 378 // Wait for the notification that the history service is ready and the URL | 387 // Wait for the notification that the history service is ready and the URL |
| 379 // DB is loaded. | 388 // DB is loaded. |
| 380 notification_registrar_.Add(this, chrome::NOTIFICATION_HISTORY_LOADED, | 389 notification_registrar_.Add(this, chrome::NOTIFICATION_HISTORY_LOADED, |
| 381 content::Source<Profile>(profile_)); | 390 content::Source<Profile>(profile_)); |
| 382 } | 391 } |
| 383 } | 392 } |
| 384 | 393 |
| 385 bool NetworkActionPredictor::TryDeleteOldEntries(HistoryService* service) { | 394 bool AutocompleteActionPredictor::TryDeleteOldEntries(HistoryService* service) { |
| 386 if (!service) | 395 if (!service) |
| 387 return false; | 396 return false; |
| 388 | 397 |
| 389 history::URLDatabase* url_db = service->InMemoryDatabase(); | 398 history::URLDatabase* url_db = service->InMemoryDatabase(); |
| 390 if (!url_db) | 399 if (!url_db) |
| 391 return false; | 400 return false; |
| 392 | 401 |
| 393 DeleteOldEntries(url_db); | 402 DeleteOldEntries(url_db); |
| 394 return true; | 403 return true; |
| 395 } | 404 } |
| 396 | 405 |
| 397 double NetworkActionPredictor::CalculateConfidence( | 406 double AutocompleteActionPredictor::CalculateConfidence( |
| 398 const string16& user_text, | 407 const string16& user_text, |
| 399 const AutocompleteMatch& match, | 408 const AutocompleteMatch& match, |
| 400 bool* is_in_db) const { | 409 bool* is_in_db) const { |
| 401 const DBCacheKey key = { user_text, match.destination_url }; | 410 const DBCacheKey key = { user_text, match.destination_url }; |
| 402 | 411 |
| 403 *is_in_db = false; | 412 *is_in_db = false; |
| 404 if (user_text.length() < kMinimumUserTextLength) | 413 if (user_text.length() < kMinimumUserTextLength) |
| 405 return 0.0; | 414 return 0.0; |
| 406 | 415 |
| 407 const DBCacheMap::const_iterator iter = db_cache_.find(key); | 416 const DBCacheMap::const_iterator iter = db_cache_.find(key); |
| 408 if (iter == db_cache_.end()) | 417 if (iter == db_cache_.end()) |
| 409 return 0.0; | 418 return 0.0; |
| 410 | 419 |
| 411 *is_in_db = true; | 420 *is_in_db = true; |
| 412 return CalculateConfidenceForDbEntry(iter); | 421 return CalculateConfidenceForDbEntry(iter); |
| 413 } | 422 } |
| 414 | 423 |
| 415 double NetworkActionPredictor::CalculateConfidenceForDbEntry( | 424 double AutocompleteActionPredictor::CalculateConfidenceForDbEntry( |
| 416 DBCacheMap::const_iterator iter) const { | 425 DBCacheMap::const_iterator iter) const { |
| 417 const DBCacheValue& value = iter->second; | 426 const DBCacheValue& value = iter->second; |
| 418 if (value.number_of_hits < kMinimumNumberOfHits) | 427 if (value.number_of_hits < kMinimumNumberOfHits) |
| 419 return 0.0; | 428 return 0.0; |
| 420 | 429 |
| 421 const double number_of_hits = value.number_of_hits * hit_weight_; | 430 const double number_of_hits = value.number_of_hits * hit_weight_; |
| 422 return number_of_hits / (number_of_hits + value.number_of_misses); | 431 return number_of_hits / (number_of_hits + value.number_of_misses); |
| 423 } | 432 } |
| 424 | 433 |
| 425 void NetworkActionPredictor::AddRow( | 434 void AutocompleteActionPredictor::AddAndUpdateRows( |
| 426 const DBCacheKey& key, | 435 const AutocompleteActionPredictorTable::Rows& rows_to_add, |
| 427 const NetworkActionPredictorDatabase::Row& row) { | 436 const AutocompleteActionPredictorTable::Rows& rows_to_update) { |
| 428 if (!initialized_) | 437 if (!initialized_) |
| 429 return; | 438 return; |
| 430 | 439 |
| 431 DBCacheValue value = { row.number_of_hits, row.number_of_misses }; | |
| 432 db_cache_[key] = value; | |
| 433 db_id_cache_[key] = row.id; | |
| 434 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | 440 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, |
| 435 base::Bind(&NetworkActionPredictorDatabase::AddRow, db_, row)); | 441 base::Bind(&AutocompleteActionPredictorTable::AddAndUpdateRows, |
| 442 db_, |
| 443 rows_to_add, |
| 444 rows_to_update)); |
| 436 | 445 |
| 437 UMA_HISTOGRAM_ENUMERATION("NetworkActionPredictor.DatabaseAction", | 446 for (AutocompleteActionPredictorTable::Rows::const_iterator it = |
| 438 DATABASE_ACTION_ADD, DATABASE_ACTION_COUNT); | 447 rows_to_add.begin(); it != rows_to_add.end(); ++it) { |
| 448 const DBCacheKey key = { it->user_text, it->url }; |
| 449 DBCacheValue value = { it->number_of_hits, it->number_of_misses }; |
| 450 |
| 451 DCHECK(db_cache_.find(key) == db_cache_.end()); |
| 452 |
| 453 db_cache_[key] = value; |
| 454 db_id_cache_[key] = it->id; |
| 455 UMA_HISTOGRAM_ENUMERATION("AutocompleteActionPredictor.DatabaseAction", |
| 456 DATABASE_ACTION_ADD, DATABASE_ACTION_COUNT); |
| 457 } |
| 458 for (AutocompleteActionPredictorTable::Rows::const_iterator it = |
| 459 rows_to_update.begin(); it != rows_to_update.end(); ++it) { |
| 460 const DBCacheKey key = { it->user_text, it->url }; |
| 461 |
| 462 DBCacheMap::iterator db_it = db_cache_.find(key); |
| 463 DCHECK(db_it != db_cache_.end()); |
| 464 DCHECK(db_id_cache_.find(key) != db_id_cache_.end()); |
| 465 |
| 466 db_it->second.number_of_hits = it->number_of_hits; |
| 467 db_it->second.number_of_misses = it->number_of_misses; |
| 468 UMA_HISTOGRAM_ENUMERATION("AutocompleteActionPredictor.DatabaseAction", |
| 469 DATABASE_ACTION_UPDATE, DATABASE_ACTION_COUNT); |
| 470 } |
| 439 } | 471 } |
| 440 | 472 |
| 441 void NetworkActionPredictor::UpdateRow( | 473 void AutocompleteActionPredictor::DeleteAllRows() { |
| 442 DBCacheMap::iterator it, | |
| 443 const NetworkActionPredictorDatabase::Row& row) { | |
| 444 if (!initialized_) | |
| 445 return; | |
| 446 | |
| 447 DCHECK(it != db_cache_.end()); | |
| 448 it->second.number_of_hits = row.number_of_hits; | |
| 449 it->second.number_of_misses = row.number_of_misses; | |
| 450 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | |
| 451 base::Bind(&NetworkActionPredictorDatabase::UpdateRow, db_, row)); | |
| 452 UMA_HISTOGRAM_ENUMERATION("NetworkActionPredictor.DatabaseAction", | |
| 453 DATABASE_ACTION_UPDATE, DATABASE_ACTION_COUNT); | |
| 454 } | |
| 455 | |
| 456 void NetworkActionPredictor::DeleteAllRows() { | |
| 457 if (!initialized_) | 474 if (!initialized_) |
| 458 return; | 475 return; |
| 459 | 476 |
| 460 db_cache_.clear(); | 477 db_cache_.clear(); |
| 461 db_id_cache_.clear(); | 478 db_id_cache_.clear(); |
| 462 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | 479 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, |
| 463 base::Bind(&NetworkActionPredictorDatabase::DeleteAllRows, db_)); | 480 base::Bind(&AutocompleteActionPredictorTable::DeleteAllRows, |
| 464 UMA_HISTOGRAM_ENUMERATION("NetworkActionPredictor.DatabaseAction", | 481 db_)); |
| 482 UMA_HISTOGRAM_ENUMERATION("AutocompleteActionPredictor.DatabaseAction", |
| 465 DATABASE_ACTION_DELETE_ALL, DATABASE_ACTION_COUNT); | 483 DATABASE_ACTION_DELETE_ALL, DATABASE_ACTION_COUNT); |
| 466 } | 484 } |
| 467 | 485 |
| 468 void NetworkActionPredictor::DeleteRowsWithURLs(const std::set<GURL>& urls) { | 486 void AutocompleteActionPredictor::DeleteRowsWithURLs( |
| 487 const std::set<GURL>& urls) { |
| 469 if (!initialized_) | 488 if (!initialized_) |
| 470 return; | 489 return; |
| 471 | 490 |
| 472 std::vector<NetworkActionPredictorDatabase::Row::Id> id_list; | 491 std::vector<AutocompleteActionPredictorTable::Row::Id> id_list; |
| 473 | 492 |
| 474 for (DBCacheMap::iterator it = db_cache_.begin(); it != db_cache_.end();) { | 493 for (DBCacheMap::iterator it = db_cache_.begin(); it != db_cache_.end();) { |
| 475 if (urls.find(it->first.url) != urls.end()) { | 494 if (urls.find(it->first.url) != urls.end()) { |
| 476 const DBIdCacheMap::iterator id_it = db_id_cache_.find(it->first); | 495 const DBIdCacheMap::iterator id_it = db_id_cache_.find(it->first); |
| 477 DCHECK(id_it != db_id_cache_.end()); | 496 DCHECK(id_it != db_id_cache_.end()); |
| 478 id_list.push_back(id_it->second); | 497 id_list.push_back(id_it->second); |
| 479 db_id_cache_.erase(id_it); | 498 db_id_cache_.erase(id_it); |
| 480 db_cache_.erase(it++); | 499 db_cache_.erase(it++); |
| 481 } else { | 500 } else { |
| 482 ++it; | 501 ++it; |
| 483 } | 502 } |
| 484 } | 503 } |
| 485 | 504 |
| 486 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | 505 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, |
| 487 base::Bind(&NetworkActionPredictorDatabase::DeleteRows, db_, id_list)); | 506 base::Bind(&AutocompleteActionPredictorTable::DeleteRows, |
| 488 UMA_HISTOGRAM_ENUMERATION("NetworkActionPredictor.DatabaseAction", | 507 db_, |
| 508 id_list)); |
| 509 UMA_HISTOGRAM_ENUMERATION("AutocompleteActionPredictor.DatabaseAction", |
| 489 DATABASE_ACTION_DELETE_SOME, DATABASE_ACTION_COUNT); | 510 DATABASE_ACTION_DELETE_SOME, DATABASE_ACTION_COUNT); |
| 490 } | 511 } |
| 491 | 512 |
| 492 void NetworkActionPredictor::BeginTransaction() { | 513 AutocompleteActionPredictor::TransitionalMatch::TransitionalMatch() { |
| 493 if (!initialized_) | |
| 494 return; | |
| 495 | |
| 496 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | |
| 497 base::Bind(&NetworkActionPredictorDatabase::BeginTransaction, db_)); | |
| 498 } | 514 } |
| 499 | 515 |
| 500 void NetworkActionPredictor::CommitTransaction() { | 516 AutocompleteActionPredictor::TransitionalMatch::~TransitionalMatch() { |
| 501 if (!initialized_) | |
| 502 return; | |
| 503 | |
| 504 content::BrowserThread::PostTask(content::BrowserThread::DB, FROM_HERE, | |
| 505 base::Bind(&NetworkActionPredictorDatabase::CommitTransaction, db_)); | |
| 506 } | 517 } |
| 507 | 518 |
| 508 NetworkActionPredictor::TransitionalMatch::TransitionalMatch() { | 519 } // namespace predictors |
| 509 } | |
| 510 | |
| 511 NetworkActionPredictor::TransitionalMatch::~TransitionalMatch() { | |
| 512 } | |
| OLD | NEW |