程序退出时关闭执行程序服务

时间:2013-12-10 05:01:30

标签: java executorservice

我正在使用expiry实现缓存。我使用ScheduledThreadExecutor来安排从缓存中删除条目。我的问题是执行程序永远不会被关闭。我在executor.shutdown()中尝试了shutdownHook方法,但即使我的主程序执行完毕也没有执行。我也不喜欢终结者。我的代码如下。我希望在主程序退出时执行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() ;
}

}

2 个答案:

答案 0 :(得分:3)

感谢您的回复。我正在创建一个图书馆。因此,除非库的用户明确调用executor.shutdown(),否则无法知道何时调用shutdown()。无论如何,我得到了解决方案。我使执行程序成为守护程序线程,以便在主程序退出时自动终止。制作守护程序执行程序的代码如下所示

scheduler =  Executors.newScheduledThreadPool(maxSize,new ThreadFactory() {
        @Override
        public Thread newThread(Runnable r) {
            Thread t = new Thread(r);
            t.setDaemon(true);
            return t;
        }
    });

谢谢你们。希望这会帮助某人

答案 1 :(得分:0)

只有在所有已提交的任务完成后,

executor.shutdown()才会关闭线程池。看起来你有计时器计划的任务,使执行程序运行。取消它们,或致电shutdownNow