This is an automated email from the ASF dual-hosted git repository. hui pushed a commit to branch ml/windowSet in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit ceeb55dc8bc867cd5c9b161afa602cda9c71c9e8 Author: Minghui Liu <[email protected]> AuthorDate: Tue Nov 1 20:30:51 2022 +0800 simple test --- .../iotdb/session/it/IoTDBFetchWindowSetIT.java | 187 +++++++++++++++++++++ 1 file changed, 187 insertions(+) diff --git a/integration-test/src/test/java/org/apache/iotdb/session/it/IoTDBFetchWindowSetIT.java b/integration-test/src/test/java/org/apache/iotdb/session/it/IoTDBFetchWindowSetIT.java new file mode 100644 index 0000000000..1e31bd5c83 --- /dev/null +++ b/integration-test/src/test/java/org/apache/iotdb/session/it/IoTDBFetchWindowSetIT.java @@ -0,0 +1,187 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.iotdb.session.it; + +import org.apache.iotdb.it.env.EnvFactory; +import org.apache.iotdb.it.framework.IoTDBTestRunner; +import org.apache.iotdb.itbase.category.ClusterIT; +import org.apache.iotdb.session.ISession; +import org.apache.iotdb.session.SessionDataSet; + +import org.junit.AfterClass; +import org.junit.Assert; +import org.junit.BeforeClass; +import org.junit.Test; +import org.junit.experimental.categories.Category; +import org.junit.runner.RunWith; + +import java.sql.Connection; +import java.sql.Statement; +import java.util.Arrays; +import java.util.List; + +import static org.junit.Assert.fail; + +@RunWith(IoTDBTestRunner.class) +@Category({ClusterIT.class}) +public class IoTDBFetchWindowSetIT { + + private static final String[] sqls = + new String[] { + "SET STORAGE GROUP TO root.sg1", + "create aligned timeseries root.sg1.d1(s1 FLOAT encoding=RLE, s2 INT32 encoding=Gorilla compression=SNAPPY, s3 INT64, s4 BOOLEAN, s5 TEXT)", + "create timeseries root.sg1.d2.s1 WITH DATATYPE=FLOAT, encoding=RLE", + "create timeseries root.sg1.d2.s2 WITH DATATYPE=INT32, encoding=Gorilla", + "create timeseries root.sg1.d2.s3 WITH DATATYPE=INT64", + "create timeseries root.sg1.d2.s4 WITH DATATYPE=BOOLEAN", + "create timeseries root.sg1.d2.s5 WITH DATATYPE=TEXT", + "insert into root.sg1.d1(time, s1, s2, s3, s4, s5) aligned values(1, 1.0, 1, 1, TRUE, 'aligned_test1')", + "insert into root.sg1.d1(time, s1, s2, s3, s5) aligned values(2, 2.0, 2, 2, 'aligned_test2')", + "insert into root.sg1.d1(time, s1, s3, s4, s5) aligned values(3, 3.0, 3, FALSE, 'aligned_test3')", + "insert into root.sg1.d1(time, s1, s2, s4, s5) aligned values(4, 4.0, 4, TRUE, 'aligned_test4')", + "insert into root.sg1.d1(time, s1, s2, s4, s5) aligned values(5, 5.0, 5, TRUE, 'aligned_test5')", + "insert into root.sg1.d1(time, s1, s2, s3, s4) aligned values(6, 6.0, 6, 6, TRUE)", + "insert into root.sg1.d1(time, s1, s2, s3, s4, s5) aligned values(7, 7.0, 7, 7, FALSE, 'aligned_test7')", + "insert into root.sg1.d1(time, s1, s2, s3, s5) aligned values(8, 8.0, 8, 8, 'aligned_test8')", + "insert into root.sg1.d1(time, s1, s2, s3, s4, s5) aligned values(9, 9.0, 9, 9, FALSE, 'aligned_test9')", + "insert into root.sg1.d1(time, s2, s3, s4, s5) aligned values(10, 10, 10, TRUE, 'aligned_test10')", + "insert into root.sg1.d2(time, s1, s2, s3, s4, s5) values(1, 1.0, 1, 1, TRUE, 'non_aligned_test1')", + "insert into root.sg1.d2(time, s1, s2, s3, s5) values(2, 2.0, 2, 2, 'non_aligned_test2')", + "insert into root.sg1.d2(time, s1, s3, s4, s5) values(3, 3.0, 3, FALSE, 'non_aligned_test3')", + "insert into root.sg1.d2(time, s1, s2, s4, s5) values(4, 4.0, 4, TRUE, 'non_aligned_test4')", + "insert into root.sg1.d2(time, s1, s2, s4, s5) values(5, 5.0, 5, TRUE, 'non_aligned_test5')", + "insert into root.sg1.d2(time, s1, s2, s3, s4) values(6, 6.0, 6, 6, TRUE)", + "insert into root.sg1.d2(time, s1, s2, s3, s4, s5) values(7, 7.0, 7, 7, FALSE, 'non_aligned_test7')", + "insert into root.sg1.d2(time, s1, s2, s3, s5) values(8, 8.0, 8, 8, 'non_aligned_test8')", + "insert into root.sg1.d2(time, s1, s2, s3, s4, s5) values(9, 9.0, 9, 9, FALSE, 'non_aligned_test9')", + "insert into root.sg1.d2(time, s2, s3, s4, s5) values(10, 10, 10, TRUE, 'non_aligned_test10')", + "flush", + "insert into root.sg1.d1(time, s1, s3, s4, s5) aligned values(3, 30000.0, 30000, TRUE, 'aligned_unseq_test3')", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(11, 11.0, 11, 11)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(12, 12.0, 12, 12)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(13, 13.0, 13, 13)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(14, 14.0, 14, 14)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(15, 15.0, 15, 15)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(16, 16.0, 16, 16)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(17, 17.0, 17, 17)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(18, 18.0, 18, 18)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(19, 19.0, 19, 19)", + "insert into root.sg1.d1(time, s1, s2, s3) aligned values(20, 20.0, 20, 20)", + "insert into root.sg1.d2(time, s1, s2, s3) values(11, 11.0, 11, 11)", + "insert into root.sg1.d2(time, s1, s2, s3) values(12, 12.0, 12, 12)", + "insert into root.sg1.d2(time, s1, s2, s3) values(13, 13.0, 13, 13)", + "insert into root.sg1.d2(time, s1, s2, s3) values(14, 14.0, 14, 14)", + "insert into root.sg1.d2(time, s1, s2, s3) values(15, 15.0, 15, 15)", + "insert into root.sg1.d2(time, s1, s2, s3) values(16, 16.0, 16, 16)", + "insert into root.sg1.d2(time, s1, s2, s3) values(17, 17.0, 17, 17)", + "insert into root.sg1.d2(time, s1, s2, s3) values(18, 18.0, 18, 18)", + "insert into root.sg1.d2(time, s1, s2, s3) values(19, 19.0, 19, 19)", + "insert into root.sg1.d2(time, s1, s2, s3) values(20, 20.0, 20, 20)", + "flush", + "insert into root.sg1.d1(time, s1, s2, s3, s4, s5) aligned values(13, 130000.0, 130000, 130000, TRUE, 'aligned_unseq_test13')", + "insert into root.sg1.d1(time, s3, s4) aligned values(21, 21, TRUE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(22, 22, TRUE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(23, 23, TRUE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(24, 24, TRUE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(25, 25, TRUE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(26, 26, FALSE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(27, 27, FALSE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(28, 28, FALSE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(29, 29, FALSE)", + "insert into root.sg1.d1(time, s3, s4) aligned values(30, 30, FALSE)", + "insert into root.sg1.d2(time, s3, s4) values(21, 21, TRUE)", + "insert into root.sg1.d2(time, s3, s4) values(22, 22, TRUE)", + "insert into root.sg1.d2(time, s3, s4) values(23, 23, TRUE)", + "insert into root.sg1.d2(time, s3, s4) values(24, 24, TRUE)", + "insert into root.sg1.d2(time, s3, s4) values(25, 25, TRUE)", + "insert into root.sg1.d2(time, s3, s4) values(26, 26, FALSE)", + "insert into root.sg1.d2(time, s3, s4) values(27, 27, FALSE)", + "insert into root.sg1.d2(time, s3, s4) values(28, 28, FALSE)", + "insert into root.sg1.d2(time, s3, s4) values(29, 29, FALSE)", + "insert into root.sg1.d2(time, s3, s4) values(30, 30, FALSE)", + "flush", + "insert into root.sg1.d1(time, s1, s3, s4) aligned values(23, 230000.0, 230000, FALSE)", + "insert into root.sg1.d1(time, s2, s5) aligned values(31, 31, 'aligned_test31')", + "insert into root.sg1.d1(time, s2, s5) aligned values(32, 32, 'aligned_test32')", + "insert into root.sg1.d1(time, s2, s5) aligned values(33, 33, 'aligned_test33')", + "insert into root.sg1.d1(time, s2, s5) aligned values(34, 34, 'aligned_test34')", + "insert into root.sg1.d1(time, s2, s5) aligned values(35, 35, 'aligned_test35')", + "insert into root.sg1.d1(time, s2, s5) aligned values(36, 36, 'aligned_test36')", + "insert into root.sg1.d1(time, s2, s5) aligned values(37, 37, 'aligned_test37')", + "insert into root.sg1.d1(time, s2, s5) aligned values(38, 38, 'aligned_test38')", + "insert into root.sg1.d1(time, s2, s5) aligned values(39, 39, 'aligned_test39')", + "insert into root.sg1.d1(time, s2, s5) aligned values(40, 40, 'aligned_test40')", + "insert into root.sg1.d2(time, s2, s5) values(31, 31, 'non_aligned_test31')", + "insert into root.sg1.d2(time, s2, s5) values(32, 32, 'non_aligned_test32')", + "insert into root.sg1.d2(time, s2, s5) values(33, 33, 'non_aligned_test33')", + "insert into root.sg1.d2(time, s2, s5) values(34, 34, 'non_aligned_test34')", + "insert into root.sg1.d2(time, s2, s5) values(35, 35, 'non_aligned_test35')", + "insert into root.sg1.d2(time, s2, s5) values(36, 36, 'non_aligned_test36')", + "insert into root.sg1.d2(time, s2, s5) values(37, 37, 'non_aligned_test37')", + "insert into root.sg1.d2(time, s2, s5) values(38, 38, 'non_aligned_test38')", + "insert into root.sg1.d2(time, s2, s5) values(39, 39, 'non_aligned_test39')", + "insert into root.sg1.d2(time, s2, s5) values(40, 40, 'non_aligned_test40')", + }; + + public static void insertData() { + try (Connection connection = EnvFactory.getEnv().getConnection(); + Statement statement = connection.createStatement()) { + + // create aligned and non-aligned time series + // TODO change it to executeBatch way when it's supported in new cluster + for (String sql : sqls) { + statement.execute(sql); + } + } catch (Exception e) { + e.printStackTrace(); + fail(e.getMessage()); + } + } + + @BeforeClass + public static void setUp() throws Exception { + EnvFactory.getEnv().initBeforeClass(); + insertData(); + } + + @AfterClass + public static void tearDown() throws Exception { + EnvFactory.getEnv().cleanAfterClass(); + } + + @Test + public void fetchWindowSetTest() { + try (ISession session = EnvFactory.getEnv().getSessionConnection()) { + List<String> queryPaths = Arrays.asList("root.sg1.d1.s1", "root.sg1.d2.s1"); + List<Integer> indexes = Arrays.asList(1, 3, 5, 7); + List<SessionDataSet> windowSet = + session.fetchWindowSet(queryPaths, null, 0, 20, 2, 2, indexes); + Assert.assertEquals(indexes.size(), windowSet.size()); + for (SessionDataSet window : windowSet) { + while (window.hasNext()) { + System.out.println(window.next()); + } + } + } catch (Exception e) { + e.printStackTrace(); + fail(e.getMessage()); + } + } +}
