Notify DataUseTabModel of navigations and tab closures

chrome/browser/profiles/profile_impl_io_data.cc

Index: chrome/browser/profiles/profile_impl_io_data.cc
diff --git a/chrome/browser/profiles/profile_impl_io_data.cc b/chrome/browser/profiles/profile_impl_io_data.cc
index 5638f8c84da657ec45fa5305830c673f2966ddaf..635d8b8e5d9500b1e9365e59eec88b871799f6e6 100644
--- a/chrome/browser/profiles/profile_impl_io_data.cc
+++ b/chrome/browser/profiles/profile_impl_io_data.cc
@@ -65,6 +65,13 @@
 #include "net/url_request/url_request_job_factory_impl.h"
 #include "storage/browser/quota/special_storage_policy.h"
 
+#if defined(OS_ANDROID)
+#include "chrome/browser/android/data_usage/data_use_tab_model.h"
+#include "chrome/browser/android/data_usage/data_use_ui_tab_model.h"
+#include "chrome/browser/android/data_usage/data_use_ui_tab_model_factory.h"
+#include "chrome/browser/android/data_usage/external_data_use_observer.h"
+#endif  // defined(OS_ANDROID)
+
 namespace {
 
 net::BackendType ChooseCacheBackendType() {
@@ -226,6 +233,19 @@ ProfileImplIOData::Handle::CreateMainRequestContextGetter(
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   LazyInitialize();
   DCHECK(!main_request_context_getter_.get());
+
+#if defined(OS_ANDROID)

[mmenke, 2015/11/13 22:49:55]

I don't know what this stuff is for, but I assume you only want to do it for
non-incognito profiles?

[bengr, 2015/11/14 03:54:39]

Yes, should only be for non-incognito.

[tbansal1, 2015/11/16 17:49:08]

Done.

[tbansal1, 2015/11/16 17:49:08]

Done.

+  chrome::android::DataUseUITabModel* data_use_ui_tab_model =
+      chrome::android::DataUseUITabModelFactory::GetForBrowserContext(
+          Profile::FromBrowserContext(profile_));

[mmenke, 2015/11/13 22:49:55]

Remove the "Profile::FromBrowserContext"?

[tbansal1, 2015/11/16 17:49:08]

Done.

+  if (data_use_ui_tab_model) {
+    data_use_ui_tab_model->set_data_use_tab_model(
+        io_thread->external_data_use_observer()
+            ->data_use_tab_model()
+            ->GetWeakPtr());
+  }
+#endif  // defined(OS_ANDROID)
+
   main_request_context_getter_ = ChromeURLRequestContextGetter::Create(
       profile_, io_data_, protocol_handlers, request_interceptors.Pass());