Search in sources :

Example 1 with PoolExhaustedException

use of org.apache.tomcat.jdbc.pool.PoolExhaustedException in project tomcat by apache.

the class Bug53367 method testPool.

@Test
public void testPool() throws SQLException, InterruptedException {
    DriverManager.setLoginTimeout(1);
    PoolProperties poolProperties = new DefaultProperties();
    int threadsCount = 3;
    poolProperties.setMaxActive(threadsCount);
    poolProperties.setMaxIdle(threadsCount);
    poolProperties.setMinIdle(0);
    poolProperties.setMaxWait(5000);
    poolProperties.setInitialSize(0);
    poolProperties.setRemoveAbandoned(true);
    poolProperties.setRemoveAbandonedTimeout(300);
    poolProperties.setRollbackOnReturn(true);
    poolProperties.setFairQueue(fairQueue);
    final DataSource ds = new DataSource(poolProperties);
    final CountDownLatch openedLatch = new CountDownLatch(threadsCount);
    final CountDownLatch closedLatch = new CountDownLatch(threadsCount);
    final CountDownLatch toCloseLatch = new CountDownLatch(1);
    for (int i = 0; i < threadsCount; i++) {
        new Thread(new Runnable() {

            @Override
            public void run() {
                try {
                    Connection connection = ds.getConnection();
                    openedLatch.countDown();
                    toCloseLatch.await();
                    connection.close();
                    closedLatch.countDown();
                } catch (Exception e) {
                    System.err.println("Step 1:" + e.getMessage());
                }
            }
        }).start();
    }
    openedLatch.await();
    ConnectionPool pool = ds.getPool();
    //Now we have 3 initialized busy connections
    Assert.assertEquals(0, pool.getIdle());
    Assert.assertEquals(threadsCount, pool.getActive());
    Assert.assertEquals(threadsCount, pool.getSize());
    List<Thread> threads = new ArrayList<>();
    for (int i = 0; i < threadsCount; i++) {
        Thread thread = new Thread(new Runnable() {

            @Override
            public void run() {
                try {
                    ds.getConnection();
                } catch (Exception e) {
                    System.err.println("Step 2:" + e.getMessage());
                }
            }
        });
        thread.start();
        threads.add(thread);
    }
    for (Thread thread : threads) {
        thread.interrupt();
    }
    for (Thread thread : threads) {
        thread.join();
    }
    //Still 3 active connections
    Assert.assertEquals(0, pool.getIdle());
    Assert.assertEquals(threadsCount, pool.getActive());
    Assert.assertEquals(threadsCount, pool.getSize());
    toCloseLatch.countDown();
    closedLatch.await();
    //Here comes the bug! No more active connections and unable to establish new connections.
    // <-- Should be threadsCount (3) here
    Assert.assertEquals(threadsCount, pool.getIdle());
    Assert.assertEquals(0, pool.getActive());
    Assert.assertEquals(threadsCount, pool.getSize());
    final AtomicInteger failedCount = new AtomicInteger();
    final ArrayBlockingQueue<Connection> cons = new ArrayBlockingQueue<>(threadsCount);
    threads.clear();
    for (int i = 0; i < threadsCount; i++) {
        Thread thread = new Thread(new Runnable() {

            @Override
            public void run() {
                try {
                    cons.add(ds.getConnection());
                } catch (PoolExhaustedException e) {
                    failedCount.incrementAndGet();
                    System.err.println("Step 3:" + e.getMessage());
                } catch (Exception e) {
                    System.err.println("Step 4:" + e.getMessage());
                    throw new RuntimeException(e);
                }
            }
        });
        thread.start();
        threads.add(thread);
    }
    for (Thread thread : threads) {
        thread.join();
    }
    Assert.assertEquals(0, failedCount.get());
    Assert.assertEquals(0, pool.getIdle());
    Assert.assertEquals(threadsCount, pool.getActive());
    Assert.assertEquals(threadsCount, pool.getSize());
    for (Connection con : cons) {
        con.close();
    }
    Assert.assertEquals(threadsCount, pool.getIdle());
    Assert.assertEquals(0, pool.getActive());
    Assert.assertEquals(threadsCount, pool.getSize());
}
Also used : ConnectionPool(org.apache.tomcat.jdbc.pool.ConnectionPool) DefaultProperties(org.apache.tomcat.jdbc.test.DefaultProperties) PoolExhaustedException(org.apache.tomcat.jdbc.pool.PoolExhaustedException) Connection(java.sql.Connection) ArrayList(java.util.ArrayList) CountDownLatch(java.util.concurrent.CountDownLatch) SQLException(java.sql.SQLException) PoolExhaustedException(org.apache.tomcat.jdbc.pool.PoolExhaustedException) PoolProperties(org.apache.tomcat.jdbc.pool.PoolProperties) DataSource(org.apache.tomcat.jdbc.pool.DataSource) ArrayBlockingQueue(java.util.concurrent.ArrayBlockingQueue) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) Test(org.junit.Test)

Aggregations

Connection (java.sql.Connection)1 SQLException (java.sql.SQLException)1 ArrayList (java.util.ArrayList)1 ArrayBlockingQueue (java.util.concurrent.ArrayBlockingQueue)1 CountDownLatch (java.util.concurrent.CountDownLatch)1 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)1 ConnectionPool (org.apache.tomcat.jdbc.pool.ConnectionPool)1 DataSource (org.apache.tomcat.jdbc.pool.DataSource)1 PoolExhaustedException (org.apache.tomcat.jdbc.pool.PoolExhaustedException)1 PoolProperties (org.apache.tomcat.jdbc.pool.PoolProperties)1 DefaultProperties (org.apache.tomcat.jdbc.test.DefaultProperties)1 Test (org.junit.Test)1