最簡單的LRU演算法java實現

yangxi_001發表於2013-11-26
演算法Java
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就搞定了。
以下是程式碼，增加的是執行緒安全的程式碼

  1. import java.util.LinkedHashMap;  
  2. import java.util.concurrent.locks.Lock;  
  3. import java.util.concurrent.locks.ReentrantLock;  
  4.   
  5. public class LRULinkedHashMap<K, V> extends LinkedHashMap<K, V>  
  6. {  
  7.     private final int maxCapacity;  
  8.     private static final float DEFAULT_LOAD_FACTOR = 0.75f;  
  9.     private final Lock lock = new ReentrantLock();  
  10.   
  11.     public LRULinkedHashMap(int maxCapacity)  
  12.     {  
  13.         super(maxCapacity, DEFAULT_LOAD_FACTOR, true);  
  14.         this.maxCapacity = maxCapacity;  
  15.     }  
  16.   
  17.     @Override  
  18.     protected boolean removeEldestEntry(java.util.Map.Entry<K, V> eldest)  
  19.     {  
  20.         return size() > maxCapacity;  
  21.     }  
  22.   
  23.     @Override  
  24.     public V get(Object key)  
  25.     {  
  26.         try {  
  27.             lock.lock();  
  28.             return super.get(key);  
  29.         }  
  30.         finally {  
  31.             lock.unlock();  
  32.         }  
  33.     }  
  34.   
  35.     @Override  
  36.     public V put(K key, V value)  
  37.     {  
  38.         try {  
  39.             lock.lock();  
  40.             return super.put(key, value);  
  41.         }  
  42.         finally {  
  43.             lock.unlock();  
  44.         }  
  45.     }  
  46. }  

相關文章