Java druid、hikari连接池链接方法配合jdbctemplate使用,trino链接池方法

该博客展示了如何使用Trino进行数据库连接,并通过Druid和Hikari连接池进行管理,同时演示了JdbcTemplate的使用。测试代码包括初始化数据库、设置连接池参数以及执行查询操作。

测试trino及druid、hikari连接池链接方法,jdbctemplate使用。

找了很久的接入各种类型数据库(es、hive、mysql、clickhouse、……)的方案,最后被告知trino。还没让做的时候就在思考怎么连上trino进行配置啥的,然后就简单写了个测试类,如下:
想着代码随时删,网上使用连接池链接trino的例子也很少有人分享,为了让有需要者快速复制开发,就贴出来:
依赖:

Hikari 的依赖这里面 :
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>
不然自己加:
 <dependency> 
 <groupId>com.zaxxer</groupId>
     <artifactId>HikariCP</artifactId> 
     <version>4.0.3</version> 
 </dependency>


druid的依赖:
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid</artifactId>
<version>1.2.8</version>
</dependency>】

trino
<dependency>
    <groupId>io.trino</groupId>
    <artifactId>trino-jdbc</artifactId>
<version>378</version>
</dependency>

代码自己测试过,可以用:


import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidPooledConnection;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import lombok.extern.slf4j.Slf4j;
import org.junit.Assert;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.test.context.junit.jupiter.SpringExtension;

import javax.sql.DataSource;
import java.sql.*;
import java.util.*;

import static java.util.concurrent.TimeUnit.MINUTES;
import static java.util.concurrent.TimeUnit.SECONDS;

/**
 * 测试trino及druid、hikari连接池链接方法,jdbctemplate使用。
 * Hikari 的依赖这里面 :
 * <dependency>
 * <groupId>org.springframework.boot</groupId>
 * <artifactId>spring-boot-starter-jdbc</artifactId>
 * </dependency>
 *
 * druid的依赖:
 * <dependency>
 *     <groupId>com.alibaba</groupId>
 * <artifactId>druid</artifactId>
 * <version>1.2.8</version>
 * </dependency>
 *
 * <dependency>
 *     <groupId>io.trino</groupId>
 *     <artifactId>trino-jdbc</artifactId>
 * <version>378</version>
 * </dependency>
 *
 * @date 2022/4/24 11:10
 */
@Slf4j
@SpringBootTest
@ExtendWith(SpringExtension.class)
public class ConnectionPool_Druid_Hikari_Trino_Test {
    /** 构造Hikari连接池 */
    public static HikariDataSource getHikariConfig() throws SQLException {
        // 1. 初始化数据库
        initDataBase("default");
        // 2. 返回连接池
        HikariConfig config = new HikariConfig();
        // 最小空闲连接
        config.setMinimumIdle(10);
        // 最大连接数
        config.setMaximumPoolSize(50);
        // 自动提交从池中返回的连接
        config.setAutoCommit(true);
        // 空闲连接超时时间
        config.setIdleTimeout(MINUTES.toMillis(10));
        // 连接最大存活时间
        config.setMaxLifetime(MINUTES.toMillis(30));
        // 连接超时时间
        config.setConnectionTimeout(SECONDS.toMillis(30));
        config.setJdbcUrl("jdbc:trino://127.0.0.1:49228/es");
        config.setUsername("elasticsearch");
        return new HikariDataSource(config);
    }

    /** 初始化数据库,如果数据库不存在就创建数据 */
    private static void initDataBase(String dbName) throws SQLException {
        Connection connection = null;
        Statement statement = null;
        try {
            Properties props = new Properties();
            // 获取数据库的备注信息
            // props.put("remarksReporting","true");
            props.put("user", "elasticsearch");
            // props.put("password",dbInfoConfig.getPassword());
            // 1.获取连接
            Class.forName("io.trino.jdbc.TrinoDriver");
            connection = DriverManager.getConnection("jdbc:trino://127.0.0.1:49228/es", props);
            // 2.查询是否存在数据库
            ResultSet catalogs = connection.getMetaData().getCatalogs();
            Set<String> databases = new HashSet<>();
            while (catalogs.next()) {
                databases.add(catalogs.getString(1));
            }
            if (!databases.contains(dbName)) {
                // 如果没有这个库, 新增这个库
                String sql = String.format("CREATE DATABASE %s", dbName);
                statement = connection.createStatement();
                statement.execute(sql);
            }
        } catch (Exception e) {
            log.error("初始化数据库失败", e);
        } finally {
            if (null != statement) {
                statement.close();
            }
            if (null != connection) {
                connection.close();
            }
        }
    }

