Chromium Code Reviews| Index: components/ntp_snippets/content_suggestions_service.cc |
| diff --git a/components/ntp_snippets/content_suggestions_service.cc b/components/ntp_snippets/content_suggestions_service.cc |
| index f52a9809d09dcae42dbad4cd9a62ff9598870423..b689990b54198c6ae380ac04e6fbb0b1f707fa39 100644 |
| --- a/components/ntp_snippets/content_suggestions_service.cc |
| +++ b/components/ntp_snippets/content_suggestions_service.cc |
| @@ -150,11 +150,19 @@ void ContentSuggestionsService::DismissCategory(Category category) { |
| if (providers_it == providers_by_category_.end()) |
| return; |
| + dismissed_providers_by_category_[providers_it->first] = providers_it->second; |
| providers_by_category_.erase(providers_it); |
| categories_.erase( |
| std::find(categories_.begin(), categories_.end(), category)); |
| } |
| +void ContentSuggestionsService::RestoreDismissedCategories() { |
| + for (const auto& category_provider_pair : dismissed_providers_by_category_) |
|
Marc Treib
2016/10/11 07:35:58
Braces if the body doesn't fit on one line.
Michael van Ouwerkerk
2016/10/11 12:04:16
Done.
|
| + RegisterCategoryIfRequired(category_provider_pair.second, |
| + category_provider_pair.first); |
| + dismissed_providers_by_category_.clear(); |
|
Marc Treib
2016/10/11 07:35:58
Should RegisterCategoryIfRequired remove the entry
Michael van Ouwerkerk
2016/10/11 12:04:16
Done.
|
| +} |
| + |
| void ContentSuggestionsService::AddObserver(Observer* observer) { |
| observers_.AddObserver(observer); |
| } |