Add stats collection for telemetry startup.warm.blank_page test.

components/html_viewer/stats_collection_controller.cc

Index: components/html_viewer/stats_collection_controller.cc
diff --git a/components/html_viewer/stats_collection_controller.cc b/components/html_viewer/stats_collection_controller.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5b6087dbf9fa0070aa5e0365f7efc77a0942e587
--- /dev/null
+++ b/components/html_viewer/stats_collection_controller.cc
@@ -0,0 +1,162 @@
+// 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 "components/html_viewer/stats_collection_controller.h"
+
+#include "base/command_line.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/metrics/histogram.h"
+#include "base/metrics/statistics_recorder.h"
+#include "base/time/time.h"
+#include "components/startup_metric_utils/startup_metric_utils.h"
+#include "gin/handle.h"
+#include "gin/object_template_builder.h"
+#include "mojo/application/public/cpp/application_impl.h"
+#include "mojo/services/tracing/public/cpp/switches.h"
+#include "third_party/WebKit/public/web/WebKit.h"
+#include "third_party/WebKit/public/web/WebLocalFrame.h"
+
+namespace html_viewer {
+
+namespace {
+
+// Initialize the histogram data using the given startup performance times.
+// TODO(msw): Use TimeTicks to avoid system clock changes: crbug.com/521164
+void GetStartupPerformanceTimesCallbackImpl(
+    tracing::StartupPerformanceTimesPtr times) {
+  base::StatisticsRecorder::Initialize();
+
+  const base::Time shell_process_creation_time =
+      base::Time::FromInternalValue(times->shell_process_creation_time);
+  startup_metric_utils::RecordSavedMainEntryPointTime(
+      shell_process_creation_time);
+  // TODO(msw): Use |browser_message_loop_start_time|, instead of letting
+  //            OnBrowserStartupComplete incorrectly call base::Time::Now()...
+  startup_metric_utils::OnBrowserStartupComplete(false);
+
+  // TODO(msw): Consolidate with chrome's Browser::OnWindowDidShow()...
+  const base::Time browser_window_display_time =
+      base::Time::FromInternalValue(times->browser_window_display_time);
+  base::TimeDelta browser_window_display_delta =
+      browser_window_display_time - shell_process_creation_time;
+  UMA_HISTOGRAM_LONG_TIMES("Startup.BrowserWindowDisplay",
+                           browser_window_display_delta);
+
+  // TODO(msw): Consolidate with chrome's PreMainMessageLoopRunImpl()...
+  // TODO(msw): Need to measure the "browser_open_start" time for this delta...
+  const base::Time browser_open_tabs_time =
+      base::Time::FromInternalValue(times->browser_open_tabs_time);
+  base::TimeDelta browser_open_tabs_delta =
+      browser_open_tabs_time - shell_process_creation_time;
+  UMA_HISTOGRAM_LONG_TIMES_100("Startup.BrowserOpenTabs",
+                               browser_open_tabs_delta);
+
+  // TODO(msw): Consolidate with chrome's first_web_contents_profiler.cc...
+  const base::Time first_web_contents_main_frame_load_time =
+      base::Time::FromInternalValue(
+          times->first_web_contents_main_frame_load_time);
+  base::TimeDelta first_web_contents_main_frame_load_delta =
+      first_web_contents_main_frame_load_time - shell_process_creation_time;
+  UMA_HISTOGRAM_LONG_TIMES_100("Startup.FirstWebContents.MainFrameLoad",
+                               first_web_contents_main_frame_load_delta);
+
+  // TODO(msw): Consolidate with chrome's first_web_contents_profiler.cc...
+  const base::Time first_visually_non_empty_layout_time =
+      base::Time::FromInternalValue(
+          times->first_visually_non_empty_layout_time);
+  base::TimeDelta first_web_contents_non_empty_paint_delta =
+      first_visually_non_empty_layout_time - shell_process_creation_time;
+  UMA_HISTOGRAM_LONG_TIMES_100("Startup.FirstWebContents.NonEmptyPaint",
+                               first_web_contents_non_empty_paint_delta);
+}
+
+}  // namespace
+
+// static
+gin::WrapperInfo StatsCollectionController::kWrapperInfo = {
+    gin::kEmbedderNativeGin};
+
+// static
+tracing::StartupPerformanceDataCollectorPtr StatsCollectionController::Install(
+    blink::WebFrame* frame,
+    mojo::ApplicationImpl* app) {
+  // Only make startup tracing available when running in the context of a test.
+  if (!app ||
+      !base::CommandLine::ForCurrentProcess()->HasSwitch(
+          tracing::kEnableStatsCollectionBindings)) {
+    return nullptr;
+  }
+
+  v8::Isolate* isolate = blink::mainThreadIsolate();
+  v8::HandleScope handle_scope(isolate);
+  v8::Local<v8::Context> context = frame->mainWorldScriptContext();
+  if (context.IsEmpty())
+    return nullptr;
+
+  v8::Context::Scope context_scope(context);
+
+  mojo::URLRequestPtr request(mojo::URLRequest::New());
+  request->url = mojo::String::From("mojo:tracing");
+  scoped_ptr<mojo::ApplicationConnection> connection =
+      app->ConnectToApplication(request.Pass());
+  if (!connection)
+    return nullptr;
+  tracing::StartupPerformanceDataCollectorPtr collector_for_controller;
+  tracing::StartupPerformanceDataCollectorPtr collector_for_caller;
+  connection->ConnectToService(&collector_for_controller);
+  connection->ConnectToService(&collector_for_caller);
+
+  gin::Handle<StatsCollectionController> controller = gin::CreateHandle(
+      isolate, new StatsCollectionController(collector_for_controller.Pass()));
+  DCHECK(!controller.IsEmpty());
+  v8::Local<v8::Object> global = context->Global();
+  global->Set(gin::StringToV8(isolate, "statsCollectionController"),
+              controller.ToV8());
+  return collector_for_caller.Pass();
+}
+
+StatsCollectionController::StatsCollectionController(
+    tracing::StartupPerformanceDataCollectorPtr collector)
+    : startup_performance_data_collector_(collector.Pass()) {}
+
+StatsCollectionController::~StatsCollectionController() {}
+
+gin::ObjectTemplateBuilder StatsCollectionController::GetObjectTemplateBuilder(
+    v8::Isolate* isolate) {
+  return gin::Wrappable<StatsCollectionController>::GetObjectTemplateBuilder(
+             isolate)
+      .SetMethod("getHistogram", &StatsCollectionController::GetHistogram)
+      .SetMethod("getBrowserHistogram",
+                 &StatsCollectionController::GetBrowserHistogram);
+}
+
+std::string StatsCollectionController::GetHistogram(
+    const std::string& histogram_name) {
+  DCHECK(base::CommandLine::ForCurrentProcess()->HasSwitch(
+      tracing::kEnableStatsCollectionBindings));
+
+  static bool startup_histogram_initialized = false;
+  if (!startup_histogram_initialized) {
+    // Get the startup performance times from the tracing service.
+    auto callback = base::Bind(&GetStartupPerformanceTimesCallbackImpl);
+    startup_performance_data_collector_->GetStartupPerformanceTimes(callback);
+    startup_performance_data_collector_.WaitForIncomingResponse();
+    DCHECK(base::StatisticsRecorder::IsActive());
+    startup_histogram_initialized = true;
+  }
+
+  std::string histogram_json = "{}";
+  base::HistogramBase* histogram =
+      base::StatisticsRecorder::FindHistogram(histogram_name);
+  if (histogram)
+    histogram->WriteJSON(&histogram_json);
+  return histogram_json;
+}
+
+std::string StatsCollectionController::GetBrowserHistogram(
+    const std::string& histogram_name) {
+  return GetHistogram(histogram_name);
+}
+
+}  // namespace html_viewer