继承LinkedHashMap,合理使用构造方法public LinkedHashMap(int initialCapacity, float loadFactor, boolean accessOrder)和protected boolean removeEldestEntry(Map.Entry<K,V> eldest)方法就可以自己构建简单的Cache了;
LinkedHashMap构造方法里,第一个参数为初始化的容量,初始化的容量肯定不低于该值,第二个参数是负载系数,用于自动扩展容量,当数据量跟容量占比达到系数时就自动扩展,第三个参数设置数据的访问顺序,设置true可使数据按照访问顺序排序,访问一个数据后就会将数据移动到尾端,设置false则按照数据插入顺序排序,可以使用第三个参数实现LRU和FIFO缓存;
removeEldestEntry方法使用protected修饰的,插入数据的时候会自动调用,参数为最老的数据(排最前面的数据),返回值表示是否移除该数据,默认实现返回false,即永不移除,可以重写该方法实现缓存数据的自动淘汰;
通过设置accessOrder参数和重写removeEldestEntry方法可以实现LRUCache和FIFOCache;
Demo:
/**
* 2023年8月30日下午3:34:16
*/
package testMyCache;
import java.util.LinkedHashMap;
/**
* @author XWF
*
*/
public class MyLruCache<K, V> extends LinkedHashMap<K, V> {
private static final long serialVersionUID = -4851815364152248408L;
private int cacheSize;
public MyLruCache(int cacheSize){
super(cacheSize + 1, 1.0f, true);//true按访问顺序LRU,false按添加顺序可FIFO
this.cacheSize = cacheSize;
}
@Override
protected boolean removeEldestEntry(java.util.Map.Entry<K, V> eldest) {
return this.size() > cacheSize;
}
}
/**
* 2023年8月30日下午3:33:39
*/
package testMyCache;
/**
* @author XWF
*
*/
public class TestMyCache {
/**
* @param args
*/
public static void main(String[] args) {
MyLruCache<Integer, String> lruCache = new MyLruCache<>(3);
lruCache.put(1, "A");
lruCache.put(2, "B");
lruCache.put(3, "C");
System.out.println(lruCache);
System.out.println("get(1)=" + lruCache.get(1)); //访问第一个数据
System.out.println("访问后:" + lruCache);
lruCache.put(4, "D");
System.out.println(lruCache);
lruCache.put(5, "E");
System.out.println(lruCache);
}
}
执行结果:
代码里设置size为3,accessOrder设置为true,按访问顺序排序,访问1后排序从123变为231,再插入4则移除2,插入5移除3;
当把顺序设置为false,按插入顺序排序,就可以变成FIFOCache了,执行结果:
访问1后顺序不变,最老的数据还是1;