183 lines
4.8 KiB
Java
183 lines
4.8 KiB
Java
package DataStructures.Caches;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Least recently used (LRU)
|
|
* <p>
|
|
* Discards the least recently used items first.
|
|
* This algorithm requires keeping track of what was used when,
|
|
* which is expensive if one wants to make sure the algorithm always discards
|
|
* the least recently used item.
|
|
* https://en.wikipedia.org/wiki/Cache_replacement_policies#Least_recently_used_(LRU)
|
|
*
|
|
* @param <K> key type
|
|
* @param <V> value type
|
|
*/
|
|
public class LRUCache<K, V> {
|
|
private final Map<K, Entry<K, V>> data = new HashMap<>();
|
|
private Entry<K, V> head;
|
|
private Entry<K, V> tail;
|
|
private int cap;
|
|
private static final int DEFAULT_CAP = 100;
|
|
|
|
public LRUCache() {
|
|
setCapacity(DEFAULT_CAP);
|
|
}
|
|
|
|
public LRUCache(int cap) {
|
|
setCapacity(cap);
|
|
}
|
|
|
|
private void setCapacity(int newCapacity) {
|
|
checkCapacity(newCapacity);
|
|
for (int i = data.size(); i > newCapacity; i--) {
|
|
Entry<K, V> evicted = evict();
|
|
data.remove(evicted.getKey());
|
|
}
|
|
this.cap = newCapacity;
|
|
}
|
|
|
|
private Entry<K, V> evict() {
|
|
if (head == null) {
|
|
throw new RuntimeException("cache cannot be empty!");
|
|
}
|
|
Entry<K, V> evicted = head;
|
|
head = evicted.getNextEntry();
|
|
head.setPreEntry(null);
|
|
evicted.setNextEntry(null);
|
|
return evicted;
|
|
}
|
|
|
|
private void checkCapacity(int capacity) {
|
|
if (capacity <= 0) {
|
|
throw new RuntimeException("capacity must greater than 0!");
|
|
}
|
|
}
|
|
|
|
public V get(K key) {
|
|
if (!data.containsKey(key)) {
|
|
return null;
|
|
}
|
|
final Entry<K, V> entry = data.get(key);
|
|
moveNodeToLast(entry);
|
|
return entry.getValue();
|
|
}
|
|
|
|
private void moveNodeToLast(Entry<K, V> entry) {
|
|
if (tail == entry) {
|
|
return;
|
|
}
|
|
final Entry<K, V> preEntry = entry.getPreEntry();
|
|
final Entry<K, V> nextEntry = entry.getNextEntry();
|
|
if (preEntry != null) {
|
|
preEntry.setNextEntry(nextEntry);
|
|
}
|
|
if (nextEntry != null) {
|
|
nextEntry.setPreEntry(preEntry);
|
|
}
|
|
if (head == entry) {
|
|
head = nextEntry;
|
|
}
|
|
tail.setNextEntry(entry);
|
|
entry.setPreEntry(tail);
|
|
entry.setNextEntry(null);
|
|
tail = entry;
|
|
}
|
|
|
|
public void put(K key, V value) {
|
|
if (data.containsKey(key)) {
|
|
final Entry<K, V> existingEntry = data.get(key);
|
|
existingEntry.setValue(value);
|
|
moveNodeToLast(existingEntry);
|
|
return;
|
|
}
|
|
Entry<K, V> newEntry;
|
|
if (data.size() == cap) {
|
|
newEntry = evict();
|
|
data.remove(newEntry.getKey());
|
|
} else {
|
|
newEntry = new Entry<>();
|
|
}
|
|
|
|
newEntry.setKey(key);
|
|
newEntry.setValue(value);
|
|
addNewEntry(newEntry);
|
|
data.put(key, newEntry);
|
|
}
|
|
|
|
private void addNewEntry(Entry<K, V> newEntry) {
|
|
if (data.isEmpty()) {
|
|
head = newEntry;
|
|
tail = newEntry;
|
|
return;
|
|
}
|
|
tail.setNextEntry(newEntry);
|
|
newEntry.setPreEntry(tail);
|
|
newEntry.setNextEntry(null);
|
|
tail = newEntry;
|
|
}
|
|
|
|
static final class Entry<I, J> {
|
|
private Entry<I, J> preEntry;
|
|
private Entry<I, J> nextEntry;
|
|
private I key;
|
|
private J value;
|
|
|
|
public Entry() {
|
|
}
|
|
|
|
public Entry(Entry<I, J> preEntry, Entry<I, J> nextEntry, I key, J value) {
|
|
this.preEntry = preEntry;
|
|
this.nextEntry = nextEntry;
|
|
this.key = key;
|
|
this.value = value;
|
|
}
|
|
|
|
public Entry<I, J> getPreEntry() {
|
|
return preEntry;
|
|
}
|
|
|
|
public void setPreEntry(Entry<I, J> preEntry) {
|
|
this.preEntry = preEntry;
|
|
}
|
|
|
|
public Entry<I, J> getNextEntry() {
|
|
return nextEntry;
|
|
}
|
|
|
|
public void setNextEntry(Entry<I, J> nextEntry) {
|
|
this.nextEntry = nextEntry;
|
|
}
|
|
|
|
public I getKey() {
|
|
return key;
|
|
}
|
|
|
|
public void setKey(I key) {
|
|
this.key = key;
|
|
}
|
|
|
|
public J getValue() {
|
|
return value;
|
|
}
|
|
|
|
public void setValue(J value) {
|
|
this.value = value;
|
|
}
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
final LRUCache<String, Integer> cache = new LRUCache<>(2);
|
|
cache.put("Key1", 1);
|
|
cache.put("Key2", 2);
|
|
cache.put("Key3", 3);
|
|
cache.put("Key4", 4);
|
|
System.out.println("getValue(Key1): " + cache.get("Key1"));
|
|
System.out.println("getValue(Key2): " + cache.get("Key2"));
|
|
System.out.println("getValue(Key3): " + cache.get("Key3"));
|
|
System.out.println("getValue(Key4): " + cache.get("Key4"));
|
|
}
|
|
}
|