Apache Kyuubi JDBC快速入门指南

原创于 2025-06-25 09:19:05 发布 · 305 阅读

CC 4.0 BY-SA版权

Apache Kyuubi JDBC快速入门指南

什么是Apache Kyuubi

Apache Kyuubi是一个分布式SQL引擎，提供统一的JDBC接口来访问多种数据源。它兼容HiveServer2协议，这意味着可以使用Hive JDBC驱动来连接Kyuubi服务。本文将详细介绍如何使用JDBC方式快速连接Kyuubi服务。

JDBC驱动选择

虽然Kyuubi兼容Hive JDBC驱动，但官方推荐使用Kyuubi提供的增强版Hive JDBC驱动。这个驱动基于Hive 3.1.x版本开发，修复了原版驱动的一些功能缺失问题。

获取JDBC驱动

可以通过Maven依赖方式获取Kyuubi JDBC驱动：

<dependency>
    <groupId>org.apache.kyuubi</groupId>
    <artifactId>kyuubi-hive-jdbc-shaded</artifactId>
    <version>1.7.0</version>
</dependency>

这个依赖包含了所有必要的类库，无需额外引入其他依赖。

基础连接示例

非Kerberos认证连接

对于不需要Kerberos认证的Kyuubi服务，连接非常简单：

import java.sql.*;

public class KyuubiJDBCExample {
    // 指定驱动类名
    private static final String DRIVER = "org.apache.kyuubi.jdbc.KyuubiHiveDriver";
    // JDBC连接URL，默认端口10009
    private static final String URL = "jdbc:kyuubi://localhost:10009/default";
    
    public static void main(String[] args) {
        try {
            // 加载驱动
            Class.forName(DRIVER);
            // 建立连接
            try (Connection conn = DriverManager.getConnection(URL)) {
                // 创建语句对象
                try (Statement stmt = conn.createStatement()) {
                    // 执行查询
                    try (ResultSet rs = stmt.executeQuery("SHOW DATABASES")) {
                        // 处理结果集
                        while (rs.next()) {
                            System.out.println("数据库名称: " + rs.getString(1));
                        }
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

这个示例展示了最基本的连接和查询流程，适合开发测试环境使用。

安全连接示例

Kerberos认证连接

在生产环境中，通常需要Kerberos认证来确保连接安全。以下是使用Kerberos认证的示例：

import java.sql.*;

public class SecureKyuubiJDBCExample {
    private static final String DRIVER = "org.apache.kyuubi.jdbc.KyuubiHiveDriver";
    // JDBC URL模板，包含Kerberos认证参数
    private static final String URL_TEMPLATE = "jdbc:kyuubi://kyuubi-server:10009/default;" +
            "kyuubiClientPrincipal=%s;kyuubiClientKeytab=%s;kyuubiServerPrincipal=%s";
    
    public static void main(String[] args) {
        if (args.length < 3) {
            System.out.println("用法: java SecureKyuubiJDBCExample <clientPrincipal> <keytabPath> <serverPrincipal>");
            return;
        }
        
        String clientPrincipal = args[0];  // 客户端Kerberos主体
        String keytabPath = args[1];      // Keytab文件路径
        String serverPrincipal = args[2];  // 服务端Kerberos主体
        
        try {
            Class.forName(DRIVER);
            // 构建完整的JDBC URL
            String jdbcUrl = String.format(URL_TEMPLATE, 
                clientPrincipal, keytabPath, serverPrincipal);
            
            // 建立安全连接
            try (Connection conn = DriverManager.getConnection(jdbcUrl)) {
                System.out.println("成功连接到Kyuubi服务");
                
                // 执行元数据查询
                try (Statement stmt = conn.createStatement();
                     ResultSet rs = stmt.executeQuery("SELECT current_database()")) {
                    if (rs.next()) {
                        System.out.println("当前数据库: " + rs.getString(1));
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

连接参数详解

Kyuubi JDBC连接URL支持多种参数配置：

基础格式：jdbc:kyuubi://<host>:<port>/<database>
Kerberos相关参数：
- kyuubiClientPrincipal：客户端Kerberos主体
- kyuubiClientKeytab：Keytab文件路径
- kyuubiServerPrincipal：服务端Kerberos主体
其他常用参数：
- user：用户名（非Kerberos认证时使用）
- password：密码
- sessionConf：会话级别配置