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/discardable_memory_provider.h" | 5 #include "base/memory/discardable_memory_provider.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/containers/hash_tables.h" | 8 #include "base/containers/hash_tables.h" |
9 #include "base/containers/mru_cache.h" | 9 #include "base/containers/mru_cache.h" |
10 #include "base/debug/trace_event.h" | 10 #include "base/debug/trace_event.h" |
(...skipping 11 matching lines...) Expand all Loading... |
22 static const size_t kDefaultBytesToReclaimUnderModeratePressure = | 22 static const size_t kDefaultBytesToReclaimUnderModeratePressure = |
23 kDefaultDiscardableMemoryLimit / 2; | 23 kDefaultDiscardableMemoryLimit / 2; |
24 | 24 |
25 } // namespace | 25 } // namespace |
26 | 26 |
27 DiscardableMemoryProvider::DiscardableMemoryProvider() | 27 DiscardableMemoryProvider::DiscardableMemoryProvider() |
28 : allocations_(AllocationMap::NO_AUTO_EVICT), | 28 : allocations_(AllocationMap::NO_AUTO_EVICT), |
29 bytes_allocated_(0), | 29 bytes_allocated_(0), |
30 discardable_memory_limit_(kDefaultDiscardableMemoryLimit), | 30 discardable_memory_limit_(kDefaultDiscardableMemoryLimit), |
31 bytes_to_reclaim_under_moderate_pressure_( | 31 bytes_to_reclaim_under_moderate_pressure_( |
32 kDefaultBytesToReclaimUnderModeratePressure) { | 32 kDefaultBytesToReclaimUnderModeratePressure), |
| 33 memory_pressure_listener_( |
| 34 base::Bind(&DiscardableMemoryProvider::NotifyMemoryPressure, |
| 35 Unretained(this))) { |
33 } | 36 } |
34 | 37 |
35 DiscardableMemoryProvider::~DiscardableMemoryProvider() { | 38 DiscardableMemoryProvider::~DiscardableMemoryProvider() { |
36 DCHECK(allocations_.empty()); | 39 DCHECK(allocations_.empty()); |
37 DCHECK_EQ(0u, bytes_allocated_); | 40 DCHECK_EQ(0u, bytes_allocated_); |
38 } | 41 } |
39 | 42 |
40 void DiscardableMemoryProvider::RegisterMemoryPressureListener() { | 43 void DiscardableMemoryProvider::NotifyMemoryPressure( |
41 AutoLock lock(lock_); | 44 MemoryPressureListener::MemoryPressureLevel pressure_level) { |
42 DCHECK(base::MessageLoop::current()); | 45 switch (pressure_level) { |
43 DCHECK(!memory_pressure_listener_); | 46 case MemoryPressureListener::MEMORY_PRESSURE_MODERATE: |
44 memory_pressure_listener_.reset( | 47 Purge(); |
45 new MemoryPressureListener( | 48 return; |
46 base::Bind(&DiscardableMemoryProvider::OnMemoryPressure, | 49 case MemoryPressureListener::MEMORY_PRESSURE_CRITICAL: |
47 Unretained(this)))); | 50 PurgeAll(); |
48 } | 51 return; |
| 52 } |
49 | 53 |
50 void DiscardableMemoryProvider::UnregisterMemoryPressureListener() { | 54 NOTREACHED(); |
51 AutoLock lock(lock_); | |
52 DCHECK(memory_pressure_listener_); | |
53 memory_pressure_listener_.reset(); | |
54 } | 55 } |
55 | 56 |
56 void DiscardableMemoryProvider::SetDiscardableMemoryLimit(size_t bytes) { | 57 void DiscardableMemoryProvider::SetDiscardableMemoryLimit(size_t bytes) { |
57 AutoLock lock(lock_); | 58 AutoLock lock(lock_); |
58 discardable_memory_limit_ = bytes; | 59 discardable_memory_limit_ = bytes; |
59 EnforcePolicyWithLockAcquired(); | 60 EnforcePolicyWithLockAcquired(); |
60 } | 61 } |
61 | 62 |
62 void DiscardableMemoryProvider::SetBytesToReclaimUnderModeratePressure( | 63 void DiscardableMemoryProvider::SetBytesToReclaimUnderModeratePressure( |
63 size_t bytes) { | 64 size_t bytes) { |
64 AutoLock lock(lock_); | 65 AutoLock lock(lock_); |
65 bytes_to_reclaim_under_moderate_pressure_ = bytes; | 66 bytes_to_reclaim_under_moderate_pressure_ = bytes; |
66 } | 67 } |
67 | 68 |
68 void DiscardableMemoryProvider::Register( | 69 void DiscardableMemoryProvider::Register( |
69 const DiscardableMemory* discardable, size_t bytes) { | 70 const DiscardableMemory* discardable, size_t bytes) { |
70 AutoLock lock(lock_); | 71 AutoLock lock(lock_); |
71 // A registered memory listener is currently required. This DCHECK can be | |
72 // moved or removed if we decide that it's useful to relax this condition. | |
73 DCHECK(memory_pressure_listener_); | |
74 DCHECK(allocations_.Peek(discardable) == allocations_.end()); | 72 DCHECK(allocations_.Peek(discardable) == allocations_.end()); |
75 allocations_.Put(discardable, Allocation(bytes)); | 73 allocations_.Put(discardable, Allocation(bytes)); |
76 } | 74 } |
77 | 75 |
78 void DiscardableMemoryProvider::Unregister( | 76 void DiscardableMemoryProvider::Unregister( |
79 const DiscardableMemory* discardable) { | 77 const DiscardableMemory* discardable) { |
80 AutoLock lock(lock_); | 78 AutoLock lock(lock_); |
81 AllocationMap::iterator it = allocations_.Peek(discardable); | 79 AllocationMap::iterator it = allocations_.Peek(discardable); |
82 if (it == allocations_.end()) | 80 if (it == allocations_.end()) |
83 return; | 81 return; |
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
164 AutoLock lock(lock_); | 162 AutoLock lock(lock_); |
165 AllocationMap::const_iterator it = allocations_.Peek(discardable); | 163 AllocationMap::const_iterator it = allocations_.Peek(discardable); |
166 return it != allocations_.end() && it->second.memory; | 164 return it != allocations_.end() && it->second.memory; |
167 } | 165 } |
168 | 166 |
169 size_t DiscardableMemoryProvider::GetBytesAllocatedForTest() const { | 167 size_t DiscardableMemoryProvider::GetBytesAllocatedForTest() const { |
170 AutoLock lock(lock_); | 168 AutoLock lock(lock_); |
171 return bytes_allocated_; | 169 return bytes_allocated_; |
172 } | 170 } |
173 | 171 |
174 void DiscardableMemoryProvider::OnMemoryPressure( | |
175 MemoryPressureListener::MemoryPressureLevel pressure_level) { | |
176 switch (pressure_level) { | |
177 case MemoryPressureListener::MEMORY_PRESSURE_MODERATE: | |
178 Purge(); | |
179 return; | |
180 case MemoryPressureListener::MEMORY_PRESSURE_CRITICAL: | |
181 PurgeAll(); | |
182 return; | |
183 } | |
184 | |
185 NOTREACHED(); | |
186 } | |
187 | |
188 void DiscardableMemoryProvider::Purge() { | 172 void DiscardableMemoryProvider::Purge() { |
189 AutoLock lock(lock_); | 173 AutoLock lock(lock_); |
190 | 174 |
191 if (bytes_to_reclaim_under_moderate_pressure_ == 0) | 175 if (bytes_to_reclaim_under_moderate_pressure_ == 0) |
192 return; | 176 return; |
193 | 177 |
194 size_t limit = 0; | 178 size_t limit = 0; |
195 if (bytes_to_reclaim_under_moderate_pressure_ < bytes_allocated_) | 179 if (bytes_to_reclaim_under_moderate_pressure_ < bytes_allocated_) |
196 limit = bytes_allocated_ - bytes_to_reclaim_under_moderate_pressure_; | 180 limit = bytes_allocated_ - bytes_to_reclaim_under_moderate_pressure_; |
197 | 181 |
(...skipping 24 matching lines...) Expand all Loading... |
222 it->second.memory = NULL; | 206 it->second.memory = NULL; |
223 } | 207 } |
224 } | 208 } |
225 | 209 |
226 void DiscardableMemoryProvider::EnforcePolicyWithLockAcquired() { | 210 void DiscardableMemoryProvider::EnforcePolicyWithLockAcquired() { |
227 PurgeLRUWithLockAcquiredUntilUsageIsWithin(discardable_memory_limit_); | 211 PurgeLRUWithLockAcquiredUntilUsageIsWithin(discardable_memory_limit_); |
228 } | 212 } |
229 | 213 |
230 } // namespace internal | 214 } // namespace internal |
231 } // namespace base | 215 } // namespace base |
OLD | NEW |