我正在实现与expiry
的高速缓存。我正在使用ScheduledThreadExecutor
来安排从缓存中删除条目。我的问题是执行者永远不会关闭。我在shutdownHook
中尝试过executor.shutdown()
方法,但是即使在我的主程序执行完成后它也没有执行。我也不喜欢终结者。我的代码如下。当主程序退出时,我想要执行closeCache()
方法。程序退出时的关机执行程序服务
public class TimeCacheManual<K,V> {
private final int maxSize;
private final long timeToLive;
private Map<K, V> keyValueMap;
private Map<K,ScheduledFuture > keySchedulerMap;
private Queue<K> keys;
private final ScheduledExecutorService scheduler;
/*
* creates new instance of TimeBasedEvictionCache.
* @param maxSize must be greater than zero
* @param timeToLive must be greater than zero
* @throws IllegalArgumentException if {@code maxSize<1||timeToLive<1}
* */
public TimeCacheManual(int maxSize,long timeToLive) {
if(maxSize<1||timeToLive<1){
throw new IllegalArgumentException();
}
this.maxSize = maxSize;
this.timeToLive = timeToLive;
keyValueMap = new ConcurrentHashMap<K, V>(maxSize);
keySchedulerMap = new ConcurrentHashMap<K, ScheduledFuture>(maxSize);
keys = new ConcurrentLinkedQueue<K>();
scheduler = Executors.newScheduledThreadPool(maxSize);
}
/*
* adds a key value pair to the cache.
* @param key
* @param value associated with key
*/
public synchronized void put(K key,V value) {
if (keyValueMap.containsKey(key)) {
refreshKey(key);
}
else{
keys.add(key);
}
keyValueMap.put(key, value);
scheduleEviction(key); // schedules eviction of the key after timeToLive
}
/*
* schedules eviction of particular key after timeToLive
* @param key
*/
private void scheduleEviction(final K key){
ScheduledFuture sf= scheduler.schedule(new Runnable(){
@Override public void run(){
keys.remove(key);
keyValueMap.remove(key);
}
},
timeToLive,
TimeUnit.MILLISECONDS);
keySchedulerMap.put(key,sf);
}
/*
* used to get a value associated with a given key. returns null if no value is associated with given key
* @param key
* @return value associated with key, null if no value is associated with particular key
*/
public synchronized V get(K key) {
refreshKey(key);
scheduleEviction(key);
return keyValueMap.get(key);
}
/*
* updates the order of keys according to a particular policy
* @param key to be refreshed
*/
private void refreshKey(K key){ // refreshing the order of keys
keySchedulerMap.get(key).cancel(true) ;
keys.remove(key); //LRU policy
keys.add(key);
}
public void closeCache(){
scheduler.shutdownNow() ;
}
}