Adding previews information to PLM UKM

chrome/browser/page_load_metrics/metrics_web_contents_observer.h

 // Copyright 2015 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.
 
 #ifndef CHROME_BROWSER_PAGE_LOAD_METRICS_METRICS_WEB_CONTENTS_OBSERVER_H_
 #define CHROME_BROWSER_PAGE_LOAD_METRICS_METRICS_WEB_CONTENTS_OBSERVER_H_
 
 #include <map>
 #include <memory>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/observer_list.h"
+#include "base/optional.h"
 #include "base/time/time.h"
 #include "chrome/browser/page_load_metrics/page_load_metrics_observer.h"
 #include "chrome/common/page_load_metrics/page_load_metrics.mojom.h"
 #include "chrome/common/page_load_metrics/page_load_timing.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
+#include "components/ukm/ukm_source.h"
 #include "content/public/browser/render_widget_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_binding_set.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/browser/web_contents_user_data.h"
 #include "content/public/common/resource_type.h"
 #include "third_party/WebKit/public/platform/WebInputEvent.h"
 
 namespace content {
 class NavigationHandle;
@@ skipping 109 matching lines @@
 
   // Register / unregister TestingObservers. Should only be called from tests.
   void AddTestingObserver(TestingObserver* observer);
   void RemoveTestingObserver(TestingObserver* observer);
 
   // public only for testing
   void OnTimingUpdated(content::RenderFrameHost* render_frame_host,
                        const mojom::PageLoadTiming& timing,
                        const mojom::PageLoadMetadata& metadata);
 
+  // Gets the Source ID for the load related to |navigation_handle| if it is
+  // pending, can be empty if the load is not pending.
+  base::Optional<ukm::SourceId> GetUKMSourceIdForNavigationHandle(
+      content::NavigationHandle* navigation_handle);
+
+  // Gets the Source ID for the currently committed load if there is a committed
+  // load.
+  base::Optional<ukm::SourceId> GetUKMSourceForCommittedLoad();
+
  private:
   friend class content::WebContentsUserData<MetricsWebContentsObserver>;
 
   // page_load_metrics::mojom::PageLoadMetrics implementation.
   void UpdateTiming(mojom::PageLoadTimingPtr timing,
                     mojom::PageLoadMetadataPtr metadata) override;
 
   // Called from legacy IPC.
   void OnUpdateTimingOverIPC(content::RenderFrameHost* render_frame_host,
                              const mojom::PageLoadTiming& timing,
@@ skipping 68 matching lines @@
   base::ObserverList<TestingObserver> testing_observers_;
   content::WebContentsFrameBindingSet<mojom::PageLoadMetrics>
       page_load_metrics_binding_;
 
   DISALLOW_COPY_AND_ASSIGN(MetricsWebContentsObserver);
 };
 
 }  // namespace page_load_metrics
 
 #endif  // CHROME_BROWSER_PAGE_LOAD_METRICS_METRICS_WEB_CONTENTS_OBSERVER_H_