扩展聚合语句中支持“group by level=n”的功能
Group by level语句中可以支持多个聚合函数的存在。
实现方案
将FilePathUtils.getPathByLevel()的接口为放到AggregationPlan中,pathIndex作为AggregationPlan的成员。
Map<String, AggregateResult> getPathByLevel()
返回值finalPaths的类型是Map<String, AggregateResult>,Key为由聚合函数和序列名拼接而成的字符串,如avg(root.sg.d1.s1),用于区分多个聚合函数的相同序列。该字符串会显示为查询结果的列名。
IoTDB> select count(temperature),avg(temperature) from root.sg.*.* group by level=2
+-----------------------------+-----------------------------+---------------------------+---------------------------+
|count(root.sg.d1.temperature)|count(root.sg.d2.temperature)|avg(root.sg.d1.temperature)|avg(root.sg.d2.temperature)|
+-----------------------------+-----------------------------+---------------------------+---------------------------+
| 2| 2| 114.0| 45.915000915527344|
+-----------------------------+-----------------------------+------------------- -------+---------------------------+
考虑到需要支持多种聚合函数及多种数据类型的需求,将mergeRecordByPath接受的参数类型改为中间类型AggregateResult,修改为:
RowRecord mergeRecordByPath(List<AggregateResult> aggResults, Map<String, AggregateResult> finalPaths, Map<Integer, String>pathIndex)
方法逻辑
mergeRecordByPath的参数意义:
aggResults为对所有序列进行聚合后的结果列表,即一阶段聚合的结果集。
finalPaths的key为最终返回序列名,value为二次聚合计算结果, 需要在该方法中计算得到。
pathIndex为一个映射,映射内容为<聚合结果在aggResults列表中的index, 所属的返回序列名>,用于定位一阶段聚合结果在最终聚合结果集中的位置。
基本逻辑为:对于aggResults中的每一个聚合结果,从pathIndex中找到对应的最终返回序列,对于位于同一返回序列的聚合结果进行二次聚合操作。二次聚合的操作交由AggregateResult类型计算,见AggregateResult.java中的merge()方法。
结果集 Header修改
修改cli返回的result header
如下图中,改为count(root.ln.wf01..temperature)和count(root.ln.wf02..temperature)