tab_helper_frame_node_source.h 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. // Copyright 2019 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. #ifndef COMPONENTS_PERFORMANCE_MANAGER_TAB_HELPER_FRAME_NODE_SOURCE_H_
  5. #define COMPONENTS_PERFORMANCE_MANAGER_TAB_HELPER_FRAME_NODE_SOURCE_H_
  6. #include "components/performance_manager/frame_node_source.h"
  7. #include "base/observer_list_types.h"
  8. #include "base/scoped_multi_source_observation.h"
  9. #include "components/performance_manager/performance_manager_tab_helper.h"
  10. namespace performance_manager {
  11. // An implementation of FrameNodeSource that uses PerformanceManagerTabHelper to
  12. // get frame node information.
  13. class TabHelperFrameNodeSource : public FrameNodeSource,
  14. public PerformanceManagerTabHelper::Observer {
  15. public:
  16. TabHelperFrameNodeSource();
  17. TabHelperFrameNodeSource(const TabHelperFrameNodeSource&) = delete;
  18. TabHelperFrameNodeSource& operator=(const TabHelperFrameNodeSource&) = delete;
  19. ~TabHelperFrameNodeSource() override;
  20. // FrameNodeSource:
  21. FrameNodeImpl* GetFrameNode(
  22. content::GlobalRenderFrameHostId render_process_host_id) override;
  23. void SubscribeToFrameNode(
  24. content::GlobalRenderFrameHostId render_process_host_id,
  25. OnbeforeFrameNodeRemovedCallback on_before_frame_node_removed_callback)
  26. override;
  27. void UnsubscribeFromFrameNode(
  28. content::GlobalRenderFrameHostId render_process_host_id) override;
  29. // PerformanceManagerTabHelper::Observer:
  30. void OnBeforeFrameNodeRemoved(
  31. PerformanceManagerTabHelper* performance_manager_tab_helper,
  32. FrameNodeImpl* frame_node) override;
  33. private:
  34. // Adds |frame_node| to the set of observed frame nodes associated with
  35. // |performance_manager_tab_helper|. Returns true if |frame_node| was the
  36. // first frame added to that set.
  37. bool AddObservedFrameNode(
  38. PerformanceManagerTabHelper* performance_manager_tab_helper,
  39. FrameNodeImpl* frame_node);
  40. // Removes |frame_node| to the set of observed frame nodes associated with
  41. // |performance_manager_tab_helper|. Returns true if |frame_node| was the
  42. // last frame removed from that set.
  43. bool RemoveObservedFrameNode(
  44. PerformanceManagerTabHelper* performance_manager_tab_helper,
  45. FrameNodeImpl* frame_node);
  46. // Maps each observed frame node to their callback.
  47. base::flat_map<FrameNodeImpl*, OnbeforeFrameNodeRemovedCallback>
  48. frame_node_callbacks_;
  49. // Maps each tab helper to the set of observed frame nodes that belongs to
  50. // that tab helper.
  51. base::flat_map<PerformanceManagerTabHelper*, base::flat_set<FrameNodeImpl*>>
  52. observed_frame_nodes_;
  53. // Observes frame node deletions.
  54. base::ScopedMultiSourceObservation<PerformanceManagerTabHelper,
  55. PerformanceManagerTabHelper::Observer>
  56. performance_manager_tab_helper_observations_;
  57. };
  58. } // namespace performance_manager
  59. #endif // COMPONENTS_PERFORMANCE_MANAGER_TAB_HELPER_FRAME_NODE_SOURCE_H_