1. 程式人生 > >最簡單的LRU演算法實現,執行緒安全的

最簡單的LRU演算法實現,執行緒安全的

LRU演算法用途之廣就不說了,凡是要用cache的地方都可以見到它的身影。特別是執行緒多,併發高,資料量大的環境下。

jdk1.5真好,在LinkedHashMap.java的原始碼中直接有這樣的字樣、“This kind of map is well-suited to building LRU caches.......The removeEldestEntry(Map.Entry) method may be overridden to impose a policy for removing stale mappings automatically when new mappings are added to the map.” 就是說自己寫一下removeEldestEntry就搞定了。難為我當年自己寫了一個。

以下是程式碼,我增加的是執行緒安全的程式碼
[code]
import java.util.LinkedHashMap;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

public class LRULinkedHashMap<K, V> extends LinkedHashMap<K, V>
{
private final int maxCapacity;
private static final float DEFAULT_LOAD_FACTOR = 0.75f;
private final Lock lock = new ReentrantLock();

public LRULinkedHashMap(int maxCapacity)
{
super(maxCapacity, DEFAULT_LOAD_FACTOR, true);
this.maxCapacity = maxCapacity;
}

@Override
protected boolean removeEldestEntry(java.util.Map.Entry<K, V> eldest)
{
return size() > maxCapacity;
}

@Override
public V get(Object key)
{
try {
lock.lock();
return super.get(key);
}
finally {
lock.unlock();
}
}

@Override
public V put(K key, V value)
{
try {
lock.lock();
return super.put(key, value);
}
finally {
lock.unlock();
}
}
}
[/code]
注意,只有get和put是執行緒安全的。

本來這貼是想放到入門論壇的,不過想了一下,要用LRU演算法的,知道ReentrantLock的人好像也不能算新手了。