Avi Drissman | 8ba1bad | 2022-09-13 19:22:36 | [diff] [blame] | 1 | // Copyright 2019 The Chromium Authors |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
Albert J. Wong | 23183a1 | 2021-08-13 00:23:35 | [diff] [blame] | 5 | #include "components/memory_pressure/multi_source_memory_pressure_monitor.h" |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 6 | |
Hans Wennborg | c3cffa6 | 2020-04-27 10:09:12 | [diff] [blame] | 7 | #include "base/check_op.h" |
Avi Drissman | 12be031 | 2023-01-11 09:16:09 | [diff] [blame] | 8 | #include "base/functional/bind.h" |
Sebastien Marchand | 6c5354e | 2020-02-07 19:38:25 | [diff] [blame] | 9 | #include "base/metrics/histogram_functions.h" |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 10 | #include "base/metrics/histogram_macros.h" |
Sebastien Marchand | 6c5354e | 2020-02-07 19:38:25 | [diff] [blame] | 11 | #include "base/time/time.h" |
Eric Seckler | f6c544f | 2020-06-02 10:49:21 | [diff] [blame] | 12 | #include "base/trace_event/base_tracing.h" |
ssid | 8734ea7 | 2021-03-17 18:43:58 | [diff] [blame] | 13 | #include "base/tracing_buildflags.h" |
Albert J. Wong | 23183a1 | 2021-08-13 00:23:35 | [diff] [blame] | 14 | #include "components/memory_pressure/system_memory_pressure_evaluator.h" |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 15 | |
ssid | 8734ea7 | 2021-03-17 18:43:58 | [diff] [blame] | 16 | #if BUILDFLAG(ENABLE_BASE_TRACING) |
| 17 | #include "base/trace_event/memory_pressure_level_proto.h" // no-presubmit-check |
| 18 | #endif |
| 19 | |
Albert J. Wong | 23183a1 | 2021-08-13 00:23:35 | [diff] [blame] | 20 | namespace memory_pressure { |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 21 | |
| 22 | MultiSourceMemoryPressureMonitor::MultiSourceMemoryPressureMonitor() |
| 23 | : current_pressure_level_( |
| 24 | base::MemoryPressureListener::MEMORY_PRESSURE_LEVEL_NONE), |
| 25 | dispatch_callback_(base::BindRepeating( |
| 26 | &base::MemoryPressureListener::NotifyMemoryPressure)), |
Sebastien Marchand | fcd2548 | 2021-05-03 16:19:46 | [diff] [blame] | 27 | aggregator_(this), |
| 28 | level_reporter_(current_pressure_level_) {} |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 29 | |
Jenny Wong | 6592dd4 | 2019-09-26 17:25:07 | [diff] [blame] | 30 | MultiSourceMemoryPressureMonitor::~MultiSourceMemoryPressureMonitor() { |
| 31 | // Destroy system evaluator early while the remaining members of this class |
| 32 | // still exist. MultiSourceMemoryPressureMonitor implements |
| 33 | // MemoryPressureVoteAggregator::Delegate, and |
| 34 | // delegate_->OnMemoryPressureLevelChanged() gets indirectly called during |
| 35 | // ~SystemMemoryPressureEvaluator(). |
| 36 | system_evaluator_.reset(); |
| 37 | } |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 38 | |
Patrick Monette | 216bbdc3 | 2023-02-17 22:27:00 | [diff] [blame] | 39 | void MultiSourceMemoryPressureMonitor::MaybeStartPlatformVoter() { |
Sebastien Marchand | c9d4cd1 | 2019-10-03 22:48:01 | [diff] [blame] | 40 | system_evaluator_ = |
| 41 | SystemMemoryPressureEvaluator::CreateDefaultSystemEvaluator(this); |
Ryan Powell | bdc2759 | 2019-09-04 22:59:39 | [diff] [blame] | 42 | } |
| 43 | |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 44 | base::MemoryPressureListener::MemoryPressureLevel |
| 45 | MultiSourceMemoryPressureMonitor::GetCurrentPressureLevel() const { |
| 46 | DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| 47 | return current_pressure_level_; |
| 48 | } |
| 49 | |
| 50 | std::unique_ptr<MemoryPressureVoter> |
| 51 | MultiSourceMemoryPressureMonitor::CreateVoter() { |
| 52 | DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
Wez | 861e292 | 2020-03-01 21:14:00 | [diff] [blame] | 53 | return aggregator_.CreateVoter(); |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 54 | } |
| 55 | |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 56 | void MultiSourceMemoryPressureMonitor::OnMemoryPressureLevelChanged( |
| 57 | base::MemoryPressureListener::MemoryPressureLevel level) { |
| 58 | DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
Sebastien Marchand | 6c5354e | 2020-02-07 19:38:25 | [diff] [blame] | 59 | DCHECK_NE(current_pressure_level_, level); |
| 60 | |
Sebastien Marchand | fcd2548 | 2021-05-03 16:19:46 | [diff] [blame] | 61 | level_reporter_.OnMemoryPressureLevelChanged(level); |
| 62 | |
ssid | fff7ab4 | 2021-03-04 20:03:11 | [diff] [blame] | 63 | TRACE_EVENT_INSTANT( |
Sebastien Marchand | 9df1f3d | 2020-03-23 18:54:31 | [diff] [blame] | 64 | "base", "MultiSourceMemoryPressureMonitor::OnMemoryPressureLevelChanged", |
ssid | fff7ab4 | 2021-03-04 20:03:11 | [diff] [blame] | 65 | [&](perfetto::EventContext ctx) { |
| 66 | auto* event = ctx.event<perfetto::protos::pbzero::ChromeTrackEvent>(); |
| 67 | auto* data = event->set_chrome_memory_pressure_notification(); |
ssid | 8734ea7 | 2021-03-17 18:43:58 | [diff] [blame] | 68 | data->set_level( |
| 69 | base::trace_event::MemoryPressureLevelToTraceEnum(level)); |
ssid | fff7ab4 | 2021-03-04 20:03:11 | [diff] [blame] | 70 | }); |
Sebastien Marchand | 9df1f3d | 2020-03-23 18:54:31 | [diff] [blame] | 71 | |
Ryan Powell | 30287d4 | 2019-08-15 15:48:32 | [diff] [blame] | 72 | current_pressure_level_ = level; |
| 73 | } |
| 74 | |
| 75 | void MultiSourceMemoryPressureMonitor::OnNotifyListenersRequested() { |
| 76 | DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| 77 | dispatch_callback_.Run(current_pressure_level_); |
| 78 | } |
| 79 | |
Ryan Powell | 00e61aea | 2019-09-04 18:33:44 | [diff] [blame] | 80 | void MultiSourceMemoryPressureMonitor::SetSystemEvaluator( |
| 81 | std::unique_ptr<SystemMemoryPressureEvaluator> evaluator) { |
| 82 | DCHECK(!system_evaluator_); |
| 83 | system_evaluator_ = std::move(evaluator); |
| 84 | } |
| 85 | |
Patrick Monette | 216bbdc3 | 2023-02-17 22:27:00 | [diff] [blame] | 86 | void MultiSourceMemoryPressureMonitor::SetDispatchCallbackForTesting( |
| 87 | const DispatchCallback& callback) { |
| 88 | DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| 89 | // Must be called before `Start()`. |
| 90 | DCHECK(!system_evaluator_); |
| 91 | dispatch_callback_ = callback; |
| 92 | } |
| 93 | |
Albert J. Wong | 23183a1 | 2021-08-13 00:23:35 | [diff] [blame] | 94 | } // namespace memory_pressure |