DataReductionProxyStatisticsPrefs should support WeakPtr

components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc

Index: components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc
diff --git a/components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc b/components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc
index 71d206e262ac18329d6267a954c12ac4d94040d7..cbacece8ab67965b22aad35370693f3e426ce6ad 100644
--- a/components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc
+++ b/components/data_reduction_proxy/core/browser/data_reduction_proxy_io_data.cc
@@ -25,13 +25,11 @@ namespace data_reduction_proxy {
 
 DataReductionProxyIOData::DataReductionProxyIOData(
     const Client& client,
-    scoped_ptr<DataReductionProxyStatisticsPrefs> statistics_prefs,
     DataReductionProxySettings* settings,
     net::NetLog* net_log,
     scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,
     scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner)
     : client_(client),
-      statistics_prefs_(statistics_prefs.Pass()),
       settings_(settings),
       net_log_(net_log),
       io_task_runner_(io_task_runner),
@@ -62,17 +60,18 @@ void DataReductionProxyIOData::InitOnUIThread(PrefService* pref_service) {
   enabled_.MoveToThread(io_task_runner_);
 }
 
-void  DataReductionProxyIOData::ShutdownOnUIThread() {
+void DataReductionProxyIOData::ShutdownOnUIThread() {
   DCHECK(!shutdown_on_ui_);
   DCHECK(ui_task_runner_->BelongsToCurrentThread());
-  if (statistics_prefs_) {
-    statistics_prefs_->WritePrefs();
-    statistics_prefs_->ShutdownOnUIThread();
-  }
   enabled_.Destroy();
   shutdown_on_ui_ = true;
 }
 
+void DataReductionProxyIOData::SetDataReductionProxyStatisticsPrefs(
+    base::WeakPtr<DataReductionProxyStatisticsPrefs> statistics_prefs) {
+  statistics_prefs_ = statistics_prefs;
+}
+
 bool DataReductionProxyIOData::IsEnabled() const {
   DCHECK(io_task_runner_->BelongsToCurrentThread());
   return enabled_.GetValue() ||
@@ -87,15 +86,6 @@ DataReductionProxyIOData::CreateInterceptor() {
       params_.get(), usage_stats_.get(), event_store_.get()));
 }
 
-void DataReductionProxyIOData::EnableCompressionStatisticsLogging(
-    PrefService* prefs,
-    const base::TimeDelta& commit_delay) {
-  DCHECK(ui_task_runner_->BelongsToCurrentThread());
-  statistics_prefs_.reset(
-      new DataReductionProxyStatisticsPrefs(
-          prefs, ui_task_runner_, commit_delay));
-}
-
 scoped_ptr<DataReductionProxyNetworkDelegate>
 DataReductionProxyIOData::CreateNetworkDelegate(
     scoped_ptr<net::NetworkDelegate> wrapped_network_delegate,
@@ -106,11 +96,10 @@ DataReductionProxyIOData::CreateNetworkDelegate(
           wrapped_network_delegate.Pass(), params_.get(),
           auth_request_handler_.get(), configurator_.get()));
   if (track_proxy_bypass_statistics && !usage_stats_) {
-    usage_stats_.reset(
-        new data_reduction_proxy::DataReductionProxyUsageStats(
+    usage_stats_.reset(new data_reduction_proxy::DataReductionProxyUsageStats(
         params_.get(), settings_, ui_task_runner_));
     network_delegate->InitStatisticsPrefsAndUMA(
-        ui_task_runner_, statistics_prefs_.get(), &enabled_,
+        ui_task_runner_, statistics_prefs_, &enabled_,
         usage_stats_.get());
   }
   return network_delegate.Pass();