Browse Source

Fix memory leak in ConcurrentReferenceHashMap

Update ConcurrentReferenceHashMap to protect against references that
have been garbage collected but for some reason do not appear as a
`pollForPurge` result.

Also added purgeUnreferencedEntries() method to allow for programmatic
cleanup.

Issue: SPR-11440
(cherry picked from commit 2b4c81e6)
pull/475/head
Phillip Webb 12 years ago
parent
commit
09c57203bb
  1. 21
      spring-core/src/main/java/org/springframework/util/ConcurrentReferenceHashMap.java

21
spring-core/src/main/java/org/springframework/util/ConcurrentReferenceHashMap.java

@ -334,6 +334,19 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -334,6 +334,19 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
}
}
/**
* Remove any entries that have been garbage collected and are no longer referenced.
* Under normal circumstances garbage collected entries are automatically purged as
* items are added or removed from the Map. This method can be used to force a purge,
* and is useful when the Map is read frequently but updated less often.
*/
public void purgeUnreferencedEntries() {
for (Segment segment : this.segments) {
segment.restructureIfNecessary(false);
}
}
@Override
public int size() {
int size = 0;
@ -508,7 +521,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -508,7 +521,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
* references that have been garbage collected.
* @param allowResize if resizing is permitted
*/
private void restructureIfNecessary(boolean allowResize) {
protected final void restructureIfNecessary(boolean allowResize) {
boolean needsResize = ((this.count > 0) && (this.count >= this.resizeThreshold));
Reference<K, V> reference = this.referenceManager.pollForPurge();
if ((reference != null) || (needsResize && allowResize)) {
@ -546,7 +559,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -546,7 +559,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
restructured[i] = null;
}
while (reference != null) {
if (!toPurge.contains(reference)) {
if (!toPurge.contains(reference) && (reference.get() != null)) {
int index = getIndex(reference.getHash(), restructured);
restructured[index] = this.referenceManager.createReference(
reference.get(), reference.getHash(),
@ -560,7 +573,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -560,7 +573,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
if (resizing) {
setReferences(restructured);
}
this.count = countAfterRestructure;
this.count = Math.max(countAfterRestructure, 0);
} finally {
unlock();
}
@ -961,6 +974,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -961,6 +974,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
enqueue();
clear();
}
}
@ -991,6 +1005,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen @@ -991,6 +1005,7 @@ public class ConcurrentReferenceHashMap<K, V> extends AbstractMap<K, V> implemen
enqueue();
clear();
}
}
}

Loading…
Cancel
Save