Как безопасно отключить GenericKeyedObjectPool?
У меня есть GenericKeyedObjectPool
в моем приложении. Я могу закрыть это с close
метод, но как мне ждать, пока клиенты вернут (и пул уничтожит) каждый заимствованный объект в пул?
Мне нужно что-то вроде ExecutorService.awaitTermination
,
2 ответа
Создать обертку для GenericKeyedObjectPool
который имеет необходимый awaitTermination
метод. Вы можете проверить close
и returnObject
вызывает и уменьшает значение защелки, если пул закрыт, и каждый объект был возвращен (= общее количество экземпляров, заимствованных из этого пула в текущий момент, равно нулю).
public final class ListenablePool<K, V> {
private final GenericKeyedObjectPool<K, V> delegate;
private final CountDownLatch closeLatch = new CountDownLatch(1);
private final AtomicBoolean closed = new AtomicBoolean();
public ListenablePool(final KeyedPoolableObjectFactory<K, V> factory) {
this.delegate = new GenericKeyedObjectPool<K, V>(factory);
}
public V borrowObject(final K key) throws Exception {
return delegate.borrowObject(key);
}
public void returnObject(final K key, final V obj) throws Exception {
try {
delegate.returnObject(key, obj);
} finally {
countDownIfRequired();
}
}
private void countDownIfRequired() {
if (closed.get() && delegate.getNumActive() == 0) {
closeLatch.countDown();
}
}
public void close() throws Exception {
try {
delegate.close();
} finally {
closed.set(true);
countDownIfRequired();
}
}
public void awaitTermination() throws InterruptedException {
closeLatch.await();
}
public void awaitTermination(final long timeout, final TimeUnit unit)
throws InterruptedException {
closeLatch.await(timeout, unit);
}
public int getNumActive() {
return delegate.getNumActive();
}
// other delegate methods
}
Вы должны быть в безопасности, просто закрыв его, не дожидаясь возвращения всех объектов.
Из документов GenericKeyedObjectPool.html # close ():
Закрывает бассейн. Как только пул будет закрыт, заимствовать заемщик OObject() с IllegalStateException, но returnObject(Object) и invalidateObject(Object) продолжат работать, а возвращенные объекты будут уничтожены при возврате.