Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(152)

Side by Side Diff: components/ntp_snippets/remote/remote_suggestions_provider.cc

Issue 2568033005: [NTP::SectionOrder] Replace CategoryFactory with a category ranker. (Closed)
Patch Set: download provider tests. Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 1 // Copyright 2015 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 "components/ntp_snippets/remote/remote_suggestions_provider.h" 5 #include "components/ntp_snippets/remote/remote_suggestions_provider.h"
6 6
7 #include <algorithm> 7 #include <algorithm>
8 #include <iterator> 8 #include <iterator>
9 #include <utility> 9 #include <utility>
10 10
11 #include "base/command_line.h" 11 #include "base/command_line.h"
12 #include "base/feature_list.h" 12 #include "base/feature_list.h"
13 #include "base/location.h" 13 #include "base/location.h"
14 #include "base/memory/ptr_util.h" 14 #include "base/memory/ptr_util.h"
15 #include "base/metrics/histogram_macros.h" 15 #include "base/metrics/histogram_macros.h"
16 #include "base/metrics/sparse_histogram.h" 16 #include "base/metrics/sparse_histogram.h"
17 #include "base/path_service.h" 17 #include "base/path_service.h"
18 #include "base/stl_util.h" 18 #include "base/stl_util.h"
19 #include "base/strings/string_number_conversions.h" 19 #include "base/strings/string_number_conversions.h"
20 #include "base/strings/utf_string_conversions.h" 20 #include "base/strings/utf_string_conversions.h"
21 #include "base/task_runner_util.h" 21 #include "base/task_runner_util.h"
22 #include "base/time/default_clock.h" 22 #include "base/time/default_clock.h"
23 #include "base/time/default_tick_clock.h" 23 #include "base/time/default_tick_clock.h"
24 #include "base/time/time.h" 24 #include "base/time/time.h"
25 #include "base/values.h" 25 #include "base/values.h"
26 #include "components/data_use_measurement/core/data_use_user_data.h" 26 #include "components/data_use_measurement/core/data_use_user_data.h"
27 #include "components/history/core/browser/history_service.h" 27 #include "components/history/core/browser/history_service.h"
28 #include "components/image_fetcher/image_decoder.h" 28 #include "components/image_fetcher/image_decoder.h"
29 #include "components/image_fetcher/image_fetcher.h" 29 #include "components/image_fetcher/image_fetcher.h"
30 #include "components/ntp_snippets/category_rankers/category_ranker.h"
30 #include "components/ntp_snippets/features.h" 31 #include "components/ntp_snippets/features.h"
31 #include "components/ntp_snippets/pref_names.h" 32 #include "components/ntp_snippets/pref_names.h"
32 #include "components/ntp_snippets/remote/remote_suggestions_database.h" 33 #include "components/ntp_snippets/remote/remote_suggestions_database.h"
33 #include "components/ntp_snippets/switches.h" 34 #include "components/ntp_snippets/switches.h"
34 #include "components/ntp_snippets/user_classifier.h" 35 #include "components/ntp_snippets/user_classifier.h"
35 #include "components/prefs/pref_registry_simple.h" 36 #include "components/prefs/pref_registry_simple.h"
36 #include "components/prefs/pref_service.h" 37 #include "components/prefs/pref_service.h"
37 #include "components/variations/variations_associated_data.h" 38 #include "components/variations/variations_associated_data.h"
38 #include "grit/components_strings.h" 39 #include "grit/components_strings.h"
39 #include "ui/base/l10n/l10n_util.h" 40 #include "ui/base/l10n/l10n_util.h"
(...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after
217 if (callback.is_null()) { 218 if (callback.is_null()) {
218 return; 219 return;
219 } 220 }
220 callback.Run(status, std::vector<ContentSuggestion>()); 221 callback.Run(status, std::vector<ContentSuggestion>());
221 } 222 }
222 223
223 } // namespace 224 } // namespace
224 225
225 RemoteSuggestionsProvider::RemoteSuggestionsProvider( 226 RemoteSuggestionsProvider::RemoteSuggestionsProvider(
226 Observer* observer, 227 Observer* observer,
227 CategoryFactory* category_factory,
228 PrefService* pref_service, 228 PrefService* pref_service,
229 const std::string& application_language_code, 229 const std::string& application_language_code,
230 CategoryRanker* category_ranker,
230 const UserClassifier* user_classifier, 231 const UserClassifier* user_classifier,
231 NTPSnippetsScheduler* scheduler, 232 NTPSnippetsScheduler* scheduler,
232 std::unique_ptr<NTPSnippetsFetcher> snippets_fetcher, 233 std::unique_ptr<NTPSnippetsFetcher> snippets_fetcher,
233 std::unique_ptr<image_fetcher::ImageFetcher> image_fetcher, 234 std::unique_ptr<image_fetcher::ImageFetcher> image_fetcher,
234 std::unique_ptr<image_fetcher::ImageDecoder> image_decoder, 235 std::unique_ptr<image_fetcher::ImageDecoder> image_decoder,
235 std::unique_ptr<RemoteSuggestionsDatabase> database, 236 std::unique_ptr<RemoteSuggestionsDatabase> database,
236 std::unique_ptr<RemoteSuggestionsStatusService> status_service) 237 std::unique_ptr<RemoteSuggestionsStatusService> status_service)
237 : ContentSuggestionsProvider(observer, category_factory), 238 : ContentSuggestionsProvider(observer),
238 state_(State::NOT_INITED), 239 state_(State::NOT_INITED),
239 pref_service_(pref_service), 240 pref_service_(pref_service),
240 articles_category_( 241 articles_category_(
241 category_factory->FromKnownCategory(KnownCategories::ARTICLES)), 242 Category::FromKnownCategory(KnownCategories::ARTICLES)),
242 application_language_code_(application_language_code), 243 application_language_code_(application_language_code),
244 category_ranker_(category_ranker),
243 user_classifier_(user_classifier), 245 user_classifier_(user_classifier),
244 scheduler_(scheduler), 246 scheduler_(scheduler),
245 snippets_fetcher_(std::move(snippets_fetcher)), 247 snippets_fetcher_(std::move(snippets_fetcher)),
246 image_fetcher_(std::move(image_fetcher)), 248 image_fetcher_(std::move(image_fetcher)),
247 image_decoder_(std::move(image_decoder)), 249 image_decoder_(std::move(image_decoder)),
248 database_(std::move(database)), 250 database_(std::move(database)),
249 status_service_(std::move(status_service)), 251 status_service_(std::move(status_service)),
250 fetch_when_ready_(false), 252 fetch_when_ready_(false),
251 nuke_when_initialized_(false), 253 nuke_when_initialized_(false),
252 thumbnail_requests_throttler_( 254 thumbnail_requests_throttler_(
(...skipping 338 matching lines...) Expand 10 before | Expand all | Expand 10 after
591 DCHECK(base::ContainsKey(category_contents_, articles_category_)); 593 DCHECK(base::ContainsKey(category_contents_, articles_category_));
592 594
593 base::TimeDelta database_load_time = 595 base::TimeDelta database_load_time =
594 base::TimeTicks::Now() - database_load_start_; 596 base::TimeTicks::Now() - database_load_start_;
595 UMA_HISTOGRAM_MEDIUM_TIMES("NewTabPage.Snippets.DatabaseLoadTime", 597 UMA_HISTOGRAM_MEDIUM_TIMES("NewTabPage.Snippets.DatabaseLoadTime",
596 database_load_time); 598 database_load_time);
597 599
598 NTPSnippet::PtrVector to_delete; 600 NTPSnippet::PtrVector to_delete;
599 for (std::unique_ptr<NTPSnippet>& snippet : snippets) { 601 for (std::unique_ptr<NTPSnippet>& snippet : snippets) {
600 Category snippet_category = 602 Category snippet_category =
601 category_factory()->FromRemoteCategory(snippet->remote_category_id()); 603 Category::FromRemoteCategory(snippet->remote_category_id());
604 // The category is added to the ranker to enforce the order among remote
605 // categories.
606 category_ranker_->AppendCategoryIfNecessary(snippet_category);
tschumann 2016/12/15 11:50:26 in google3, a common idiom is to call such methods
vitaliii 2016/12/15 15:30:13 I do not like "Maybe", because then it is not clea
tschumann 2016/12/15 18:23:13 For now it's fine. I'm not sure I agree with the "
vitaliii 2016/12/16 08:15:43 Ack.
602 auto content_it = category_contents_.find(snippet_category); 607 auto content_it = category_contents_.find(snippet_category);
603 // We should already know about the category. 608 // We should already know about the category.
604 if (content_it == category_contents_.end()) { 609 if (content_it == category_contents_.end()) {
605 DLOG(WARNING) << "Loaded a suggestion for unknown category " 610 DLOG(WARNING) << "Loaded a suggestion for unknown category "
606 << snippet_category << " from the DB; deleting"; 611 << snippet_category << " from the DB; deleting";
607 to_delete.emplace_back(std::move(snippet)); 612 to_delete.emplace_back(std::move(snippet));
608 continue; 613 continue;
609 } 614 }
610 CategoryContent* content = &content_it->second; 615 CategoryContent* content = &content_it->second;
611 if (snippet->is_dismissed()) { 616 if (snippet->is_dismissed()) {
(...skipping 129 matching lines...) Expand 10 before | Expand all | Expand 10 after
741 for (NTPSnippetsFetcher::FetchedCategory& fetched_category : 746 for (NTPSnippetsFetcher::FetchedCategory& fetched_category :
742 *fetched_categories) { 747 *fetched_categories) {
743 // TODO(tschumann): Remove this histogram once we only talk to the content 748 // TODO(tschumann): Remove this histogram once we only talk to the content
744 // suggestions cloud backend. 749 // suggestions cloud backend.
745 if (fetched_category.category == articles_category_) { 750 if (fetched_category.category == articles_category_) {
746 UMA_HISTOGRAM_SPARSE_SLOWLY( 751 UMA_HISTOGRAM_SPARSE_SLOWLY(
747 "NewTabPage.Snippets.NumArticlesFetched", 752 "NewTabPage.Snippets.NumArticlesFetched",
748 std::min(fetched_category.snippets.size(), 753 std::min(fetched_category.snippets.size(),
749 static_cast<size_t>(kMaxSnippetCount + 1))); 754 static_cast<size_t>(kMaxSnippetCount + 1)));
750 } 755 }
756 category_ranker_->AppendCategoryIfNecessary(fetched_category.category);
751 CategoryContent* content = 757 CategoryContent* content =
752 UpdateCategoryInfo(fetched_category.category, fetched_category.info); 758 UpdateCategoryInfo(fetched_category.category, fetched_category.info);
753 content->included_in_last_server_response = true; 759 content->included_in_last_server_response = true;
754 SanitizeReceivedSnippets(content->dismissed, &fetched_category.snippets); 760 SanitizeReceivedSnippets(content->dismissed, &fetched_category.snippets);
755 IntegrateSnippets(content, std::move(fetched_category.snippets)); 761 IntegrateSnippets(content, std::move(fetched_category.snippets));
756 } 762 }
757 } 763 }
758 764
759 // TODO(tschumann): The snippets fetcher needs to signal errors so that we 765 // TODO(tschumann): The snippets fetcher needs to signal errors so that we
760 // know why we received no data. If an error occured, none of the following 766 // know why we received no data. If an error occured, none of the following
(...skipping 524 matching lines...) Expand 10 before | Expand all | Expand 10 after
1285 &included_in_last_server_response)) { 1291 &included_in_last_server_response)) {
1286 DLOG(WARNING) << "Invalid category pref value, missing '" 1292 DLOG(WARNING) << "Invalid category pref value, missing '"
1287 << kCategoryContentProvidedByServer << "': " << *entry; 1293 << kCategoryContentProvidedByServer << "': " << *entry;
1288 continue; 1294 continue;
1289 } 1295 }
1290 bool allow_fetching_more_results = false; 1296 bool allow_fetching_more_results = false;
1291 // This wasn't always around, so it's okay if it's missing. 1297 // This wasn't always around, so it's okay if it's missing.
1292 dict->GetBoolean(kCategoryContentAllowFetchingMore, 1298 dict->GetBoolean(kCategoryContentAllowFetchingMore,
1293 &allow_fetching_more_results); 1299 &allow_fetching_more_results);
1294 1300
1295 Category category = category_factory()->FromIDValue(id); 1301 Category category = Category::FromIDValue(id);
1302 // The ranker may not persist the order of remote categories.
1303 category_ranker_->AppendCategoryIfNecessary(category);
1296 // TODO(tschumann): The following has a bad smell that category 1304 // TODO(tschumann): The following has a bad smell that category
1297 // serialization / deserialization should not be done inside this 1305 // serialization / deserialization should not be done inside this
1298 // class. We should move that into a central place that also knows how to 1306 // class. We should move that into a central place that also knows how to
1299 // parse data we received from remote backends. 1307 // parse data we received from remote backends.
1300 CategoryInfo info = 1308 CategoryInfo info =
1301 category == articles_category_ 1309 category == articles_category_
1302 ? BuildArticleCategoryInfo(title) 1310 ? BuildArticleCategoryInfo(title)
1303 : BuildRemoteCategoryInfo(title, allow_fetching_more_results); 1311 : BuildRemoteCategoryInfo(title, allow_fetching_more_results);
1304 CategoryContent* content = UpdateCategoryInfo(category, info); 1312 CategoryContent* content = UpdateCategoryInfo(category, info);
1305 content->included_in_last_server_response = 1313 content->included_in_last_server_response =
1306 included_in_last_server_response; 1314 included_in_last_server_response;
1307 } 1315 }
1308 } 1316 }
1309 1317
1310 void RemoteSuggestionsProvider::StoreCategoriesToPrefs() { 1318 void RemoteSuggestionsProvider::StoreCategoriesToPrefs() {
1311 // Collect all the CategoryContents. 1319 // Collect all the CategoryContents.
1312 std::vector<std::pair<Category, const CategoryContent*>> to_store; 1320 std::vector<std::pair<Category, const CategoryContent*>> to_store;
1313 for (const auto& entry : category_contents_) { 1321 for (const auto& entry : category_contents_) {
1314 to_store.emplace_back(entry.first, &entry.second); 1322 to_store.emplace_back(entry.first, &entry.second);
1315 } 1323 }
1316 // Sort them into the proper category order. 1324 // The ranker may not persist the order, thus, it is stored by the provider.
1317 std::sort(to_store.begin(), to_store.end(), 1325 std::sort(to_store.begin(), to_store.end(),
1318 [this](const std::pair<Category, const CategoryContent*>& left, 1326 [this](const std::pair<Category, const CategoryContent*>& left,
1319 const std::pair<Category, const CategoryContent*>& right) { 1327 const std::pair<Category, const CategoryContent*>& right) {
1320 return category_factory()->CompareCategories(left.first, 1328 return category_ranker_->Compare(left.first, right.first);
1321 right.first);
1322 }); 1329 });
1323 // Convert the relevant info into a base::ListValue for storage. 1330 // Convert the relevant info into a base::ListValue for storage.
1324 base::ListValue list; 1331 base::ListValue list;
1325 for (const auto& entry : to_store) { 1332 for (const auto& entry : to_store) {
1326 const Category& category = entry.first; 1333 const Category& category = entry.first;
1327 const CategoryContent& content = *entry.second; 1334 const CategoryContent& content = *entry.second;
1328 auto dict = base::MakeUnique<base::DictionaryValue>(); 1335 auto dict = base::MakeUnique<base::DictionaryValue>();
1329 dict->SetInteger(kCategoryContentId, category.id()); 1336 dict->SetInteger(kCategoryContentId, category.id());
1330 // TODO(tschumann): Persist other properties of the CategoryInfo. 1337 // TODO(tschumann): Persist other properties of the CategoryInfo.
1331 dict->SetString(kCategoryContentTitle, content.info.title()); 1338 dict->SetString(kCategoryContentTitle, content.info.title());
(...skipping 14 matching lines...) Expand all
1346 RemoteSuggestionsProvider::CategoryContent::CategoryContent(CategoryContent&&) = 1353 RemoteSuggestionsProvider::CategoryContent::CategoryContent(CategoryContent&&) =
1347 default; 1354 default;
1348 1355
1349 RemoteSuggestionsProvider::CategoryContent::~CategoryContent() = default; 1356 RemoteSuggestionsProvider::CategoryContent::~CategoryContent() = default;
1350 1357
1351 RemoteSuggestionsProvider::CategoryContent& 1358 RemoteSuggestionsProvider::CategoryContent&
1352 RemoteSuggestionsProvider::CategoryContent::operator=(CategoryContent&&) = 1359 RemoteSuggestionsProvider::CategoryContent::operator=(CategoryContent&&) =
1353 default; 1360 default;
1354 1361
1355 } // namespace ntp_snippets 1362 } // namespace ntp_snippets
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698