123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- // Copyright 2013 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "base/memory/memory_pressure_listener.h"
- #include <atomic>
- #include "base/observer_list.h"
- #include "base/observer_list_threadsafe.h"
- #include "base/threading/sequenced_task_runner_handle.h"
- #include "base/trace_event/base_tracing.h"
- #include "base/tracing_buildflags.h"
- #if BUILDFLAG(ENABLE_BASE_TRACING)
- #include "base/trace_event/memory_pressure_level_proto.h" // no-presubmit-check
- #endif
- namespace base {
- namespace {
- // This class is thread safe and internally synchronized.
- class MemoryPressureObserver {
- public:
- // There is at most one MemoryPressureObserver and it is never deleted.
- ~MemoryPressureObserver() = delete;
- void AddObserver(MemoryPressureListener* listener, bool sync) {
- // TODO(crbug.com/1063868): DCHECK instead of silently failing when a
- // MemoryPressureListener is created in a non-sequenced context. Tests will
- // need to be adjusted for that to work.
- if (SequencedTaskRunnerHandle::IsSet()) {
- async_observers_->AddObserver(listener);
- }
- if (sync) {
- AutoLock lock(sync_observers_lock_);
- sync_observers_.AddObserver(listener);
- }
- }
- void RemoveObserver(MemoryPressureListener* listener) {
- async_observers_->RemoveObserver(listener);
- AutoLock lock(sync_observers_lock_);
- sync_observers_.RemoveObserver(listener);
- }
- void Notify(
- MemoryPressureListener::MemoryPressureLevel memory_pressure_level) {
- async_observers_->Notify(FROM_HERE, &MemoryPressureListener::Notify,
- memory_pressure_level);
- AutoLock lock(sync_observers_lock_);
- for (auto& observer : sync_observers_)
- observer.SyncNotify(memory_pressure_level);
- }
- private:
- const scoped_refptr<ObserverListThreadSafe<MemoryPressureListener>>
- async_observers_ =
- base::MakeRefCounted<ObserverListThreadSafe<MemoryPressureListener>>(
- ObserverListPolicy::EXISTING_ONLY);
- ObserverList<MemoryPressureListener>::Unchecked sync_observers_;
- Lock sync_observers_lock_;
- };
- // Gets the shared MemoryPressureObserver singleton instance.
- MemoryPressureObserver* GetMemoryPressureObserver() {
- static auto* const observer = new MemoryPressureObserver();
- return observer;
- }
- std::atomic<bool> g_notifications_suppressed;
- } // namespace
- MemoryPressureListener::MemoryPressureListener(
- const base::Location& creation_location,
- const MemoryPressureListener::MemoryPressureCallback& callback)
- : callback_(callback), creation_location_(creation_location) {
- GetMemoryPressureObserver()->AddObserver(this, false);
- }
- MemoryPressureListener::MemoryPressureListener(
- const base::Location& creation_location,
- const MemoryPressureListener::MemoryPressureCallback& callback,
- const MemoryPressureListener::SyncMemoryPressureCallback&
- sync_memory_pressure_callback)
- : callback_(callback),
- sync_memory_pressure_callback_(sync_memory_pressure_callback),
- creation_location_(creation_location) {
- GetMemoryPressureObserver()->AddObserver(this, true);
- }
- MemoryPressureListener::~MemoryPressureListener() {
- GetMemoryPressureObserver()->RemoveObserver(this);
- }
- void MemoryPressureListener::Notify(MemoryPressureLevel memory_pressure_level) {
- TRACE_EVENT(
- "base", "MemoryPressureListener::Notify",
- [&](perfetto::EventContext ctx) {
- auto* event = ctx.event<perfetto::protos::pbzero::ChromeTrackEvent>();
- auto* data = event->set_chrome_memory_pressure_notification();
- data->set_level(
- trace_event::MemoryPressureLevelToTraceEnum(memory_pressure_level));
- data->set_creation_location_iid(
- base::trace_event::InternedSourceLocation::Get(&ctx,
- creation_location_));
- });
- callback_.Run(memory_pressure_level);
- }
- void MemoryPressureListener::SyncNotify(
- MemoryPressureLevel memory_pressure_level) {
- if (!sync_memory_pressure_callback_.is_null()) {
- sync_memory_pressure_callback_.Run(memory_pressure_level);
- }
- }
- // static
- void MemoryPressureListener::NotifyMemoryPressure(
- MemoryPressureLevel memory_pressure_level) {
- DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE);
- TRACE_EVENT_INSTANT(
- trace_event::MemoryDumpManager::kTraceCategory,
- "MemoryPressureListener::NotifyMemoryPressure",
- [&](perfetto::EventContext ctx) {
- auto* event = ctx.event<perfetto::protos::pbzero::ChromeTrackEvent>();
- auto* data = event->set_chrome_memory_pressure_notification();
- data->set_level(
- trace_event::MemoryPressureLevelToTraceEnum(memory_pressure_level));
- });
- if (AreNotificationsSuppressed())
- return;
- DoNotifyMemoryPressure(memory_pressure_level);
- }
- // static
- bool MemoryPressureListener::AreNotificationsSuppressed() {
- return g_notifications_suppressed.load(std::memory_order_acquire);
- }
- // static
- void MemoryPressureListener::SetNotificationsSuppressed(bool suppress) {
- g_notifications_suppressed.store(suppress, std::memory_order_release);
- }
- // static
- void MemoryPressureListener::SimulatePressureNotification(
- MemoryPressureLevel memory_pressure_level) {
- // Notify all listeners even if regular pressure notifications are suppressed.
- DoNotifyMemoryPressure(memory_pressure_level);
- }
- // static
- void MemoryPressureListener::DoNotifyMemoryPressure(
- MemoryPressureLevel memory_pressure_level) {
- DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE);
- GetMemoryPressureObserver()->Notify(memory_pressure_level);
- }
- } // namespace base
|