DuckDB数据库GROUP BY ROLLUP优化导致查询结果异常问题分析

在数据库查询优化过程中，DuckDB的DUPLICATE_GROUPS优化器在某些特定场景下会导致GROUP BY ROLLUP操作产生不正确的结果。本文将深入分析这一问题的技术细节和解决方案。

问题现象

当使用GROUP BY ROLLUP配合JOIN操作时，启用DUPLICATE_GROUPS优化会导致查询结果出现重复行和错误的分组聚合值。具体表现为：

1. 在启用优化时，结果集中出现了不应该存在的重复行
2. 分组聚合的NULL占位符位置不正确
3. ROLLUP的分层聚合结果不符合预期

技术背景

GROUP BY ROLLUP是SQL中实现多层次聚合的重要功能，它会按照指定的列顺序生成从最详细到最汇总的多层次聚合结果。DuckDB的DUPLICATE_GROUPS优化原本是为了消除重复分组计算而设计的性能优化手段。

问题复现

通过以下测试用例可以稳定复现该问题：

-- 创建测试表
CREATE TABLE t1(col1 INT, col2 INT);
CREATE TABLE t2(col3 INT);

-- 插入测试数据
INSERT INTO t1 VALUES (1, 1);
INSERT INTO t2 VALUES (1);

-- 问题查询
SELECT col1, col2, col3 
FROM t1 JOIN t2 ON t1.col1 = t2.col3 
GROUP BY ROLLUP(col1, col2, col3) 
ORDER BY 1, 2, 3;

结果对比

启用优化时的错误结果：

1, 1, 1
1, 1, 1  ← 错误重复行
1, NULL, 1 ← 错误的分组
NULL, NULL, NULL

禁用优化时的正确结果：

1, 1, 1
1, 1, NULL ← 正确的部分汇总
1, NULL, NULL ← 正确的更高层汇总
NULL, NULL, NULL

问题根源分析

经过代码分析，DUPLICATE_GROUPS优化在处理ROLLUP操作时存在以下问题：

1. 未能正确处理JOIN后产生的中间结果与ROLLUP分组的对应关系
2. 在消除重复分组时错误保留了本应被聚合的行
3. 对NULL值在分组中的处理逻辑存在缺陷

解决方案

开发团队已经确认该问题并提交了修复方案。主要修改包括：

1. 增强DUPLICATE_GROUPS优化对ROLLUP操作的特殊处理
2. 完善JOIN与GROUP BY联合使用时的优化器逻辑
3. 修正分组聚合中NULL值的处理方式

最佳实践建议

对于需要使用GROUP BY ROLLUP的场景，建议：

1. 在DuckDB 1.2.2版本中暂时禁用DUPLICATE_GROUPS优化
2. 升级到包含修复的新版本
3. 对关键业务查询进行结果验证
4. 考虑使用WITH ROLLUP语法替代GROUP BY ROLLUP

总结

数据库查询优化器与复杂SQL特性的交互常常会产生微妙的边界条件问题。这次DuckDB的DUPLICATE_GROUPS优化与ROLLUP的兼容性问题提醒我们，在使用高级SQL功能时需要特别注意查询结果的正确性验证。开发团队已经迅速响应并修复了该问题，体现了开源项目的快速迭代优势。