Как безопасно отключить GenericKeyedObjectPool?

Question

У меня есть GenericKeyedObjectPool в моем приложении. Я могу закрыть его с помощью метода close, но как мне ждать возврата клиентов (и пул уничтожает) каждый заимствованный объект в пул?

Мне нужно что-то вроде ExecutorService.awaitTermination.

Answer 1

Создайте обертку для 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 
} 

Answer 2

Вы должны быть в безопасности, просто закрыв его, не дожидаясь возвращения всех объектов.

Из документов в GenericKeyedObjectPool.html#close():

Замыкает бассейн. Как только пул закрыт, заимствование объекта() завершится неудачно с IllegalStateException, но returnObject (Object) и invalidateObject (Object) будет продолжать работать, с возвращенными объектами , уничтоженным при возврате.