ServiceWorker: Keep the worker alive until FetchEvent.waitUntil settles

content/browser/service_worker/service_worker_version.cc

 // Copyright 2013 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 "content/browser/service_worker/service_worker_version.h"
 
 #include <stddef.h>
 
 #include <limits>
 #include <map>
@@ skipping 548 matching lines @@
   if (running_status() == RUNNING) {
     DCHECK(start_callbacks_.empty());
     task.Run();
     return;
   }
   StartWorker(purpose,
               base::Bind(&RunTaskAfterStartWorker, weak_factory_.GetWeakPtr(),
                          error_callback, task));
 }
 
+void ServiceWorkerVersion::DispatchFetchEvent(
+    int response_request_id,
+    int finish_request_id,
+    const ServiceWorkerFetchRequest& fetch_request,
+    const FetchResponseCallback& response_callback) {
+  ServiceWorkerMsg_FetchEvent message(response_request_id, finish_request_id,
+                                      fetch_request);
+
+  DCHECK_EQ(RUNNING, running_status());
+  PendingRequest<StatusCallback>* response_request =
+      custom_requests_.Lookup(response_request_id);
+  PendingRequest<StatusCallback>* finish_request =
+      custom_requests_.Lookup(finish_request_id);
+  DCHECK(response_request) << "Invalid request id for response request";
+  DCHECK(!response_request->listener)
+      << "Request already dispatched an IPC for response request";
+  DCHECK(finish_request) << "Invalid request id for finish request";
+  DCHECK(!finish_request->listener)
+      << "Request already dispatched an IPC for finish request";
+
+  ServiceWorkerStatusCode status = embedded_worker_->SendMessage(message);
+  if (status != SERVICE_WORKER_OK) {
+    base::ThreadTaskRunnerHandle::Get()->PostTask(
+        FROM_HERE, base::Bind(response_request->callback, status));
+    base::ThreadTaskRunnerHandle::Get()->PostTask(
+        FROM_HERE, base::Bind(finish_request->callback, status));
+    custom_requests_.Remove(response_request_id);
+    custom_requests_.Remove(finish_request_id);
+    return;
+  }
+
+  response_request->listener.reset(
+      new EventResponseHandler<ServiceWorkerHostMsg_FetchEventResponse,
+                               FetchResponseCallback>(
+          embedded_worker()->AsWeakPtr(), response_request_id,
+          response_callback));
+  finish_request->listener.reset(
+      new EventResponseHandler<ServiceWorkerHostMsg_FetchEventFinished,
+                               WebStatusCallback>(
+          embedded_worker()->AsWeakPtr(), finish_request_id,
+          base::Bind(&ServiceWorkerVersion::OnSimpleEventResponse, this)));
+}
+
 void ServiceWorkerVersion::AddControllee(
     ServiceWorkerProviderHost* provider_host) {
   const std::string& uuid = provider_host->client_uuid();
   CHECK(!provider_host->client_uuid().empty());
   DCHECK(!ContainsKey(controllee_map_, uuid));
   controllee_map_[uuid] = provider_host;
   // Keep the worker alive a bit longer right after a new controllee is added.
   RestartTick(&idle_time_);
   FOR_EACH_OBSERVER(Listener, listeners_,
                     OnControlleeAdded(this, provider_host));
@@ skipping 1124 matching lines @@
 void ServiceWorkerVersion::OnBeginEvent() {
   if (should_exclude_from_uma_ || running_status() != RUNNING ||
       idle_time_.is_null()) {
     return;
   }
   ServiceWorkerMetrics::RecordTimeBetweenEvents(base::TimeTicks::Now() -
                                                 idle_time_);
 }
 
 }  // namespace content