Apache Kyuubi中Presto连接器表名大小写匹配问题解析

问题背景

在Apache Kyuubi项目中，当使用Presto连接器查询Oracle等数据库时，发现了一个关于表名大小写匹配的问题。Oracle数据库中的表名通常使用大写字母命名，而Presto返回的错误信息中表名却转换为小写，导致正则表达式匹配失败，最终抛出未捕获的异常。

技术细节分析

问题的核心在于表存在性检查时的正则表达式匹配逻辑。当前实现中，代码使用严格区分大小写的正则表达式来匹配错误信息：

regex = r"Table\ \'.*{}\'\ does\ not\ exist".format(re.escape(table_name))

当Oracle数据库中有一个名为"MY_TABLE"的表时，如果用户查询该表但表不存在，Presto可能会返回类似"Table 'my_table' does not exist"的错误信息。由于正则表达式区分大小写，导致匹配失败，进而无法正确抛出NoSuchTableError异常。

影响范围

该问题主要影响以下场景：

1. 使用Presto连接器查询Oracle数据库
2. 数据库表名包含大写字母
3. 执行表存在性检查操作

类似的问题也出现在连接Databricks时，因为Databricks返回的错误信息格式不同（如"[TABLE_OR_VIEW_NOT_FOUND]"），同样会导致正则匹配失败。

解决方案

修复方案相对简单直接：在正则匹配时添加re.IGNORECASE标志，使匹配过程不区分大小写。这样可以确保无论错误信息中的表名是大写、小写还是混合大小写，都能正确匹配。

修改后的代码示例如下：

if msg and re.search(regex, msg, re.IGNORECASE):
    raise exc.NoSuchTableError(table_name)
else:
    raise

实现验证

为了确保修复的有效性，开发团队添加了相应的单元测试(UT)来验证：

1. 模拟Oracle表名大小写场景
2. 验证Presto返回小写表名错误信息时的处理
3. 确保修复后能正确抛出NoSuchTableError异常

总结

这个问题的修复虽然代码改动不大，但对于提升Kyuubi与Presto连接器在各种数据库环境下的兼容性具有重要意义。特别是在企业环境中，Oracle等数据库广泛使用大写表名，这个修复确保了在这些场景下表存在性检查功能的可靠性。

对于用户来说，建议关注相关版本的发布，及时升级以获得这一修复。同时，如果在使用过程中遇到类似的表名大小写相关问题，可以考虑检查是否由此类匹配问题引起。