Java Code Examples for com.zaxxer.hikari.HikariDataSource#close()

/**
 * @see <a href="https://github.com/brettwooldridge/HikariCP/issues/742">Support graceful shutdown of connection
 * pool</a>
 */
private boolean terminateHikariDataSource(HikariDataSource dataSource) {
  HikariPoolMXBean poolMXBean = dataSource.getHikariPoolMXBean();

  //evict idle connections
  poolMXBean.softEvictConnections();

  if (poolMXBean.getActiveConnections() > 0 && retryTimes < MAX_RETRY_TIMES) {
    logger.warn("Data source {} still has {} active connections, will retry in {} ms.", dataSource,
        poolMXBean.getActiveConnections(), RETRY_DELAY_IN_MILLISECONDS);
    return false;
  }

  if (poolMXBean.getActiveConnections() > 0) {
    logger.warn("Retry times({}) >= {}, force closing data source {}, with {} active connections!", retryTimes,
        MAX_RETRY_TIMES, dataSource, poolMXBean.getActiveConnections());
  }

  dataSource.close();

  return true;
}
 

private void removeObsoleteTenants(DatabaseConfiguration[] configurations) {

        // Are there Tenants, that have been removed:
        Set<String> tenantNamesFromConfiguration = Arrays.stream(configurations)
                .map(x -> x.getTenant())
                .collect(Collectors.toSet());

        for (String tenant : tenants.keySet()) {

            // There is currently a Tenant, which is not listed anymore:
            if(!tenantNamesFromConfiguration.contains(tenant)) {

                // So get the DataSource first ...
                HikariDataSource dataSource = tenants.get(tenant);

                // ... close all existing connections:
                dataSource.close();

                // ... and remove it:
                tenants.remove(tenant);
            }
        }
    }
 

private void createDatabase(String dbName) {
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setJdbcUrl("jdbc:mysql://localhost:33307/conductor");
    dataSource.setUsername("root");
    dataSource.setPassword("root");
    dataSource.setAutoCommit(false);

    dataSource.setMaximumPoolSize(2);

    try (Connection connection = dataSource.getConnection()) {
        try(Statement statement = connection.createStatement()) {
            statement.execute("CREATE DATABASE IF NOT EXISTS "+dbName);
        }
    } catch (SQLException sqlException) {
        logger.error("Unable to create default connection for docker mysql db", sqlException);
        throw new RuntimeException(sqlException);
    }finally {
        dataSource.close();
    }
}
 

@Test
public void defaultTest2() throws InterruptedException, SQLException {
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDataSourceClassName(DATA_SOURCE_CLASS_NAME);
    dataSource.addDataSourceProperty("url", JDBC_URL);

    try {
        Connection connection = dataSource.getConnection();
        Assert.assertNotNull(connection);

        Thread.sleep(500);

        connection.close();

        Thread.sleep(500);

        PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();
        verifier.printCache();

        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.HikariDataSource.HikariDataSource()"));
        verifier.verifyTrace(event(serviceType, getConnectionMethod1));
        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.pool.BaseHikariPool.BaseHikariPool(com.zaxxer.hikari.HikariConfig, java.lang.String, java.lang.String)"));
        verifier.verifyTrace(event(serviceType, proxyConnectionMethod));

    } finally {
        if (dataSource != null) {
            dataSource.close();
        }
    }
}
 

@Test
public void defaultTest3() throws InterruptedException, SQLException, NoSuchMethodException {
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDataSourceClassName(DATA_SOURCE_CLASS_NAME);
    dataSource.addDataSourceProperty("url", JDBC_URL);

    try {
        Connection connection = dataSource.getConnection("", "");
        Assert.assertNotNull(connection);

        Thread.sleep(500);

        connection.close();

        Thread.sleep(500);

        PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();
        verifier.printCache();

        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.HikariDataSource.HikariDataSource()"));
        verifier.verifyTrace(event(serviceType, getConnectionMethod2, annotation(AnnotationKey.ARGS0.getName(), "")));
        verifier.verifyTrace(event(serviceType, proxyConnectionMethod));
    } finally {
        if (dataSource != null) {
            dataSource.close();
        }
    }
}
 

@Override
public void contextDestroyed(ServletContextEvent event) {
  // This function is called when the Servlet is destroyed.
  HikariDataSource pool = (HikariDataSource) event.getServletContext().getAttribute("my-pool");
  if (pool != null) {
    pool.close();
  }
}
 

@Override
public void contextDestroyed(ServletContextEvent event) {
  // This function is called when the Servlet is destroyed.
  HikariDataSource pool = (HikariDataSource) event.getServletContext().getAttribute("my-pool");
  if (pool != null) {
    pool.close();
  }
}
 

private void performSimpleTest(String jdbcUrl) throws SQLException {
    HikariDataSource dataSource = getDataSource(jdbcUrl, 1);
    new QueryRunner(dataSource).query("SELECT 1 FROM dual", new ResultSetHandler<Object>() {
        @Override
        public Object handle(ResultSet rs) throws SQLException {
            rs.next();
            int resultSetInt = rs.getInt(1);
            assertEquals("A basic SELECT query succeeds", 1, resultSetInt);
            return true;
        }
    });
    dataSource.close();
}
 

