This is an automated email from the ASF dual-hosted git repository. caogaofei pushed a commit to branch beyyes/count_time_fe in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 0df0870f4dd56ae41d3e1030ad486c63096945a2 Author: Beyyes <[email protected]> AuthorDate: Thu Jul 27 14:42:03 2023 +0800 add test code for count time --- .../java/org/apache/iotdb/AggregationExample.java | 49 ++++++++++++++++++++++ .../datanode1conf/iotdb-common.properties | 1 + .../datanode2conf/iotdb-common.properties | 1 + .../datanode3conf/iotdb-common.properties | 2 + .../udf/builtin/BuiltinAggregationFunction.java | 4 +- 5 files changed, 56 insertions(+), 1 deletion(-) diff --git a/example/session/src/main/java/org/apache/iotdb/AggregationExample.java b/example/session/src/main/java/org/apache/iotdb/AggregationExample.java new file mode 100644 index 00000000000..81ea2089fd3 --- /dev/null +++ b/example/session/src/main/java/org/apache/iotdb/AggregationExample.java @@ -0,0 +1,49 @@ +package org.apache.iotdb; + +import org.apache.iotdb.common.rpc.thrift.TAggregationType; +import org.apache.iotdb.isession.SessionDataSet; +import org.apache.iotdb.isession.util.Version; +import org.apache.iotdb.rpc.IoTDBConnectionException; +import org.apache.iotdb.rpc.StatementExecutionException; +import org.apache.iotdb.session.Session; + +import java.util.ArrayList; +import java.util.List; + +public class AggregationExample { + + private static Session session; + + private static final String LOCAL_HOST = "127.0.0.1"; + + public static void main(String[] args) throws IoTDBConnectionException { + session = + new Session.Builder() + .host(LOCAL_HOST) + .port(6668) + .username("root") + .password("root") + .version(Version.V_1_0) + .build(); + session.open(false); + + List<String> paths = new ArrayList<>(); + paths.add("root.cpu.d1.**"); + // paths.add("root.cpu.d1.s1"); + // paths.add("root.cpu.d1.s1"); + + List<TAggregationType> aggregations = new ArrayList<>(); + aggregations.add(TAggregationType.COUNT); + // aggregations.add(TAggregationType.MAX_VALUE); + // aggregations.add(TAggregationType.COUNT_TIME); + try (SessionDataSet sessionDataSet = session.executeAggregationQuery(paths, aggregations)) { + System.out.println(sessionDataSet.getColumnNames()); + sessionDataSet.setFetchSize(1024); + while (sessionDataSet.hasNext()) { + System.out.println(sessionDataSet.next()); + } + } catch (StatementExecutionException e) { + throw new RuntimeException(e); + } + } +} diff --git a/iotdb-core/datanode/src/test/resources/datanode1conf/iotdb-common.properties b/iotdb-core/datanode/src/test/resources/datanode1conf/iotdb-common.properties index ba222515546..81deacc7880 100644 --- a/iotdb-core/datanode/src/test/resources/datanode1conf/iotdb-common.properties +++ b/iotdb-core/datanode/src/test/resources/datanode1conf/iotdb-common.properties @@ -21,3 +21,4 @@ timestamp_precision=ms udf_lib_dir=target/datanode1/ext/udf trigger_lib_dir=target/datanode1/ext/trigger pipe_lib_dir=target/datanode1/ext/pipe +enable_data_partition_inherit_policy=false diff --git a/iotdb-core/datanode/src/test/resources/datanode2conf/iotdb-common.properties b/iotdb-core/datanode/src/test/resources/datanode2conf/iotdb-common.properties index d75d1b52021..9226f2b261d 100644 --- a/iotdb-core/datanode/src/test/resources/datanode2conf/iotdb-common.properties +++ b/iotdb-core/datanode/src/test/resources/datanode2conf/iotdb-common.properties @@ -21,3 +21,4 @@ timestamp_precision=ms udf_lib_dir=target/datanode2/ext/udf trigger_lib_dir=target/datanode2/ext/trigger pipe_lib_dir=target/datanode2/ext/pipe +enable_data_partition_inherit_policy=false diff --git a/iotdb-core/datanode/src/test/resources/datanode3conf/iotdb-common.properties b/iotdb-core/datanode/src/test/resources/datanode3conf/iotdb-common.properties index ebd7d4fa59f..3098cb6ae22 100644 --- a/iotdb-core/datanode/src/test/resources/datanode3conf/iotdb-common.properties +++ b/iotdb-core/datanode/src/test/resources/datanode3conf/iotdb-common.properties @@ -21,3 +21,5 @@ timestamp_precision=ms udf_lib_dir=target/datanode3/ext/udf trigger_lib_dir=target/datanode3/ext/trigger pipe_lib_dir=target/datanode3/ext/pipe +enable_data_partition_inherit_policy=false + diff --git a/iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/udf/builtin/BuiltinAggregationFunction.java b/iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/udf/builtin/BuiltinAggregationFunction.java index 43181be7aba..a1b8a6560e9 100644 --- a/iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/udf/builtin/BuiltinAggregationFunction.java +++ b/iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/udf/builtin/BuiltinAggregationFunction.java @@ -37,7 +37,8 @@ public enum BuiltinAggregationFunction { SUM("sum"), COUNT_IF("count_if"), TIME_DURATION("time_duration"), - MODE("mode"); + MODE("mode"), + COUNT_TIME("count_time"); private final String functionName; @@ -77,6 +78,7 @@ public enum BuiltinAggregationFunction { return true; case "count_if": case "mode": + case "count_time": return false; default: throw new IllegalArgumentException("Invalid Aggregation function: " + name);
