Add ref count to service workers for extension API.

extensions/renderer/service_worker_request_sender.cc

 // Copyright 2016 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 "extensions/renderer/service_worker_request_sender.h"
 
+#include "base/guid.h"
 #include "content/public/child/worker_thread.h"
 #include "extensions/common/extension_messages.h"
 #include "extensions/renderer/worker_thread_dispatcher.h"
 
 namespace extensions {
 
 ServiceWorkerRequestSender::ServiceWorkerRequestSender(
     WorkerThreadDispatcher* dispatcher,
-    int embedded_worker_id)
-    : dispatcher_(dispatcher), embedded_worker_id_(embedded_worker_id) {}
+    int64_t service_worker_version_id)
+    : dispatcher_(dispatcher),
+      service_worker_version_id_(service_worker_version_id) {}
 
 ServiceWorkerRequestSender::~ServiceWorkerRequestSender() {}
 
 void ServiceWorkerRequestSender::SendRequest(
     content::RenderFrame* render_frame,
     bool for_io_thread,
     ExtensionHostMsg_Request_Params& params) {
   DCHECK(!render_frame && !for_io_thread);
   int worker_thread_id = content::WorkerThread::GetCurrentId();
   DCHECK_GT(worker_thread_id, 0);
   params.worker_thread_id = worker_thread_id;
-  params.embedded_worker_id = embedded_worker_id_;
+  params.service_worker_version_id = service_worker_version_id_;
+  std::string guid = base::GenerateGUID();
+  request_id_to_guid_[params.request_id] = guid;
+
+  // Keeps the worker alive during extension function call. Balanced in
+  // HandleWorkerResponse().
+  dispatcher_->Send(new ExtensionHostMsg_IncrementServiceWorkerActivity(
+      service_worker_version_id_, guid));
 
   dispatcher_->Send(new ExtensionHostMsg_RequestWorker(params));
 }
 
+void ServiceWorkerRequestSender::HandleWorkerResponse(
+    int request_id,
+    int64_t service_worker_version_id,
+    bool success,
+    const base::ListValue& response,
+    const std::string& error) {
+  RequestSender::HandleResponse(request_id, success, response, error);
+
+  std::map<int, std::string>::iterator iter =
+      request_id_to_guid_.find(request_id);
+  DCHECK(iter != request_id_to_guid_.end());
+  dispatcher_->Send(new ExtensionHostMsg_DecrementServiceWorkerActivity(
+      service_worker_version_id_, iter->second));
+  request_id_to_guid_.erase(iter);
+}
+
 }  // namespace extensions