chrome/browser/task_manager/child_process_resource_provider.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_TASK_MANAGER_CHILD_PROCESS_RESOURCE_PROVIDER_H_
   6 #define CHROME_BROWSER_TASK_MANAGER_CHILD_PROCESS_RESOURCE_PROVIDER_H_
   7
   8 #include <map>
   9 #include <vector>
  10
  11 #include "base/basictypes.h"
  12 #include "chrome/browser/task_manager/resource_provider.h"
  13 #include "content/public/browser/browser_child_process_observer.h"
  14 #include "content/public/browser/notification_registrar.h"
  15
  16 class TaskManager;
  17
  18 namespace content {
  19 struct ChildProcessData;
  20 }
  21
  22 namespace task_manager {
  23
  24 class ChildProcessResource;
  25
  26 class ChildProcessResourceProvider
  27     : public ResourceProvider,
  28       public content::BrowserChildProcessObserver {
  29  public:
  30   explicit ChildProcessResourceProvider(TaskManager* task_manager);
  31
  32   virtual Resource* GetResource(int origin_pid,
  33                                 int child_id,
  34                                 int route_id) OVERRIDE;
  35   virtual void StartUpdating() OVERRIDE;
  36   virtual void StopUpdating() OVERRIDE;
  37
  38   // content::BrowserChildProcessObserver methods:
  39   virtual void BrowserChildProcessHostConnected(
  40       const content::ChildProcessData& data) OVERRIDE;
  41   virtual void BrowserChildProcessHostDisconnected(
  42       const content::ChildProcessData& data) OVERRIDE;
  43
  44  private:
  45   virtual ~ChildProcessResourceProvider();
  46
  47   // Retrieves information about the running ChildProcessHosts (performed in the
  48   // IO thread).
  49   virtual void RetrieveChildProcessData();
  50
  51   // Notifies the UI thread that the ChildProcessHosts information have been
  52   // retrieved.
  53   virtual void ChildProcessDataRetreived(
  54       const std::vector<content::ChildProcessData>& child_processes);
  55
  56   void AddToTaskManager(const content::ChildProcessData& child_process_data);
  57
  58   TaskManager* task_manager_;
  59
  60   // Whether we are currently reporting to the task manager. Used to ignore
  61   // notifications sent after StopUpdating().
  62   bool updating_;
  63
  64   // Maps the actual resources (the ChildProcessData) to the Task Manager
  65   // resources.
  66   typedef std::map<base::ProcessHandle, ChildProcessResource*>
  67       ChildProcessMap;
  68   ChildProcessMap resources_;
  69
  70   // Maps the pids to the resources (used for quick access to the resource on
  71   // byte read notifications).
  72   typedef std::map<int, ChildProcessResource*> PidResourceMap;
  73   PidResourceMap pid_to_resources_;
  74
  75   // A scoped container for notification registries.
  76   content::NotificationRegistrar registrar_;
  77
  78   DISALLOW_COPY_AND_ASSIGN(ChildProcessResourceProvider);
  79 };
  80
  81 }  // namespace task_manager
  82
  83 #endif  // CHROME_BROWSER_TASK_MANAGER_CHILD_PROCESS_RESOURCE_PROVIDER_H_