Examples with HikariDataSource com.zaxxer.hikari.HikariDataSource used on opensource projects

Example 11 with HikariDataSource

use of com.zaxxer.hikari.HikariDataSource in project spring-boot by spring-projects.

the class HikariDataSourcePoolMetadataTests method getValidationQuery.

@Override
public void getValidationQuery() {
    HikariDataSource dataSource = createDataSource(0, 4);
    dataSource.setConnectionTestQuery("SELECT FROM FOO");
    assertThat(new HikariDataSourcePoolMetadata(dataSource).getValidationQuery()).isEqualTo("SELECT FROM FOO");
}

Example 12 with HikariDataSource

use of com.zaxxer.hikari.HikariDataSource in project spring-boot by spring-projects.

the class HikariDataSourcePoolMetadataTests method createDataSource.

private HikariDataSource createDataSource(int minSize, int maxSize) {
    HikariDataSource dataSource = (HikariDataSource) initializeBuilder().type(HikariDataSource.class).build();
    dataSource.setMinimumIdle(minSize);
    dataSource.setMaximumPoolSize(maxSize);
    return dataSource;
}

Example 13 with HikariDataSource

use of com.zaxxer.hikari.HikariDataSource in project HikariCP by brettwooldridge.

the class TestConnectionCloseBlocking method testConnectionCloseBlocking.

// @Test
public void testConnectionCloseBlocking() throws SQLException {
    HikariConfig config = newHikariConfig();
    config.setMinimumIdle(0);
    config.setMaximumPoolSize(1);
    config.setConnectionTimeout(1500);
    config.setDataSource(new CustomMockDataSource());
    long start = currentTime();
    try (HikariDataSource ds = new HikariDataSource(config);
        Connection connection = ds.getConnection()) {
        connection.close();
        // Hikari only checks for validity for connections with lastAccess > 1000 ms so we sleep for 1001 ms to force
        // Hikari to do a connection validation which will fail and will trigger the connection to be closed
        quietlySleep(1100L);
        shouldFail = true;
        // on physical connection close we sleep 2 seconds
        try (Connection connection2 = ds.getConnection()) {
            assertTrue("Waited longer than timeout", (elapsedMillis(start) < config.getConnectionTimeout()));
        }
    } catch (SQLException e) {
        assertTrue("getConnection failed because close connection took longer than timeout", (elapsedMillis(start) < config.getConnectionTimeout()));
    }
}

Example 14 with HikariDataSource

use of com.zaxxer.hikari.HikariDataSource in project HikariCP by brettwooldridge.

the class TestConnectionTimeoutRetry method testConnectionRetries2.

@Test
public void testConnectionRetries2() throws SQLException {
    HikariConfig config = newHikariConfig();
    config.setMinimumIdle(0);
    config.setMaximumPoolSize(1);
    config.setConnectionTimeout(2800);
    config.setValidationTimeout(2800);
    config.setInitializationFailTimeout(0);
    config.setConnectionTestQuery("VALUES 1");
    config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
    try (HikariDataSource ds = new HikariDataSource(config)) {
        final StubDataSource stubDataSource = ds.unwrap(StubDataSource.class);
        stubDataSource.setThrowException(new SQLException("Connection refused"));
        ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
        scheduler.schedule(new Runnable() {

            @Override
            public void run() {
                stubDataSource.setThrowException(null);
            }
        }, 300, TimeUnit.MILLISECONDS);
        long start = currentTime();
        try {
            try (Connection connection = ds.getConnection()) {
            // close immediately
            }
            long elapsed = elapsedMillis(start);
            assertTrue("Connection returned too quickly, something is wrong.", elapsed > 250);
            assertTrue("Waited too long to get a connection.", elapsed < config.getConnectionTimeout());
        } catch (SQLException e) {
            fail("Should not have timed out: " + e.getMessage());
        } finally {
            scheduler.shutdownNow();
        }
    }
}

Example 15 with HikariDataSource

use of com.zaxxer.hikari.HikariDataSource in project HikariCP by brettwooldridge.

the class TestConnectionTimeoutRetry method testConnectionIdleFill.

@Test
public void testConnectionIdleFill() throws Exception {
    StubConnection.slowCreate = false;
    HikariConfig config = newHikariConfig();
    config.setMinimumIdle(5);
    config.setMaximumPoolSize(10);
    config.setConnectionTimeout(2000);
    config.setValidationTimeout(2000);
    config.setConnectionTestQuery("VALUES 2");
    config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
    System.setProperty("com.zaxxer.hikari.housekeeping.periodMs", "400");
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    PrintStream ps = new PrintStream(baos, true);
    setSlf4jTargetStream(HikariPool.class, ps);
    try (HikariDataSource ds = new HikariDataSource(config)) {
        setSlf4jLogLevel(HikariPool.class, Level.DEBUG);
        HikariPool pool = getPool(ds);
        try (Connection connection1 = ds.getConnection();
            Connection connection2 = ds.getConnection();
            Connection connection3 = ds.getConnection();
            Connection connection4 = ds.getConnection();
            Connection connection5 = ds.getConnection();
            Connection connection6 = ds.getConnection();
            Connection connection7 = ds.getConnection()) {
            sleep(1300);
            assertSame("Total connections not as expected", 10, pool.getTotalConnections());
            assertSame("Idle connections not as expected", 3, pool.getIdleConnections());
        }
        assertSame("Total connections not as expected", 10, pool.getTotalConnections());
        assertSame("Idle connections not as expected", 10, pool.getIdleConnections());
    }
}