Master -Slave Database Configuration With Hikari Pool

Question

I have two DataSource Beans one with @Primary annotation. Individual Hikari pools are created for every DataSource.

I am trying to change the HikariDataSource from Pool 1(if connection is not available) to Pool 2 .

@Primary
@Bean(destroyMethod = "close", name = "dataSource")
public CustomHikariDataSource dataSource() throws SQLException {
    try {
        primaryDataSource = mainDataSource();
    } catch (Exception e) {
        primaryDataSource = secondaryDataSource();
    }
    HikariConfig config = new HikariConfig();
    config.setDataSource(primaryDataSource);
    config.setPoolName("POOL_PRIMARY");
    config.setAllowPoolSuspension(true);
    config.setIdleTimeout(10000);
    config.setMaxLifetime(30000);
    return new CustomHikariDataSource(config);
}

@Bean(destroyMethod = "close", name = "failoverDataSource")
public CustomHikariDataSource failoverDataSource() throws SQLException {
    secondaryDataSource = secondaryDataSource();
    HikariConfig config = new HikariConfig();
    config.setDataSource(secondaryDataSource);
    config.setPoolName("POOL_SECONDARY");
    config.setAllowPoolSuspension(true);
    return new CustomHikariDataSource(config);
}

private DataSource mainDataSource() {
    return dataSourceProperties().initializeDataSourceBuilder().build();

}

private DataSource secondaryDataSource() {
    return failoverDataSourceProperties().initializeDataSourceBuilder().build();

}

Where is the actual Problem?

Answer 1

Finally i am able to achieve it by Overriding getConnection() method from HikariDataSource.class

@Override
public Connection getConnection() throws SQLException {

    if (isClosed()) {
        throw new SQLException("HikariDataSource " + this + " has been closed.");
    }

    if (fastPathPool != null && (fastPathPool.poolState == 0 || fastPathPool.poolState == 1)) {
        try {
            fastPathPool.resumePool();
            con = fastPathPool.getConnection();
        } catch (Exception e) {

        }
        if (con.isClosed()) {
            config = pool.config;
            fastPathPool.suspendPool();
        } else
            return con;
    }

    config.setDataSource(dataSource);
    config.setAllowPoolSuspension(true);
    config.setMinimumIdle(minIdle);
    pool = new HikariPool(config);
    HikariPool result = pool;
    if (result == null) {
        synchronized (this) {
            result = pool;
            if (result == null) {
                validate();
                System.out.println("{} - Starting..." + getPoolName());
                try {
                    pool = result = new HikariPool(this);
                    this.seal();
                } catch (PoolInitializationException pie) {
                    if (pie.getCause() instanceof SQLException) {
                        throw (SQLException) pie.getCause();
                    } else {
                        throw pie;
                    }
                }
                System.out.println("{} - Start completed." + getPoolName());
            }
        }
    }

    return result.getConnection();

}

For complete class ,feel free to ping me.

Happy Coding ! :)