| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/memory/memory_pressure_listener.h" | 5 #include "base/memory/memory_pressure_listener.h" |
| 6 | 6 |
| 7 #include "base/lazy_instance.h" | 7 #include "base/lazy_instance.h" |
| 8 #include "base/observer_list_threadsafe.h" | 8 #include "base/observer_list_threadsafe.h" |
| 9 #include "base/trace_event/trace_event.h" | 9 #include "base/trace_event/trace_event.h" |
| 10 | 10 |
| 11 namespace base { | 11 namespace base { |
| 12 | 12 |
| 13 namespace { | 13 namespace { |
| 14 | 14 |
| 15 // ObserverListThreadSafe is RefCountedThreadSafe, this traits is needed | 15 class MemoryPressureObserver { |
| 16 // to ensure the LazyInstance will hold a reference to it. | 16 public: |
| 17 struct LeakyLazyObserverListTraits : | 17 MemoryPressureObserver() |
| 18 base::internal::LeakyLazyInstanceTraits< | 18 : async_observers_(new ObserverListThreadSafe<MemoryPressureListener>), |
| 19 ObserverListThreadSafe<MemoryPressureListener> > { | 19 sync_observers_(new ObserverList<MemoryPressureListener>) { |
| 20 static ObserverListThreadSafe<MemoryPressureListener>* | |
| 21 New(void* instance) { | |
| 22 ObserverListThreadSafe<MemoryPressureListener>* ret = | |
| 23 base::internal::LeakyLazyInstanceTraits< | |
| 24 ObserverListThreadSafe<MemoryPressureListener>>::New(instance); | |
| 25 // Leaky. | |
| 26 ret->AddRef(); | |
| 27 return ret; | |
| 28 } | 20 } |
| 21 |
| 22 void AddObserver(MemoryPressureListener* listener, bool sync) { |
| 23 async_observers_->AddObserver(listener); |
| 24 if (sync) { |
| 25 AutoLock lock(sync_observers_lock_); |
| 26 sync_observers_->AddObserver(listener); |
| 27 } |
| 28 } |
| 29 |
| 30 void RemoveObserver(MemoryPressureListener* listener) { |
| 31 async_observers_->RemoveObserver(listener); |
| 32 AutoLock lock(sync_observers_lock_); |
| 33 sync_observers_->RemoveObserver(listener); |
| 34 } |
| 35 |
| 36 void Notify(MemoryPressureListener::MemoryPressureLevel |
| 37 memory_pressure_level) { |
| 38 async_observers_->Notify(FROM_HERE, |
| 39 &MemoryPressureListener::Notify, memory_pressure_level); |
| 40 AutoLock lock(sync_observers_lock_); |
| 41 FOR_EACH_OBSERVER(MemoryPressureListener, *sync_observers_, |
| 42 MemoryPressureListener::SyncNotify(memory_pressure_level)); |
| 43 } |
| 44 |
| 45 private: |
| 46 scoped_refptr<ObserverListThreadSafe<MemoryPressureListener>> |
| 47 async_observers_; |
| 48 ObserverList<MemoryPressureListener>* sync_observers_; |
| 49 Lock sync_observers_lock_; |
| 50 |
| 51 DISALLOW_COPY_AND_ASSIGN(MemoryPressureObserver); |
| 29 }; | 52 }; |
| 30 | 53 |
| 31 LazyInstance< | 54 LazyInstance<MemoryPressureObserver>::Leaky g_observer = |
| 32 ObserverListThreadSafe<MemoryPressureListener>, | 55 LAZY_INSTANCE_INITIALIZER; |
| 33 LeakyLazyObserverListTraits> g_observers = LAZY_INSTANCE_INITIALIZER; | |
| 34 | 56 |
| 35 // All memory pressure notifications within this process will be suppressed if | |
| 36 // this variable is set to 1. | |
| 37 subtle::Atomic32 g_notifications_suppressed = 0; | 57 subtle::Atomic32 g_notifications_suppressed = 0; |
| 38 | 58 |
| 39 } // namespace | 59 } // namespace |
| 40 | 60 |
| 41 MemoryPressureListener::MemoryPressureListener( | 61 MemoryPressureListener::MemoryPressureListener( |
| 42 const MemoryPressureListener::MemoryPressureCallback& callback) | 62 const MemoryPressureListener::MemoryPressureCallback& callback) |
| 43 : callback_(callback) { | 63 : callback_(callback) { |
| 44 g_observers.Get().AddObserver(this); | 64 g_observer.Get().AddObserver(this, false); |
| 65 } |
| 66 |
| 67 MemoryPressureListener::MemoryPressureListener( |
| 68 const MemoryPressureListener::MemoryPressureCallback& callback, |
| 69 const MemoryPressureListener::SyncMemoryPressureCallback& |
| 70 sync_memory_pressure_callback) |
| 71 : callback_(callback), |
| 72 sync_memory_pressure_callback_(sync_memory_pressure_callback) { |
| 73 g_observer.Get().AddObserver(this, true); |
| 45 } | 74 } |
| 46 | 75 |
| 47 MemoryPressureListener::~MemoryPressureListener() { | 76 MemoryPressureListener::~MemoryPressureListener() { |
| 48 g_observers.Get().RemoveObserver(this); | 77 g_observer.Get().RemoveObserver(this); |
| 49 } | 78 } |
| 50 | 79 |
| 51 void MemoryPressureListener::Notify(MemoryPressureLevel memory_pressure_level) { | 80 void MemoryPressureListener::Notify(MemoryPressureLevel memory_pressure_level) { |
| 52 callback_.Run(memory_pressure_level); | 81 callback_.Run(memory_pressure_level); |
| 53 } | 82 } |
| 54 | 83 |
| 84 void MemoryPressureListener::SyncNotify( |
| 85 MemoryPressureLevel memory_pressure_level) { |
| 86 if (!sync_memory_pressure_callback_.is_null()) { |
| 87 sync_memory_pressure_callback_.Run(memory_pressure_level); |
| 88 } |
| 89 } |
| 90 |
| 55 // static | 91 // static |
| 56 void MemoryPressureListener::NotifyMemoryPressure( | 92 void MemoryPressureListener::NotifyMemoryPressure( |
| 57 MemoryPressureLevel memory_pressure_level) { | 93 MemoryPressureLevel memory_pressure_level) { |
| 58 DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE); | 94 DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE); |
| 59 TRACE_EVENT_INSTANT1(TRACE_DISABLED_BY_DEFAULT("memory-infra"), | 95 TRACE_EVENT_INSTANT1(TRACE_DISABLED_BY_DEFAULT("memory-infra"), |
| 60 "MemoryPressureListener::NotifyMemoryPressure", | 96 "MemoryPressureListener::NotifyMemoryPressure", |
| 61 TRACE_EVENT_SCOPE_THREAD, "level", | 97 TRACE_EVENT_SCOPE_THREAD, "level", |
| 62 memory_pressure_level); | 98 memory_pressure_level); |
| 63 if (AreNotificationsSuppressed()) | 99 if (AreNotificationsSuppressed()) |
| 64 return; | 100 return; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 79 void MemoryPressureListener::SimulatePressureNotification( | 115 void MemoryPressureListener::SimulatePressureNotification( |
| 80 MemoryPressureLevel memory_pressure_level) { | 116 MemoryPressureLevel memory_pressure_level) { |
| 81 // Notify all listeners even if regular pressure notifications are suppressed. | 117 // Notify all listeners even if regular pressure notifications are suppressed. |
| 82 DoNotifyMemoryPressure(memory_pressure_level); | 118 DoNotifyMemoryPressure(memory_pressure_level); |
| 83 } | 119 } |
| 84 | 120 |
| 85 // static | 121 // static |
| 86 void MemoryPressureListener::DoNotifyMemoryPressure( | 122 void MemoryPressureListener::DoNotifyMemoryPressure( |
| 87 MemoryPressureLevel memory_pressure_level) { | 123 MemoryPressureLevel memory_pressure_level) { |
| 88 DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE); | 124 DCHECK_NE(memory_pressure_level, MEMORY_PRESSURE_LEVEL_NONE); |
| 89 g_observers.Get().Notify(FROM_HERE, &MemoryPressureListener::Notify, | 125 |
| 90 memory_pressure_level); | 126 g_observer.Get().Notify(memory_pressure_level); |
| 91 } | 127 } |
| 92 | 128 |
| 93 } // namespace base | 129 } // namespace base |
| OLD | NEW |