功能描述


扩展聚合语句中支持“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)

  • No labels