File: throttle_service.h

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (112 lines) | stat: -rw-r--r-- 4,156 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROMEOS_ASH_COMPONENTS_THROTTLE_THROTTLE_SERVICE_H_
#define CHROMEOS_ASH_COMPONENTS_THROTTLE_THROTTLE_SERVICE_H_

#include <memory>
#include <optional>
#include <string>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/observer_list.h"
#include "base/time/time.h"
#include "base/component_export.h"
#include "chromeos/ash/components/throttle/throttle_observer.h"

namespace content {
class BrowserContext;
}

namespace ash {

// This class is the base for different throttle services on Chrome OS.
// The class holds a number of ThrottleObservers which watch for several
// conditions. When the observers change from active to inactive or vice-versa,
// OnObserverStateChanged checks if there is an active observer and calls
// ThrottleInstance accordingly.
class COMPONENT_EXPORT(CHROMEOS_ASH_COMPONENTS_THROTTLE) ThrottleService {
 public:
  class ServiceObserver : public base::CheckedObserver {
   public:
    // Notifies that throttling has been changed. If the target is now
    // throttled, the function is called with true. It is called with
    // false otherwise.
    virtual void OnThrottle(bool throttled) = 0;
  };

  explicit ThrottleService(content::BrowserContext* context);

  ThrottleService(const ThrottleService&) = delete;
  ThrottleService& operator=(const ThrottleService&) = delete;

  virtual ~ThrottleService();

  void AddServiceObserver(ServiceObserver* observer);
  void RemoveServiceObserver(ServiceObserver* observer);

  // Returns an observer whose name is |name|. Returns nullptr otherwise.
  ThrottleObserver* GetObserverByName(const std::string& name);

  // Functions for testing
  void NotifyObserverStateChangedForTesting();
  void SetObserversForTesting(
      std::vector<std::unique_ptr<ThrottleObserver>> observers);
  bool HasServiceObserverForTesting(ServiceObserver* candidate);

  bool should_throttle() const {
    // When `should_throttle_` hasn't been initialized, return true to throttle
    // the target (which is a safer bet for Chrome.)
    return !should_throttle_ || *should_throttle_;
  }

  void reset_should_throttle_for_testing() { should_throttle_.reset(); }
  const std::vector<std::unique_ptr<ThrottleObserver>>& observers_for_testing()
      const {
    return observers_;
  }

 protected:
  void AddObserver(std::unique_ptr<ThrottleObserver> observer);
  void StartObservers();
  void StopObservers();
  void OnObserverStateChanged(const ThrottleObserver* changed_observer);

  // This function is called whenever the target should be throttled or
  // unthrottled. Derived classes should implement ThrottleInstance to adjust
  // the throttling state of their relevant target.
  virtual void ThrottleInstance(bool should_throttle) = 0;

  // Whenever there is a change in the effective observer, this function is
  // called with the name of the previously effective observer and the duration
  // it was effective. Derived classes can implement this function to record UMA
  // metrics.
  virtual void RecordCpuRestrictionDisabledUMA(const std::string& observer_name,
                                               base::TimeDelta delta) {}

  const std::vector<std::unique_ptr<ThrottleObserver>>& observers() const {
    return observers_;
  }
  content::BrowserContext* context() const { return context_; }

 private:
  const raw_ptr<content::BrowserContext> context_;
  std::vector<std::unique_ptr<ThrottleObserver>> observers_;

  // True when the target should be throttled. Use optional<> to make sure this
  // service always calls ThrottleInstance() when one of the observers has
  // changed for the first time.
  std::optional<bool> should_throttle_;

  raw_ptr<ThrottleObserver> last_effective_observer_{nullptr};
  base::TimeTicks last_throttle_transition_;
  base::ObserverList<ServiceObserver> service_observers_;
  base::WeakPtrFactory<ThrottleService> weak_ptr_factory_{this};
};

}  // namespace ash

#endif  // CHROMEOS_ASH_COMPONENTS_THROTTLE_THROTTLE_SERVICE_H_