chrome/browser/ui/webui/webui_load_timer.h

// Copyright 2019 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_UI_WEBUI_WEBUI_LOAD_TIMER_H_
#define CHROME_BROWSER_UI_WEBUI_WEBUI_LOAD_TIMER_H_

#include <memory>
#include <string>

#include "base/macros.h"
#include "base/timer/elapsed_timer.h"
#include "content/public/browser/web_contents_observer.h"

// Measures and reports page load times for WebUI.
class WebuiLoadTimer : public content::WebContentsObserver {
 public:
  // Load times are reported to UMA using the provided strings which
  // must not be empty.
  // * |document_initial_load_uma_id| - corresponds to DocumentLoadedInFrame
  // * |document_load_completed_uma_id| - corresponds to
  //   DocumentOnLoadCompletedInMainFrame
  WebuiLoadTimer(content::WebContents* web_contents,
                 const std::string& document_initial_load_uma_id,
                 const std::string& document_load_completed_uma_id);
  ~WebuiLoadTimer() override;

  // WebContentsObserver
  void DidStartNavigation(
      content::NavigationHandle* navigation_handle) override;
  void DocumentLoadedInFrame(
      content::RenderFrameHost* render_frame_host) override;
  void DocumentOnLoadCompletedInMainFrame() override;

 private:
  std::string document_initial_load_uma_id_;
  std::string document_load_completed_uma_id_;
  std::unique_ptr<base::ElapsedTimer> timer_;

  DISALLOW_COPY_AND_ASSIGN(WebuiLoadTimer);
};

#endif  // CHROME_BROWSER_UI_WEBUI_WEBUI_LOAD_TIMER_H_