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

问题背景

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

问题分析

问题的核心在于表名检查逻辑中的正则表达式匹配部分。当前实现中，代码使用严格的大小写敏感匹配来检查"Table 'table_name' does not exist"这样的错误信息。当表名在数据库中为大写（如"MY_TABLE"），而Presto返回的错误信息中表名为小写（如"my_table"）时，正则表达式匹配就会失败。

技术细节

错误处理逻辑中的关键代码如下：

regex = r"Table\ \'.*{}\'\ does\ not\ exist".format(re.escape(table_name))
            
if msg and re.search(regex, msg):
    raise exc.NoSuchTableError(table_name)
else:
    raise

当table_name为大写（如"MY_TABLE"），而msg中包含小写表名（如"Table 'my_table' does not exist"）时，re.search()会返回None，导致进入else分支抛出原始异常，而不是预期的NoSuchTableError。

解决方案

针对这个问题，社区提出了在正则表达式匹配时添加re.IGNORECASE标志的解决方案。这个修改可以确保无论错误信息中的表名是大写还是小写，都能正确匹配。

修改后的代码应该类似：

regex = r"Table\ \'.*{}\'\ does\ not\ exist".format(re.escape(table_name))
            
if msg and re.search(regex, msg, re.IGNORECASE):
    raise exc.NoSuchTableError(table_name)
else:
    raise

类似问题扩展

值得注意的是，这个问题不仅存在于Presto连接Oracle的场景。在连接Databricks时也存在类似问题，因为Databricks返回的错误信息格式不同（如"[TABLE_OR_VIEW_NOT_FOUND]"），同样会导致正则表达式匹配失败。这说明错误处理机制可能需要更全面的改进，以支持不同数据源返回的不同错误信息格式。

总结

这个问题的修复虽然简单，但体现了数据库连接层处理跨数据源兼容性的重要性。在实际应用中，不同数据库系统对于标识符大小写的处理方式各不相同，连接器需要能够适应这些差异。Apache Kyuubi社区通过添加大小写不敏感的匹配，解决了Presto连接器与Oracle等数据库的兼容性问题，提高了系统的稳定性和用户体验。