@Override
public void destroy() {
    AnnotationConfigApplicationContext ctx = Main.getSpringContext();
    if (ctx != null) {
        HikariDataSource dataSource = ctx.getBean(HikariDataSource.class);
        if (dataSource != null) {
            dataSource.close();
        }
        ctx.close();
    }
}
 

/**
 * Close, close cached data source.
 */
@Override
public void close() {
    for (HikariDataSource each : cachedDataSources.values()) {
        if (!each.isClosed()) {
            each.close();
        }
    }
    cachedDataSources.clear();
    sourceDatasources.clear();
}
 

@Override
public void afterAll(final TestInvocation invocation) throws Exception {
    final ExecutionContext context = invocation.getContext();
    final HikariDataSource ds = (HikariDataSource) context.getData(Constants.KEY_DATA_SOURCE);
    ds.close();
    context.storeData(Constants.KEY_DATA_SOURCE, null);
}
 

@SneakyThrows
private List<Entity> queryColumns(TableRequest request) {
    HikariDataSource dataSource = DbUtil.buildFromTableRequest(request);
    Db db = new Db(dataSource);

    List<Entity> query = db.query(COLUMN_SQL_TEMPLATE, request.getTableName());

    dataSource.close();
    return query;
}
 

@SneakyThrows
private Entity queryTable(TableRequest request) {
    HikariDataSource dataSource = DbUtil.buildFromTableRequest(request);
    Db db = new Db(dataSource);

    String paramSql = StrUtil.EMPTY;
    if (StrUtil.isNotBlank(request.getTableName())) {
        paramSql = "and table_name = ?";
    }
    String sql = String.format(TABLE_SQL_TEMPLATE, paramSql);
    Entity entity = db.queryOne(sql, request.getTableName());

    dataSource.close();
    return entity;
}
 

@Test
public void defaultTest3() throws InterruptedException, SQLException {
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDataSourceClassName(DATA_SOURCE_CLASS_NAME);
    dataSource.addDataSourceProperty("url", JDBC_URL);

    try {
        Connection connection = dataSource.getConnection("", "");
        Assert.assertNotNull(connection);

        Thread.sleep(500);

        connection.close();

        Thread.sleep(500);

        PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();
        verifier.printCache();

        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.HikariDataSource.HikariDataSource()"));
        verifier.verifyTrace(event(serviceType, getConnectionMethod2, annotation(AnnotationKey.ARGS0.getName(), "")));
        verifier.verifyTrace(event(serviceType, proxyConnectionMethod));
    } finally {
        if (dataSource != null) {
            dataSource.close();
        }
    }
}
 

public HikariDataSource createAndTestDataSource() throws SQLException {
  HikariDataSource newDataSource = createDataSource();
  try {
    testConnection(newDataSource);
  } catch (SQLException ex) {
    logger.error("Testing connection for data source failed: {}", newDataSource.getJdbcUrl(), ex);
    newDataSource.close();
    throw ex;
  }

  return newDataSource;
}
 

private static void silentClose(HikariDataSource con) {
    if (con != null && !con.isClosed()) {
        con.close();
    }
}
 

@Test
public void defaultTest2() throws InterruptedException, SQLException, NoSuchMethodException {
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDataSourceClassName(DATA_SOURCE_CLASS_NAME);
    dataSource.addDataSourceProperty("url", JDBC_URL);

    try {
        Connection connection = dataSource.getConnection();
        Assert.assertNotNull(connection);

        Thread.sleep(500);

        connection.close();

        Thread.sleep(500);

        PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();
        verifier.printCache();

        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.HikariDataSource.HikariDataSource()"));
        verifier.verifyTrace(event(serviceType, getConnectionMethod1));
        verifier.verifyTrace(event(serviceType, "com.zaxxer.hikari.pool.BaseHikariPool.BaseHikariPool(com.zaxxer.hikari.HikariConfig, java.lang.String, java.lang.String)"));
        verifier.verifyTrace(event(serviceType, proxyConnectionMethod));

    } finally {
        if (dataSource != null) {
            dataSource.close();
        }
    }
}
 

private static void silentClose(HikariDataSource con) {
    if (con != null && !con.isClosed()) {
        con.close();
    }
}
 

@PreDestroy
public void shutdown() {
    for (HikariDataSource hiDs : DATA_SOURCE_CACHE.asMap().values()) {
        hiDs.close();
    }
}
 

/**
 * This method intentionally verifies encoding twice to ensure that the query string parameters are used when
 * Connections are created from cached containers.
 *
 * @param jdbcUrl
 * @throws SQLException
 */
private void performSimpleTestWithCharacterSet(String jdbcUrl) throws SQLException {
    HikariDataSource datasource1 = verifyCharacterSet(jdbcUrl);
    HikariDataSource datasource2 = verifyCharacterSet(jdbcUrl);
    datasource1.close();
    datasource2.close();
}