Как безопасно отключить 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) продолжат работать, а возвращенные объекты будут уничтожены при возврате.

Другие вопросы по тегам