Adding previews information to PLM UKM

chrome/browser/page_load_metrics/metrics_web_contents_observer.cc

 // 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.
 
 #include "chrome/browser/page_load_metrics/metrics_web_contents_observer.h"
 
 #include <algorithm>
 #include <ostream>
 #include <string>
 #include <utility>
@@ skipping 666 matching lines @@
   if (observer_) {
     observer_->RemoveTestingObserver(this);
     observer_ = nullptr;
   }
 }
 
 void MetricsWebContentsObserver::TestingObserver::OnGoingAway() {
   observer_ = nullptr;
 }
 
+base::Optional<ukm::SourceId>
+MetricsWebContentsObserver::GetUKMSourceIdForNavigationHandle(
+    content::NavigationHandle* navigation_handle) {
+  base::Optional<ukm::SourceId> source_id;
+  auto it = provisional_loads_.find(navigation_handle);
+  if (it != provisional_loads_.end())
+    source_id = it->second->source_id();
+  return source_id;
+}
+
+base::Optional<ukm::SourceId>
+MetricsWebContentsObserver::GetUKMSourceForCommittedLoad() {
+  base::Optional<ukm::SourceId> source_id;
+  if (committed_load_)
+    source_id = committed_load_->source_id();
+  return source_id;
+}
+
 }  // namespace page_load_metrics