Apache Kyuubi 引擎选择与高可用性配置问题解析

问题背景

Apache Kyuubi 是一个开源的分布式 SQL 引擎服务，为用户提供了统一的多引擎 SQL 查询接口。在实际生产环境中，用户经常需要同时配置多种查询引擎（如 Spark SQL 和 Trino）并实现服务的高可用性（HA）。然而，在 Kyuubi 1.9.1 版本中，当同时启用 HA 功能和设置 kyuubi.engine.share.level=GROUP 时，用户遇到了引擎选择失效的问题。

核心问题分析

1. 引擎选择失效问题

当用户通过 ZooKeeper 连接 Kyuubi 服务时，发现无法通过 JDBC URL 参数指定查询引擎类型。具体表现为：

• 无论设置 kyuubi.engine.type=SPARK_SQL 还是 kyuubi.engine.type=TRINO，系统都会默认使用 Spark SQL 引擎
• 只有在直接连接 Kyuubi 服务端口（如 10009）时，引擎选择参数才能生效

根本原因：JDBC URL 参数传递方式不正确。正确的参数传递应该使用 # 符号而非 ; 分隔符。例如：

# 错误方式
jdbc:hive2://zk_host:2181/;kyuubi.engine.type=SPARK_SQL;serviceDiscoveryMode=zooKeeper

# 正确方式
jdbc:hive2://zk_host:2181/;serviceDiscoveryMode=zooKeeper#kyuubi.engine.type=SPARK_SQL

2. 高可用性实现机制

用户对 Kyuubi HA 功能的期望是：当某个 Kyuubi 服务实例（Pod）故障时，正在执行的 SQL 查询能够自动转移到其他健康的实例上继续执行。然而，Kyuubi 当前的 HA 实现机制有所不同：

• 当前实现：HA 主要解决的是服务发现和负载均衡问题，而非查询任务的故障转移
• 会话管理：当 Kyuubi 实例故障时，与其关联的所有会话都会被标记为无效，正在执行的查询会被取消
• 客户端责任：客户端需要捕获异常并重新执行失败的查询

技术解决方案

1. 正确配置多引擎环境

对于需要同时支持多种查询引擎的场景，建议配置方式如下：

1. 在 kyuubi-defaults.conf 中不要设置默认引擎类型（注释掉 kyuubi.engine.type）
2. 通过 JDBC URL 的 # 后参数明确指定每次连接使用的引擎类型
3. 为不同引擎配置相应的参数前缀（如 ___batch3___. 为 Spark SQL 配置前缀）

2. 高可用性最佳实践

虽然 Kyuubi 目前不支持查询任务的自动故障转移，但可以通过以下方式提高系统可用性：

1. Spark 集群模式：设置 spark.submit.deployMode=cluster，使 Spark 驱动程序运行在独立的 Pod 中

   • 这样即使 Kyuubi Pod 终止，Spark 引擎仍可继续运行
   • 但需要注意，会话和查询状态仍会丢失，需要客户端重新连接

2. 客户端重试机制：

   • 实现自动重试逻辑，捕获连接异常后重新建立会话并执行查询
   • 设置合理的重试次数和退避策略

3. 资源隔离：

   • 为不同用户/组配置独立的引擎实例（通过 kyuubi.engine.share.level=GROUP）
   • 避免单点故障影响所有用户

未来改进方向

从技术架构角度看，实现真正的分布式会话和查询故障转移需要考虑：

1. 外部状态存储：将会话和操作状态持久化到 Redis、MySQL 或 ZooKeeper 等外部存储中
2. 状态同步机制：实现跨实例的状态同步和恢复协议
3. 查询检查点：支持大型查询的中间状态保存和恢复

总结

Apache Kyuubi 作为多引擎 SQL 服务网关，在复杂生产环境中的配置需要特别注意参数传递方式和高可用性实现的边界。当前版本中：

• 引擎选择必须通过 # 符号后的参数正确指定
• HA 功能主要解决服务发现而非查询连续性
• Spark 集群模式可以提供一定程度的引擎进程容错能力

对于关键业务场景，建议在客户端实现健壮的重试机制，并合理规划资源隔离策略，以平衡系统可用性和资源利用率。随着社区的发展，未来版本有望提供更完善的分布式会话支持，进一步简化高可用性架构的实现。