Index: chrome/renderer/page_load_metrics/page_timing_sender.cc |
diff --git a/chrome/renderer/page_load_metrics/page_timing_sender.cc b/chrome/renderer/page_load_metrics/page_timing_sender.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f06997ddef9220972fb1458031e002791010caf2 |
--- /dev/null |
+++ b/chrome/renderer/page_load_metrics/page_timing_sender.cc |
@@ -0,0 +1,77 @@ |
+// Copyright 2017 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/renderer/page_load_metrics/page_timing_sender.h" |
+ |
+#include "base/feature_list.h" |
+#include "base/memory/ptr_util.h" |
+#include "chrome/common/chrome_features.h" |
+#include "chrome/common/page_load_metrics/page_load_metrics.mojom.h" |
+#include "chrome/common/page_load_metrics/page_load_metrics_messages.h" |
+#include "content/public/common/associated_interface_provider.h" |
+#include "content/public/renderer/render_frame.h" |
+#include "ipc/ipc_sender.h" |
+ |
+namespace page_load_metrics { |
+ |
+class LegacyIPCPageTimingSender : public IPC::Sender, public PageTimingSender { |
Bryan McQuade
2017/05/19 14:06:18
does this need to implement IPC::Sender anymore? s
lpy
2017/05/19 23:52:38
Done.
|
+ public: |
+ LegacyIPCPageTimingSender(content::RenderFrame* render_frame, |
+ const int routing_id) |
+ : render_frame_(render_frame), routing_id_(routing_id) {} |
Bryan McQuade
2017/05/19 14:06:18
can the render frame ever be null? iirc we had som
lpy
2017/05/19 23:52:38
Done.
|
+ ~LegacyIPCPageTimingSender() override {} |
+ |
+ void SendTiming(const mojom::PageLoadTimingPtr& timing, |
+ const mojom::PageLoadMetadataPtr& metadata) override { |
+ Send(new PageLoadMetricsMsg_TimingUpdated(routing_id_, *timing, *metadata)); |
+ } |
+ |
+ private: |
+ // IPC::Sender implementation. |
+ bool Send(IPC::Message* message) override { |
Bryan McQuade
2017/05/19 14:06:18
let's inline this up into SendTiming() above
lpy
2017/05/19 23:52:38
Done.
|
+ if (render_frame_) |
+ return render_frame_->Send(message); |
+ delete message; |
Bryan McQuade
2017/05/19 14:06:19
do we want to delete here? I was under the impress
lpy
2017/05/19 23:52:38
https://cs.chromium.org/chromium/src/content/publi
|
+ return false; |
+ } |
+ |
+ content::RenderFrame* render_frame_ = nullptr; |
+ const int routing_id_; |
+}; |
+ |
+class MojoIPCPageTimingSender : public PageTimingSender { |
+ public: |
+ explicit MojoIPCPageTimingSender(content::RenderFrame* render_frame) |
+ : render_frame_(render_frame) {} |
+ ~MojoIPCPageTimingSender() override {} |
+ void SendTiming(const mojom::PageLoadTimingPtr& timing, |
+ const mojom::PageLoadMetadataPtr& metadata) override { |
+ DCHECK(render_frame_ && render_frame_->GetWebFrame()); |
+ if (!page_load_metrics_) { |
Bryan McQuade
2017/05/19 14:06:18
I'd rather initialize this in the constructor (ins
lpy
2017/05/19 23:52:38
Done.
|
+ render_frame_->GetRemoteAssociatedInterfaces()->GetInterface( |
+ &page_load_metrics_); |
+ } |
+ page_load_metrics_->UpdateTiming(timing->Clone(), metadata->Clone()); |
+ } |
+ |
+ private: |
+ content::RenderFrame* const render_frame_; |
+ // Use associated interface to make sure mojo messages are ordered with regard |
+ // to legacy IPC messages. |
+ mojom::PageLoadMetricsAssociatedPtr page_load_metrics_; |
+}; |
+ |
+// static |
+std::unique_ptr<PageTimingSender> PageTimingSender::CreatePageTimingSender( |
+ content::RenderFrame* render_frame, |
+ const int routing_id) { |
+ if (base::FeatureList::IsEnabled(features::kPageLoadMetricsMojofication)) { |
+ return base::WrapUnique<PageTimingSender>( |
+ new MojoIPCPageTimingSender(render_frame)); |
+ } |
+ return base::WrapUnique<PageTimingSender>( |
+ new LegacyIPCPageTimingSender(render_frame, routing_id)); |
+} |
+ |
+} // namespace page_load_metrics |