Examples with DataSource org.apache.tomcat.jdbc.pool.DataSource used on opensource projects

Example 11 with DataSource

use of org.apache.tomcat.jdbc.pool.DataSource 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());
}

Example 12 with DataSource

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

the class Bug54225 method testPool.

@Test
public void testPool() throws SQLException {
    PoolProperties poolProperties = new DefaultProperties();
    poolProperties.setMinIdle(0);
    poolProperties.setInitialSize(0);
    poolProperties.setMaxWait(5000);
    poolProperties.setRemoveAbandoned(true);
    poolProperties.setRemoveAbandonedTimeout(300);
    poolProperties.setRollbackOnReturn(true);
    poolProperties.setInitSQL(initSQL);
    final DataSource ds = new DataSource(poolProperties);
    ds.getConnection().close();
    assertNull(poolProperties.getInitSQL());
}

Example 13 with DataSource

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

the class Bug54227 method testPool.

@Test
public void testPool() throws SQLException, InterruptedException {
    PoolProperties poolProperties = new DefaultProperties();
    poolProperties.setMinIdle(0);
    poolProperties.setInitialSize(0);
    poolProperties.setMaxActive(1);
    poolProperties.setMaxWait(5000);
    poolProperties.setMaxAge(100);
    poolProperties.setRemoveAbandoned(false);
    final DataSource ds = new DataSource(poolProperties);
    Connection con;
    Connection actual1;
    Connection actual2;
    con = ds.getConnection();
    actual1 = ((PooledConnection) con).getConnection();
    con.close();
    con = ds.getConnection();
    actual2 = ((PooledConnection) con).getConnection();
    assertSame(actual1, actual2);
    con.close();
    Thread.sleep(150);
    con = ds.getConnection();
    actual2 = ((PooledConnection) con).getConnection();
    assertNotSame(actual1, actual2);
    con.close();
}

Example 14 with DataSource

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

the class Bug54978 method testIllegalValidationQueryWithLegalInit.

@Test
public void testIllegalValidationQueryWithLegalInit() throws SQLException {
    PoolProperties poolProperties = new DefaultProperties();
    poolProperties.setMinIdle(0);
    poolProperties.setInitialSize(1);
    poolProperties.setMaxActive(1);
    poolProperties.setMaxWait(5000);
    poolProperties.setMaxAge(100);
    poolProperties.setRemoveAbandoned(false);
    poolProperties.setTestOnBorrow(true);
    poolProperties.setTestOnConnect(false);
    poolProperties.setValidationQuery("sdadsada");
    poolProperties.setInitSQL("SELECT 1");
    final DataSource ds = new DataSource(poolProperties);
    ds.getConnection().close();
}

Example 15 with DataSource

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

the class Bug54978 method testIllegalValidationQuery.

@Test
public void testIllegalValidationQuery() {
    PoolProperties poolProperties = new DefaultProperties();
    poolProperties.setMinIdle(0);
    poolProperties.setInitialSize(1);
    poolProperties.setMaxActive(1);
    poolProperties.setMaxWait(5000);
    poolProperties.setMaxAge(100);
    poolProperties.setRemoveAbandoned(false);
    poolProperties.setTestOnBorrow(true);
    poolProperties.setTestOnConnect(false);
    poolProperties.setValidationQuery("sdadsada");
    final DataSource ds = new DataSource(poolProperties);
    try {
        ds.getConnection().close();
        fail("Validation should have failed.");
    } catch (SQLException x) {
    }
}