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

components/html_viewer/html_document.cc

 // Copyright 2014 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/html_document.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/single_thread_task_runner.h"
 #include "base/stl_util.h"
 #include "base/strings/string_util.h"
 #include "base/thread_task_runner_handle.h"
 #include "components/devtools_service/public/cpp/switches.h"
 #include "components/html_viewer/blink_input_events_type_converters.h"
 #include "components/html_viewer/blink_url_request_type_converters.h"
 #include "components/html_viewer/devtools_agent_impl.h"
 #include "components/html_viewer/geolocation_client_impl.h"
 #include "components/html_viewer/global_state.h"
 #include "components/html_viewer/media_factory.h"
+#include "components/html_viewer/stats_collection_controller.h"
 #include "components/html_viewer/web_layer_tree_view_impl.h"
 #include "components/html_viewer/web_storage_namespace_impl.h"
 #include "components/html_viewer/web_url_loader_impl.h"
 #include "components/view_manager/public/cpp/view.h"
 #include "components/view_manager/public/cpp/view_manager.h"
 #include "components/view_manager/public/cpp/view_property.h"
 #include "components/view_manager/public/interfaces/surfaces.mojom.h"
 #include "mojo/application/public/cpp/application_impl.h"
 #include "mojo/application/public/cpp/connect.h"
 #include "mojo/application/public/interfaces/shell.mojom.h"
@@ skipping 120 matching lines @@
     : html_document_app(html_document_app),
       connection(connection),
       response(response.Pass()),
       global_state(global_state),
       delete_callback(delete_callback) {
 }
 
 HTMLDocument::CreateParams::~CreateParams() {
 }
 
+HTMLDocument::Observer::Observer(HTMLDocument* html_document)
+  : html_document_(html_document) {

[sky, 2015/08/12 23:54:36]

nit: make sure you run git cl format.

[msw, 2015/08/14 23:20:34]

Done.

+  html_document_->AddObserver(this);
+}
+
+HTMLDocument::Observer::~Observer() {
+  html_document_->RemoveObserver(this);
+}
+
 HTMLDocument::HTMLDocument(HTMLDocument::CreateParams* params)
     : app_refcount_(params->html_document_app->app_lifetime_helper()
                         ->CreateAppRefCount()),
       html_document_app_(params->html_document_app),
       response_(params->response.Pass()),
       navigator_host_(params->connection->GetServiceProvider()),
       web_view_(nullptr),
       root_(nullptr),
       view_manager_client_factory_(params->html_document_app->shell(), this),
       global_state_(params->global_state),
-      delete_callback_(params->delete_callback) {
+      delete_callback_(params->delete_callback),
+      stats_collection_controller_(nullptr) {
   params->connection->AddService(
       static_cast<InterfaceFactory<mojo::AxProvider>*>(this));
   params->connection->AddService(&view_manager_client_factory_);
 
   if (global_state_->did_init())
     Load(response_.Pass());
 }
 
 void HTMLDocument::Destroy() {
   // See comment in header for a description of lifetime.
   if (root_) {
     // Deleting the ViewManager calls back to OnViewManagerDestroyed() and
     // triggers deletion.
     delete root_->view_manager();
   } else {
     delete this;
   }
 }
 
+void HTMLDocument::AddObserver(HTMLDocument::Observer* observer) {
+  observers_.AddObserver(observer);
+}
+
+void HTMLDocument::RemoveObserver(HTMLDocument::Observer* observer) {
+  observers_.RemoveObserver(observer);
+}
+
 HTMLDocument::~HTMLDocument() {
   delete_callback_.Run(this);
 
   STLDeleteElements(&ax_providers_);
   STLDeleteElements(&ax_provider_requests_);
 
   if (web_view_)
     web_view_->close();
   if (root_)
     root_->RemoveObserver(this);
@@ skipping 38 matching lines @@
 
   // TODO(yzshen): http://crbug.com/498986 Creating DevToolsAgentImpl instances
   // causes html_viewer_apptests flakiness currently. Before we fix that we
   // cannot enable remote debugging (which is required by Telemetry tests) on
   // the bots.
   if (EnableRemoteDebugging()) {
     devtools_agent_.reset(
         new DevToolsAgentImpl(main_frame, html_document_app_->shell()));
   }
 
+  stats_collection_controller_ =
+      StatsCollectionController::Install(this, html_document_app_);
+
   GURL url(response->url);
 
   WebURLRequestExtraData* extra_data = new WebURLRequestExtraData;
   extra_data->synthetic_response = response.Pass();
 
   blink::WebURLRequest web_request;
   web_request.initialize();
   web_request.setURL(url);
   web_request.setExtraData(extra_data);
 
@@ skipping 131 matching lines @@
 
 void HTMLDocument::didAddMessageToConsole(
     const blink::WebConsoleMessage& message,
     const blink::WebString& source_name,
     unsigned source_line,
     const blink::WebString& stack_trace) {
   VLOG(1) << "[" << source_name.utf8() << "(" << source_line << ")] "
           << message.text.utf8();
 }
 
+void HTMLDocument::didStartLoading(bool toDifferentDocument) {
+  // TODO(msw): Nix observer pattern, use stats_collection_controller_ directly?
+  FOR_EACH_OBSERVER(HTMLDocument::Observer, observers_, DidStartLoading());
+}
+
+void HTMLDocument::didStopLoading() {
+  // TODO(msw): Nix observer pattern, use stats_collection_controller_ directly?
+  FOR_EACH_OBSERVER(HTMLDocument::Observer, observers_, DidStopLoading());
+}
+
+void HTMLDocument::didHandleOnloadEvents(blink::WebLocalFrame* frame) {
+  if (stats_collection_controller_)
+    stats_collection_controller_->DidHandleOnloadEvents(base::Time::Now());
+}
+
 void HTMLDocument::didFinishLoad(blink::WebLocalFrame* frame) {
   // TODO(msw): Notify AxProvider clients of updates on child frame loads.
   did_finish_load_ = true;
   // Bind any pending AxProviderImpl interface requests.
   for (auto it : ax_provider_requests_)
     ax_providers_.insert(new AxProviderImpl(web_view_, it->Pass()));
   STLDeleteElements(&ax_provider_requests_);
 }
 
 void HTMLDocument::didNavigateWithinPage(
     blink::WebLocalFrame* frame,
     const blink::WebHistoryItem& history_item,
     blink::WebHistoryCommitType commit_type) {
   if (navigator_host_.get())
     navigator_host_->DidNavigateLocally(history_item.urlString().utf8());
 }
 
+void HTMLDocument::didFirstVisuallyNonEmptyLayout(blink::WebLocalFrame* frame) {
+  if (stats_collection_controller_) {
+    stats_collection_controller_->DidFirstVisuallyNonEmptyLayout(
+        base::Time::Now());
+  }
+}
+
 blink::WebEncryptedMediaClient* HTMLDocument::encryptedMediaClient() {
   return global_state_->media_factory()->GetEncryptedMediaClient();
 }
 
 void HTMLDocument::OnViewBoundsChanged(View* view,
                                        const Rect& old_bounds,
                                        const Rect& new_bounds) {
   DCHECK_EQ(view, root_);
   UpdateWebviewSizeFromViewSize();
 }
@@ skipping 40 matching lines @@
 
 void HTMLDocument::UpdateFocus() {
   if (!web_view_)
     return;
   bool is_focused = root_ && root_->HasFocus();
   web_view_->setFocus(is_focused);
   web_view_->setIsActive(is_focused);
 }
 
 }  // namespace html_viewer