[TooManyTabs] Add TabNavigationThrottle

chrome/browser/resource_coordinator/tab_manager.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/resource_coordinator/tab_manager.h"
 
 #include <stddef.h>
 
 #include <algorithm>
 #include <set>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
-#include "base/command_line.h"
 #include "base/feature_list.h"
 #include "base/memory/memory_pressure_monitor.h"
 #include "base/metrics/field_trial.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/observer_list.h"
 #include "base/process/process.h"
 #include "base/rand_util.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
@@ skipping 10 matching lines @@
 #include "chrome/browser/resource_coordinator/tab_manager_web_contents_data.h"
 #include "chrome/browser/sessions/session_restore.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_iterator.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/tabs/tab_utils.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_features.h"
-#include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "components/metrics/system_memory_stats_recorder.h"
 #include "components/variations/variations_associated_data.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_controller.h"
+#include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_features.h"
 #include "content/public/common/page_importance_signals.h"
 
 #if defined(OS_CHROMEOS)
 #include "ash/multi_profile_uma.h"
 #include "ash/shell.h"
 #include "chrome/browser/resource_coordinator/tab_manager_delegate_chromeos.h"
 #include "components/user_manager/user_manager.h"
@@ skipping 773 matching lines @@
   }
 
   // An active tab is not purged.
   GetWebContentsData(new_contents)->set_is_purged(false);
 
   // Reload |web_contents| if it is in an active browser and discarded.
   if (IsActiveWebContentsInActiveBrowser(new_contents)) {
     ReloadWebContentsIfDiscarded(new_contents,
                                  GetWebContentsData(new_contents));
   }
+
+  ResumeTabNavigationIfNeeded(new_contents);
 }
 
 void TabManager::TabInsertedAt(TabStripModel* tab_strip_model,
                                content::WebContents* contents,
                                int index,
                                bool foreground) {
   // Only interested in background tabs, as foreground tabs get taken care of by
   // ActiveTabChanged.
   if (foreground)
     return;
@@ skipping 115 matching lines @@
 }
 
 void TabManager::RecordSwitchToTab(content::WebContents* contents) const {
   if (is_session_restore_loading_tabs_) {
     UMA_HISTOGRAM_ENUMERATION("TabManager.SessionRestore.SwitchToTab",
                               GetWebContentsData(contents)->tab_loading_state(),
                               TAB_LOADING_STATE_MAX);
   }
 }
 
+content::NavigationThrottle::ThrottleCheckResult
+TabManager::MaybeThrottleNavigation(
+    content::NavigationHandle* navigation_handle) {
+  if (!ShouldDelayNavigation(navigation_handle)) {
+    loading_contents_.insert(navigation_handle->GetWebContents());
+    return content::NavigationThrottle::PROCEED;
+  }
+
+  // TODO(zhenw): Try to set the favicon and title from history service if this
+  // navigation will be delayed.
+  GetWebContentsData(navigation_handle->GetWebContents())
+      ->SetTabLoadingState(TAB_IS_NOT_LOADING);
+  pending_navigations_.push_back(navigation_handle);
+
+  return content::NavigationThrottle::DEFER;
+}
+
+bool TabManager::ShouldDelayNavigation(
+    content::NavigationHandle* navigation_handle) const {
+  // Do not delay the navigation if no tab is currently loading.
+  if (loading_contents_.empty())
+    return false;
+
+  return true;
+}
+
+void TabManager::OnDidFinishNavigation(
+    content::NavigationHandle* navigation_handle) {
+  auto it = pending_navigations_.begin();
+  while (it != pending_navigations_.end()) {
+    content::NavigationHandle* pending_handle = *it;
+    if (pending_handle == navigation_handle) {
+      pending_navigations_.erase(it);
+      break;
+    }
+    it++;
+  }
+}
+
+void TabManager::OnDidStopLoading(content::WebContents* contents) {
+  DCHECK_EQ(TAB_IS_LOADED, GetWebContentsData(contents)->tab_loading_state());
+  loading_contents_.erase(contents);
+  LoadNextBackgroundTabIfNeeded();
+}
+
+void TabManager::OnWebContentsDestroyed(content::WebContents* contents) {
+  RemovePendingNavigationIfNeeded(contents);
+  loading_contents_.erase(contents);
+  LoadNextBackgroundTabIfNeeded();
+}
+
+void TabManager::LoadNextBackgroundTabIfNeeded() {
+  // Do not load more background tabs until all currently loading tabs have
+  // finished.
+  if (!loading_contents_.empty())
+    return;
+
+  if (pending_navigations_.empty())
+    return;
+
+  content::NavigationHandle* navigation_handle = pending_navigations_.front();
+  pending_navigations_.erase(pending_navigations_.begin());
+  ResumeNavigation(navigation_handle);
+}
+
+void TabManager::ResumeTabNavigationIfNeeded(content::WebContents* contents) {
+  content::NavigationHandle* navigation_handle =
+      RemovePendingNavigationIfNeeded(contents);
+  if (navigation_handle)
+    ResumeNavigation(navigation_handle);
+}
+
+void TabManager::ResumeNavigation(
+    content::NavigationHandle* navigation_handle) {
+  GetWebContentsData(navigation_handle->GetWebContents())
+      ->SetTabLoadingState(TAB_IS_LOADING);
+  loading_contents_.insert(navigation_handle->GetWebContents());
+
+  navigation_handle->Resume();
+}
+
+content::NavigationHandle* TabManager::RemovePendingNavigationIfNeeded(
+    content::WebContents* contents) {
+  content::NavigationHandle* navigation_handle = nullptr;
+  auto it = pending_navigations_.begin();
+  while (it != pending_navigations_.end()) {
+    navigation_handle = *it;
+    if ((*it)->GetWebContents() == contents) {
+      navigation_handle = *it;
+      pending_navigations_.erase(it);
+      break;
+    }
+    it++;
+  }
+  return navigation_handle;
+}
+
+bool TabManager::IsTabLoadingForTest(content::WebContents* contents) const {
+  if (loading_contents_.count(contents) == 1) {
+    DCHECK_EQ(TAB_IS_LOADING,
+              GetWebContentsData(contents)->tab_loading_state());
+    return true;
+  }
+
+  DCHECK_NE(TAB_IS_LOADING, GetWebContentsData(contents)->tab_loading_state());
+  return false;
+}
+
+bool TabManager::IsNavigationDelayedForTest(
+    const content::NavigationHandle* navigation_handle) const {
+  for (const auto* nav : pending_navigations_) {
+    if (nav == navigation_handle)
+      return true;
+  }
+  return false;
+}
+
 }  // namespace resource_coordinator