page_load_metrics_initialize.cc 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "weblayer/browser/page_load_metrics_initialize.h"
  5. #include <memory>
  6. #include <utility>
  7. #include "base/bind.h"
  8. #include "components/page_load_metrics/browser/metrics_web_contents_observer.h"
  9. #include "components/page_load_metrics/browser/observers/ad_metrics/ads_page_load_metrics_observer.h"
  10. #include "components/page_load_metrics/browser/page_load_metrics_embedder_base.h"
  11. #include "components/page_load_metrics/browser/page_load_metrics_memory_tracker.h"
  12. #include "components/page_load_metrics/browser/page_load_metrics_observer.h"
  13. #include "components/page_load_metrics/browser/page_load_tracker.h"
  14. #include "weblayer/browser/heavy_ad_service_factory.h"
  15. #include "weblayer/browser/i18n_util.h"
  16. #include "weblayer/browser/no_state_prefetch/no_state_prefetch_utils.h"
  17. #include "weblayer/browser/page_load_metrics_observer_impl.h"
  18. #include "weblayer/browser/weblayer_page_load_metrics_memory_tracker_factory.h"
  19. namespace content {
  20. class BrowserContext;
  21. } // namespace content
  22. namespace page_load_metrics {
  23. class PageLoadMetricsMemoryTracker;
  24. } // namespace page_load_metrics
  25. namespace weblayer {
  26. namespace {
  27. base::RepeatingCallback<void(page_load_metrics::PageLoadTracker*)>*
  28. g_callback_for_testing;
  29. class PageLoadMetricsEmbedder
  30. : public page_load_metrics::PageLoadMetricsEmbedderBase {
  31. public:
  32. PageLoadMetricsEmbedder(const PageLoadMetricsEmbedder&) = delete;
  33. PageLoadMetricsEmbedder& operator=(const PageLoadMetricsEmbedder&) = delete;
  34. explicit PageLoadMetricsEmbedder(content::WebContents* web_contents)
  35. : PageLoadMetricsEmbedderBase(web_contents) {}
  36. ~PageLoadMetricsEmbedder() override = default;
  37. // page_load_metrics::PageLoadMetricsEmbedderBase:
  38. bool IsNewTabPageUrl(const GURL& url) override { return false; }
  39. bool IsNoStatePrefetch(content::WebContents* web_contents) override {
  40. return NoStatePrefetchContentsFromWebContents(web_contents);
  41. }
  42. bool IsExtensionUrl(const GURL& url) override { return false; }
  43. bool IsSidePanel(content::WebContents* web_contents) override {
  44. // The side panel is not supported in WebLayer so this always returns false.
  45. return false;
  46. }
  47. page_load_metrics::PageLoadMetricsMemoryTracker*
  48. GetMemoryTrackerForBrowserContext(
  49. content::BrowserContext* browser_context) override {
  50. if (!base::FeatureList::IsEnabled(features::kV8PerFrameMemoryMonitoring))
  51. return nullptr;
  52. return WeblayerPageLoadMetricsMemoryTrackerFactory::GetForBrowserContext(
  53. browser_context);
  54. }
  55. protected:
  56. // page_load_metrics::PageLoadMetricsEmbedderBase:
  57. void RegisterEmbedderObservers(
  58. page_load_metrics::PageLoadTracker* tracker) override {
  59. tracker->AddObserver(std::make_unique<PageLoadMetricsObserverImpl>());
  60. if (!IsNoStatePrefetch(web_contents())) {
  61. std::unique_ptr<page_load_metrics::AdsPageLoadMetricsObserver>
  62. ads_observer =
  63. page_load_metrics::AdsPageLoadMetricsObserver::CreateIfNeeded(
  64. tracker->GetWebContents(),
  65. HeavyAdServiceFactory::GetForBrowserContext(
  66. tracker->GetWebContents()->GetBrowserContext()),
  67. base::BindRepeating(&i18n::GetApplicationLocale));
  68. if (ads_observer)
  69. tracker->AddObserver(std::move(ads_observer));
  70. }
  71. if (g_callback_for_testing)
  72. (*g_callback_for_testing).Run(tracker);
  73. }
  74. };
  75. } // namespace
  76. void InitializePageLoadMetricsForWebContents(
  77. content::WebContents* web_contents) {
  78. // Change this method? consider to modify the peer in
  79. // android_webview/browser/page_load_metrics/page_load_metrics_initialize.cc
  80. // chrome/browser/page_load_metrics/page_load_metrics_initialize.cc
  81. // as well.
  82. page_load_metrics::MetricsWebContentsObserver::CreateForWebContents(
  83. web_contents, std::make_unique<PageLoadMetricsEmbedder>(web_contents));
  84. }
  85. void SetRegisterEmbedderObserversForTesting(
  86. base::RepeatingCallback<void(page_load_metrics::PageLoadTracker*)>*
  87. callback) {
  88. g_callback_for_testing = callback;
  89. }
  90. } // namespace weblayer