Index: components/offline_pages/core/prefetch/prefetch_service_impl.cc |
diff --git a/components/offline_pages/core/prefetch/prefetch_service_impl.cc b/components/offline_pages/core/prefetch/prefetch_service_impl.cc |
index 4f557e5e6b4d4b76a96c6bcdbd9333a29f229a9f..4206c446186b3212339259f7d7fd76c8823172ca 100644 |
--- a/components/offline_pages/core/prefetch/prefetch_service_impl.cc |
+++ b/components/offline_pages/core/prefetch/prefetch_service_impl.cc |
@@ -9,34 +9,44 @@ |
#include "base/memory/ptr_util.h" |
#include "components/offline_pages/core/prefetch/offline_metrics_collector.h" |
#include "components/offline_pages/core/prefetch/prefetch_dispatcher_impl.h" |
+#include "components/offline_pages/core/prefetch/prefetch_store.h" |
namespace offline_pages { |
PrefetchServiceImpl::PrefetchServiceImpl( |
+ std::unique_ptr<OfflineMetricsCollector> offline_metrics_collector, |
+ std::unique_ptr<PrefetchDispatcher> dispatcher, |
std::unique_ptr<PrefetchGCMHandler> gcm_handler, |
- std::unique_ptr<OfflineMetricsCollector> offline_metrics_collector) |
- : gcm_handler_(std::move(gcm_handler)), |
- dispatcher_(base::MakeUnique<PrefetchDispatcherImpl>()), |
- offline_metrics_collector_(std::move(offline_metrics_collector)) {} |
+ std::unique_ptr<PrefetchStore> store) |
+ : offline_metrics_collector_(std::move(offline_metrics_collector)), |
+ prefetch_dispatcher_(std::move(dispatcher)), |
+ prefetch_gcm_handler_(std::move(gcm_handler)), |
+ prefetch_store_(std::move(store)) { |
+ prefetch_dispatcher_->SetService(this); |
+} |
PrefetchServiceImpl::~PrefetchServiceImpl() = default; |
-void PrefetchServiceImpl::ObserveContentSuggestionsService( |
- ntp_snippets::ContentSuggestionsService* service) { |
- suggested_articles_observer_ = |
- base::MakeUnique<SuggestedArticlesObserver>(service, this); |
+OfflineMetricsCollector* PrefetchServiceImpl::GetOfflineMetricsCollector() { |
+ return offline_metrics_collector_.get(); |
} |
-PrefetchDispatcher* PrefetchServiceImpl::GetDispatcher() { |
- return dispatcher_.get(); |
+PrefetchDispatcher* PrefetchServiceImpl::GetPrefetchDispatcher() { |
+ return prefetch_dispatcher_.get(); |
} |
-OfflineMetricsCollector* PrefetchServiceImpl::GetOfflineMetricsCollector() { |
- return offline_metrics_collector_.get(); |
+PrefetchGCMHandler* PrefetchServiceImpl::GetPrefetchGCMHandler() { |
+ return prefetch_gcm_handler_.get(); |
} |
-PrefetchGCMHandler* PrefetchServiceImpl::GetPrefetchGCMHandler() { |
- return gcm_handler_.get(); |
+PrefetchStore* PrefetchServiceImpl::GetPrefetchStore() { |
+ return prefetch_store_.get(); |
+} |
+ |
+void PrefetchServiceImpl::ObserveContentSuggestionsService( |
+ ntp_snippets::ContentSuggestionsService* service) { |
+ suggested_articles_observer_ = |
+ base::MakeUnique<SuggestedArticlesObserver>(service, this); |
} |
void PrefetchServiceImpl::Shutdown() {} |