Expands the chrome.experimental.processes extension API. ...

chrome/browser/extensions/extension_event_router.cc

 // Copyright (c) 2010 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 "chrome/browser/extensions/extension_event_router.h"
 
 #include "base/values.h"
 #include "chrome/browser/child_process_security_policy.h"
 #include "chrome/browser/extensions/extension_devtools_manager.h"
+#include "chrome/browser/extensions/extension_processes_api.h"
+#include "chrome/browser/extensions/extension_processes_api_constants.h"
 #include "chrome/browser/extensions/extension_tabs_module.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/renderer_host/render_process_host.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/common/render_messages.h"
 
 namespace {
 
 const char kDispatchEvent[] = "Event.dispatchJSON";
@@ skipping 35 matching lines @@
 void ExtensionEventRouter::AddEventListener(
     const std::string& event_name,
     int render_process_id) {
   DCHECK_EQ(listeners_[event_name].count(render_process_id), 0u) << event_name;
   listeners_[event_name].insert(render_process_id);
 
   if (extension_devtools_manager_.get()) {
     extension_devtools_manager_->AddEventListener(event_name,
                                                   render_process_id);
   }
+
+  // We lazily tell the TaskManager to start updating when listeners to the
+  // processes.onUpdated event arrive.
+  if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0)
+    ExtensionProcessesEventRouter::GetInstance()->ListenerAdded();
 }
 
 void ExtensionEventRouter::RemoveEventListener(
     const std::string& event_name,
     int render_process_id) {
   DCHECK_EQ(listeners_[event_name].count(render_process_id), 1u) <<
       " PID=" << render_process_id << " event=" << event_name;
   listeners_[event_name].erase(render_process_id);
 
   if (extension_devtools_manager_.get()) {
     extension_devtools_manager_->RemoveEventListener(event_name,
                                                      render_process_id);
   }
+
+  // If a processes.onUpdated event listener is removed (or a process with one
+  // exits), then we let the TaskManager know that it has one fewer listener.
+  if (event_name.compare(extension_processes_api_constants::kOnUpdated) == 0)
+    ExtensionProcessesEventRouter::GetInstance()->ListenerRemoved();
 }
 
 bool ExtensionEventRouter::HasEventListener(const std::string& event_name) {
   return (listeners_.find(event_name) != listeners_.end() &&
           !listeners_[event_name].empty());
 }
 
 void ExtensionEventRouter::DispatchEventToRenderers(
     const std::string& event_name, const std::string& event_args,
     Profile* restrict_to_profile, const GURL& event_url) {
@@ skipping 50 matching lines @@
         if (current->second.count(renderer->id()) != 0)
           RemoveEventListener(current->first, renderer->id());
       }
       break;
     }
     default:
       NOTREACHED();
       return;
   }
 }