    /**
     * springboot的依赖就包含了hikari 不然自己加 <dependency> <groupId>com.zaxxer</groupId>
     * <artifactId>HikariCP</artifactId> <version>4.0.3</version> </dependency>
     *
     * @since 2022/4/24 14:45
     */
    @Test
    void testHikariPool() throws SQLException {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(getHikariConfig());
        List<Map<String, Object>> maps =
                jdbcTemplate.queryForList("select * from es.default.zyttest");
        Assert.assertEquals(
                "[{address=四川, age=12, name=张三}, {address=四川, age=12, name=李四}]", maps.toString());
    }

    /**
     * druid也是的 <dependency> <groupId>com.alibaba</groupId> <artifactId>druid</artifactId>
     * <version>1.2.8</version> </dependency>
     *
     * @since 2022/4/24 14:45
     */
    @Test
    void testDruidPool() throws SQLException {
        // 直接使用statement来用,如果用jdbctemplate的话直接返回DruidDataSource即可
        try (final DruidPooledConnection connection = produceConnectPool()) {
            try (final PreparedStatement statement =
                    connection.prepareStatement("select * from table")) {
                try (final ResultSet resultSet = statement.executeQuery()) {
                    final ResultSetMetaData metaData = resultSet.getMetaData();
                    while (resultSet.next()) {
                        Object object = resultSet.getObject(1);
                    }
                }
            }
        }
    }

    @Test
    void testCommonConn() {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(getPrestoDataSource());
        String s = "select address from es.default.zyttest";
        RowMapper<String> rowMapper = new BeanPropertyRowMapper<>(String.class);
        List<Map<String, Object>> maps =
                jdbcTemplate.queryForList("select * from es.default.zyttest");
        Assert.assertEquals(
                "[{address=四川, age=12, name=张三}, {address=四川, age=12, name=李四}]", maps.toString());
    }

    public DataSource getPrestoDataSource() {
        return DataSourceBuilder.create()
                .driverClassName("io.trino.jdbc.TrinoDriver")
                .url("jdbc:trino://127.0.0.1:49228/es")
                .username("elasticsearch")
                .build();
    }

    public DruidPooledConnection produceConnectPool() {
        try (DruidDataSource dataSource = new DruidDataSource()) {
            // oracle : jdbc:oracle:thin:@127.0.0.1:1522/#service
            // mysql :
            // jdbc:mysql://127.0.0.1:3306/tableName?useUnicode=true&characterEncoding=UTF-8&useSSL=false&allowMultiQueries=true&autoReconnect=true&serverTimezone=Asia/Shanghai
            dataSource.setUrl("");
            dataSource.setUsername("");
            dataSource.setPassword("");
            // oracle : oracle.jdbc.OracleDriver
            // mysql : com.mysql.cj.jdbc.Driver
            // postgresql : org.postgresql.Driver
            dataSource.setDriverClassName("");
            dataSource.setValidationQuery("select 1 from dual");
            dataSource.setInitialSize(10);
            dataSource.setMinIdle(10);
            dataSource.setMaxActive(20);
            dataSource.setPoolPreparedStatements(false);
            dataSource.setRemoveAbandoned(true);
            dataSource.setRemoveAbandonedTimeout(300);
            dataSource.setLogAbandoned(false);
            return dataSource.getConnection();
        } catch (Exception e) {
            throw new RuntimeException("连接失败!原因 >> " + e.getMessage(), e);
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值