chrome/browser/task_manager/notification_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_NOTIFICATION_RESOURCE_PROVIDER_H_
   6 #define CHROME_BROWSER_TASK_MANAGER_NOTIFICATION_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/notification_observer.h"
  14 #include "content/public/browser/notification_registrar.h"
  15
  16 class BalloonHost;
  17 class TaskManager;
  18
  19 namespace task_manager {
  20
  21 class NotificationResource;
  22
  23 class NotificationResourceProvider : public ResourceProvider,
  24                                      public content::NotificationObserver {
  25  public:
  26   static NotificationResourceProvider* Create(TaskManager* task_manager);
  27
  28   // ResourceProvider interface
  29   virtual Resource* GetResource(int origin_pid,
  30                                 int child_id,
  31                                 int route_id) OVERRIDE;
  32   virtual void StartUpdating() OVERRIDE;
  33   virtual void StopUpdating() OVERRIDE;
  34
  35   // content::NotificationObserver interface
  36   virtual void Observe(int type,
  37                        const content::NotificationSource& source,
  38                        const content::NotificationDetails& details) OVERRIDE;
  39
  40  private:
  41   explicit NotificationResourceProvider(TaskManager* task_manager);
  42   virtual ~NotificationResourceProvider();
  43
  44   void AddToTaskManager(BalloonHost* balloon_host);
  45   void RemoveFromTaskManager(BalloonHost* balloon_host);
  46
  47   TaskManager* task_manager_;
  48
  49   // Maps the actual resources (BalloonHost*) to the Task Manager resources.
  50   std::map<BalloonHost*, NotificationResource*> resources_;
  51
  52   // A scoped container for notification registries.
  53   content::NotificationRegistrar registrar_;
  54
  55   bool updating_;
  56
  57   DISALLOW_COPY_AND_ASSIGN(NotificationResourceProvider);
  58 };
  59
  60 }  // namespace task_manager
  61
  62 #endif  // CHROME_BROWSER_TASK_MANAGER_NOTIFICATION_RESOURCE_PROVIDER_H_