Keep alive ServiceWorkers when devtools is attached

content/browser/service_worker/service_worker_internals_ui.cc

 // Copyright 2014 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_internals_ui.h"
 
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 277 matching lines @@
   ScopedVector<const Value> args;
   args.push_back(GetRegistrationListValue(live_registrations));
   args.push_back(GetVersionListValue(live_versions));
   args.push_back(GetRegistrationListValue(stored_registrations));
   args.push_back(new FundamentalValue(partition_id));
   args.push_back(new StringValue(context_path.value()));
   internals->web_ui()->CallJavascriptFunction("serviceworker.onPartitionData",
                                               args.get());
 }
 
+void SetKeepAliveMode(
+    ServiceWorkerContextWrapper* wrapper,
+    bool keep_alive_mode) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  if (!wrapper->context())
+    return;
+  wrapper->context()->SetKeepAliveMode(keep_alive_mode);
+}
+
+void SetKeepAliveModeForStoragePartition(
+    bool keep_alive_mode,
+    StoragePartition* storage_partition) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  scoped_refptr<ServiceWorkerContextWrapper> wrapper =
+      static_cast<ServiceWorkerContextWrapper*>(
+          storage_partition->GetServiceWorkerContext());
+  BrowserThread::PostTask(
+      BrowserThread::IO,
+      FROM_HERE,
+      base::Bind(&SetKeepAliveMode, wrapper, keep_alive_mode));
+}
+
 }  // namespace
 
 class ServiceWorkerInternalsUI::PartitionObserver
     : public ServiceWorkerContextObserver {
  public:
   PartitionObserver(int partition_id, WebUI* web_ui)
       : partition_id_(partition_id), web_ui_(web_ui) {}
   ~PartitionObserver() override {}
   // ServiceWorkerContextObserver overrides:
   void OnWorkerStarted(int64 version_id,
@@ skipping 144 matching lines @@
       base::Bind(&ServiceWorkerInternalsUI::RemoveObserverFromStoragePartition,
                  base::Unretained(this));
   BrowserContext::ForEachStoragePartition(browser_context, remove_observer_cb);
 }
 
 void ServiceWorkerInternalsUI::GetOptions(const ListValue* args) {
   DictionaryValue options;
   options.SetBoolean("debug_on_start",
                      EmbeddedWorkerDevToolsManager::GetInstance()
                          ->debug_service_worker_on_start());
+  options.SetBoolean("keep_alive",
+                     EmbeddedWorkerDevToolsManager::GetInstance()
+                         ->keep_alive_service_workers());
   web_ui()->CallJavascriptFunction("serviceworker.onOptions", options);
 }
 
 void ServiceWorkerInternalsUI::SetOption(const ListValue* args) {
   std::string option_name;
   bool option_boolean;
-  if (!args->GetString(0, &option_name) || option_name != "debug_on_start" ||
+  if (!args->GetString(0, &option_name) ||
       !args->GetBoolean(1, &option_boolean)) {
     return;
   }
-  EmbeddedWorkerDevToolsManager::GetInstance()
-      ->set_debug_service_worker_on_start(option_boolean);
+  if (option_name == "debug_on_start") {
+    EmbeddedWorkerDevToolsManager::GetInstance()
+        ->set_debug_service_worker_on_start(option_boolean);
+  } else if (option_name == "keep_alive") {
+    EmbeddedWorkerDevToolsManager::GetInstance()
+        ->set_keep_alive_service_workers(option_boolean);
+    BrowserContext::ForEachStoragePartition(
+        web_ui()->GetWebContents()->GetBrowserContext(),
+        base::Bind(&SetKeepAliveModeForStoragePartition, option_boolean));
+  }
 }
 
 void ServiceWorkerInternalsUI::GetAllRegistrations(const ListValue* args) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   BrowserContext* browser_context =
       web_ui()->GetWebContents()->GetBrowserContext();
   // Safe to use base::Unretained(this) because
   // ForEachStoragePartition is synchronous.
   BrowserContext::StoragePartitionCallback add_context_cb =
       base::Bind(&ServiceWorkerInternalsUI::AddContextFromStoragePartition,
@@ skipping 184 matching lines @@
     return;
   }
 
   base::Callback<void(ServiceWorkerStatusCode)> callback =
       base::Bind(OperationCompleteCallback, AsWeakPtr(), callback_id);
   FindRegistrationForPattern(
       context, GURL(scope_string), base::Bind(StartActiveWorker, callback));
 }
 
 }  // namespace content