在 Spring 引导应用程序中创建自定义连接池

Question

我正在编写一个 Spring 引导应用程序，它连接到 Snowflake 数据仓库并在其上执行 SQL 查询。我写了一个 Configuration class 用于配置数据源以连接到 Snowflake 数据仓库，如下所示：

@Configuration
@EnableAutoConfiguration(exclude={DataSourceAutoConfiguration.class})
public class DBConfig {

    Logger logger = LoggerFactory.getLogger(DBConfig.class);
    @Bean
    JdbcTemplate jdbcTemplate() throws IllegalAccessException, InvocationTargetException, InstantiationException {
        logger.info("-----Configuring JDBCTemplate------");
        SnowflakeBasicDataSource dataSource = new SnowflakeBasicDataSource();
        dataSource.setServerName("<myserver>.snowflakecomputing.com");
        dataSource.setUser("<my_username>");
        dataSource.setPassword("<my_password>");
        dataSource.setWarehouse("DEMO_WH");
        dataSource.setDatabaseName("DEMO_DB");
        dataSource.setSchema("PUBLIC");
        return new JdbcTemplate(dataSource);
    }
}

我的 pom.xml 如下所示：

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.1.1.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.vaibhav</groupId>
    <artifactId>snowflake-1</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>snowflake-1</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>net.snowflake</groupId>
            <artifactId>snowflake-jdbc</artifactId>
            <version>3.6.21</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web-services</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

我必须在我的 Spring 启动应用程序中为此数据源使用连接池。

如何在我的应用程序中使用 HikariCP 连接池，它可以与我的自定义数据源完美配合？

------编辑---感谢您提供解决方案，最后我的工作代码看起来像

@Configuration
@EnableAutoConfiguration(exclude = { DataSourceAutoConfiguration.class })
public class DBConfig {

    Logger logger = LoggerFactory.getLogger(DBConfig.class);

    @Bean
    JdbcTemplate jdbcTemplate() throws IllegalAccessException, InvocationTargetException, InstantiationException {
        logger.info("-----Configuring JDBCTemplate------");

        HikariConfig config = new HikariConfig();
        config.setDriverClassName("net.snowflake.client.jdbc.SnowflakeDriver");
        // config.setDataSourceProperties(properties);
        config.setJdbcUrl("jdbc:snowflake://<myserver>.snowflakecomputing.com/?warehouse=DEMO_WH&db=DEMO_DB&schema=PUBLIC");
        config.setUsername("<my_username>");
        config.setPassword("<my_password>");
        HikariDataSource ds = new HikariDataSource(config);

        return new JdbcTemplate(ds);
    }
}

Answer 1

参见设置 SnowflakeDriver with Hikari:

  HikariConfig config = new HikariConfig();    
  config.setDriverClassName("com.snowflake.client.jdbc.SnowflakeDriver");    
  config.setDataSourceProperties(properties);    
  config.setJdbcUrl(connectStr);    
  HikariDataSource ds = new HikariDataSource(config);

Answer 2

Hikari 是 Spring-boot 2+

中的默认连接池

We have nothing to do if we want to use Hikari in an application based on Spring Boot 2.x.

您可以通过 application.yml / application.properties 设置连接池的不同属性。下面是 application.yml:

的例子

spring:
  datasource
    hikari:
      maximumPoolSize: 4 # Specify maximum pool size
      minimumIdle: 1 # Specify minimum pool size
      driver-class-name: com.snowflake.client.jdbc.SnowflakeDriver

这是用于配置 Hikari CP 的 useful link。

Answer 3

如果包含 spring-jdbc，Spring 将根据可用的数据源自动创建 JdbcTemplate。所以如果上面的答案还不能让你满意，你可以试试：

@Configuration
@EnableAutoConfiguration(exclude={DataSourceAutoConfiguration.class})
public class DBConfig {

    Logger logger = LoggerFactory.getLogger(DBConfig.class);

    // TAKE NOTE THAT THIS MIGHT ALREADY BE DONE BY SPRING
    @Bean
    protected JdbcTemplate jdbcTemplate( DataSource dataSource )
    {
        return new JdbcTemplate( dataSource );
    }    

    @Bean
    protected DataSource makeDataSource() throws IllegalAccessException, InvocationTargetException, InstantiationException {
        logger.info("-----Configuring JDBCTemplate------");
        SnowflakeBasicDataSource dataSource = new SnowflakeBasicDataSource();
        dataSource.setServerName("<myserver>.snowflakecomputing.com");
        dataSource.setUser("<my_username>");
        dataSource.setPassword("<my_password>");
        dataSource.setWarehouse("DEMO_WH");
        dataSource.setDatabaseName("DEMO_DB");
        dataSource.setSchema("PUBLIC");
        return dataSource;
    }
}

在 Spring 引导应用程序中创建自定义连接池

Creating custom connection pool in Spring Boot application

java

spring

spring-boot

hikaricp

snowflake-cloud-data-platform