Track input latency while starting a ServiceWorker for a navigation hint.

content/browser/service_worker/service_worker_context_wrapper.cc

Index: content/browser/service_worker/service_worker_context_wrapper.cc
diff --git a/content/browser/service_worker/service_worker_context_wrapper.cc b/content/browser/service_worker/service_worker_context_wrapper.cc
index e14764711d3a74eebb792c750509355a6ddbbe54..64ae3ea762f36fabfff18ff825a3dd97d0accf63 100644
--- a/content/browser/service_worker/service_worker_context_wrapper.cc
+++ b/content/browser/service_worker/service_worker_context_wrapper.cc
@@ -18,6 +18,7 @@
 #include "base/logging.h"
 #include "base/profiler/scoped_tracker.h"
 #include "base/single_thread_task_runner.h"
+#include "base/stl_util.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
@@ -289,11 +290,16 @@ void ServiceWorkerContextWrapper::StartServiceWorkerForNavigationHint(
     int render_process_id,
     const ResultCallback& callback) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  ++navigation_hint_task_count_per_process_[render_process_id];
+  ResultCallback wrapped_callback =
+      base::Bind(&ServiceWorkerContextWrapper::DidFinishNavigationHintTaskOnUI,
+                 this, render_process_id, callback);
+
   RenderProcessHost* host = RenderProcessHost::FromID(render_process_id);
   if (!host ||
       !RenderProcessHostImpl::IsSuitableHost(host, host->GetBrowserContext(),
                                              document_url)) {
-    callback.Run(false);
+    wrapped_callback.Run(false);
     return;
   }
 
@@ -301,7 +307,7 @@ void ServiceWorkerContextWrapper::StartServiceWorkerForNavigationHint(
       BrowserThread::IO, FROM_HERE,
       base::Bind(
           &ServiceWorkerContextWrapper::DidCheckRenderProcessForNavigationHint,
-          this, document_url, type, render_process_id, callback));
+          this, document_url, type, render_process_id, wrapped_callback));
 }
 
 void ServiceWorkerContextWrapper::DidCheckRenderProcessForNavigationHint(
@@ -361,6 +367,23 @@ void ServiceWorkerContextWrapper::DidStartServiceWorkerForNavigationHint(
                           base::Bind(callback, code == SERVICE_WORKER_OK));
 }
 
+void ServiceWorkerContextWrapper::DidFinishNavigationHintTaskOnUI(
+    int render_process_id,
+    const ResultCallback& callback,
+    bool result) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  if (!--navigation_hint_task_count_per_process_[render_process_id])
+    navigation_hint_task_count_per_process_.erase(render_process_id);
+  callback.Run(result);
+}
+
+bool ServiceWorkerContextWrapper::IsRunningNavigationHintTask(
+    int render_process_id) const {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  return base::ContainsKey(navigation_hint_task_count_per_process_,
+                           render_process_id);
+}
+
 void ServiceWorkerContextWrapper::StartServiceWorker(
     const GURL& pattern,
     const StatusCallback& callback) {