Add ref count to service workers for extension API.

content/browser/service_worker/service_worker_context_wrapper.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_context_wrapper.h"
 
 #include <map>
 #include <set>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/barrier_closure.h"
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/lazy_instance.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/profiler/scoped_tracker.h"
 #include "base/single_thread_task_runner.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/threading/thread_task_runner_handle.h"
+#include "content/browser/service_worker/embedded_worker_registry.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_observer.h"
 #include "content/browser/service_worker/service_worker_process_manager.h"
 #include "content/browser/service_worker/service_worker_quota_client.h"
 #include "content/browser/service_worker/service_worker_version.h"
 #include "content/browser/storage_partition_impl.h"
 #include "content/common/service_worker/service_worker_utils.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "net/base/url_util.h"
@@ skipping 61 matching lines @@
                                           header_names.end());
 }
 
 bool ServiceWorkerContext::IsExcludedHeaderNameForFetchEvent(
     const std::string& header_name) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   return g_excluded_header_name_set.Get().find(header_name) !=
          g_excluded_header_name_set.Get().end();
 }
 
+// static.
+void ServiceWorkerContext::IncrementEmbeddedWorkerRefCount(
+    StoragePartition* storage_partition,
+    int embedded_worker_id) {
+  ServiceWorkerContextWrapper::IncrementEmbeddedWorkerRefCount(
+      storage_partition, embedded_worker_id);
+}
+
+// static.
+void ServiceWorkerContext::DecrementEmbeddedWorkerRefCount(
+    StoragePartition* storage_partition,
+    int embedded_worker_id) {
+  ServiceWorkerContextWrapper::DecrementEmbeddedWorkerRefCount(
+      storage_partition, embedded_worker_id);
+}
+
 ServiceWorkerContextWrapper::ServiceWorkerContextWrapper(
     BrowserContext* browser_context)
     : observer_list_(
           new base::ObserverListThreadSafe<ServiceWorkerContextObserver>()),
       process_manager_(new ServiceWorkerProcessManager(browser_context)),
       is_incognito_(false),
       storage_partition_(nullptr),
       resource_context_(nullptr) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 }
@@ skipping 606 matching lines @@
 void ServiceWorkerContextWrapper::ShutdownOnIO() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   // Can be null in tests.
   if (request_context_getter_)
     request_context_getter_->RemoveObserver(this);
   request_context_getter_ = nullptr;
   resource_context_ = nullptr;
   context_core_.reset();
 }
 
+// static.
+void ServiceWorkerContextWrapper::IncrementEmbeddedWorkerRefCount(
+    StoragePartition* storage_partition,
+    int embedded_worker_id) {
+  ServiceWorkerContextWrapper* context_wrapper =
+      static_cast<ServiceWorkerContextWrapper*>(
+          storage_partition->GetServiceWorkerContext());
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(
+          &ServiceWorkerContextWrapper::IncrementEmbeddedWorkerRefCountOnIO,
+          context_wrapper, embedded_worker_id));
+}
+
+// static.
+void ServiceWorkerContextWrapper::DecrementEmbeddedWorkerRefCount(
+    StoragePartition* storage_partition,
+    int embedded_worker_id) {
+  ServiceWorkerContextWrapper* context_wrapper =
+      static_cast<ServiceWorkerContextWrapper*>(
+          storage_partition->GetServiceWorkerContext());
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(
+          &ServiceWorkerContextWrapper::DecrementEmbeddedWorkerRefCountOnIO,
+          context_wrapper, embedded_worker_id));
+}
+
+void ServiceWorkerContextWrapper::IncrementEmbeddedWorkerRefCountOnIO(
+    int embedded_worker_id) {
+  context()
+      ->embedded_worker_registry()
+      ->GetWorker(embedded_worker_id)
+      ->IncrementRefCount();
+}
+void ServiceWorkerContextWrapper::DecrementEmbeddedWorkerRefCountOnIO(
+    int embedded_worker_id) {
+  context()
+      ->embedded_worker_registry()
+      ->GetWorker(embedded_worker_id)
+      ->DecrementRefCount();
+}
+
 void ServiceWorkerContextWrapper::DidDeleteAndStartOver(
     ServiceWorkerStatusCode status) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   if (status != SERVICE_WORKER_OK) {
     context_core_.reset();
     return;
   }
   context_core_.reset(new ServiceWorkerContextCore(context_core_.get(), this));
   DVLOG(1) << "Restarted ServiceWorkerContextCore successfully.";
 
   observer_list_->Notify(FROM_HERE,
                          &ServiceWorkerContextObserver::OnStorageWiped);
 }
 
 ServiceWorkerContextCore* ServiceWorkerContextWrapper::context() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   return context_core_.get();
 }
 
 }  // namespace content