You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by mi...@apache.org on 2023/09/27 02:58:15 UTC
[iotdb] branch rel/1.2_rest_it created (now 9d55b640d39)
This is an automated email from the ASF dual-hosted git repository.
miao pushed a change to branch rel/1.2_rest_it
in repository https://gitbox.apache.org/repos/asf/iotdb.git
at 9d55b640d39 it test
This branch includes the following new commits:
new 5bb36b68ac5 it test
new 9d55b640d39 it test
The 2 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
[iotdb] 02/02: it test
Posted by mi...@apache.org.
This is an automated email from the ASF dual-hosted git repository.
miao pushed a commit to branch rel/1.2_rest_it
in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 9d55b640d392d4bb56dcc1d5bf76524713c8d5e6
Author: Cloudwise_Luke <28...@qq.com>
AuthorDate: Wed Sep 27 10:43:38 2023 +0800
it test
---
.../org/apache/iotdb/db/it/IoTDBRestServiceIT.java | 2035 ++++++++++++++++++++
1 file changed, 2035 insertions(+)
diff --git a/integration-test/src/test/java/org/apache/iotdb/db/it/IoTDBRestServiceIT.java b/integration-test/src/test/java/org/apache/iotdb/db/it/IoTDBRestServiceIT.java
new file mode 100644
index 00000000000..69af02939bd
--- /dev/null
+++ b/integration-test/src/test/java/org/apache/iotdb/db/it/IoTDBRestServiceIT.java
@@ -0,0 +1,2035 @@
+/*
+ * 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.db.it;
+
+import org.apache.iotdb.it.env.EnvFactory;
+import org.apache.iotdb.it.env.cluster.DataNodeWrapper;
+import org.apache.iotdb.it.framework.IoTDBTestRunner;
+import org.apache.iotdb.itbase.category.ClusterIT;
+import org.apache.iotdb.itbase.category.LocalStandaloneIT;
+import org.apache.iotdb.itbase.category.RemoteIT;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import org.apache.flink.table.api.EnvironmentSettings;
+import org.apache.http.HttpEntity;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.StringEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.util.EntityUtils;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.experimental.categories.Category;
+import org.junit.runner.RunWith;
+
+import java.io.IOException;
+import java.nio.charset.Charset;
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.Base64;
+import java.util.List;
+import java.util.Map;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+@RunWith(IoTDBTestRunner.class)
+@Category({LocalStandaloneIT.class, ClusterIT.class, RemoteIT.class})
+public class IoTDBRestServiceIT {
+
+ private int port = 18080;
+
+ @Before
+ public void setUp() throws Exception {
+
+ EnvFactory.getEnv().initClusterEnvironment();
+ EnvironmentSettings settings = EnvironmentSettings.newInstance().inStreamingMode().build();
+ DataNodeWrapper portConflictDataNodeWrapper = EnvFactory.getEnv().getDataNodeWrapper(0);
+ port = portConflictDataNodeWrapper.getRestServicePort();
+ }
+
+ @After
+ public void tearDown() throws Exception {
+ EnvFactory.getEnv().cleanClusterEnvironment();
+ }
+
+ private String getAuthorization(String username, String password) {
+ return Base64.getEncoder()
+ .encodeToString((username + ":" + password).getBytes(StandardCharsets.UTF_8));
+ }
+
+ @Test
+ public void ping() {
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ HttpGet httpGet = new HttpGet("http://127.0.0.1:" + port + "/ping");
+ CloseableHttpResponse response = null;
+ try {
+ response = httpClient.execute(httpGet);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(200, response.getStatusLine().getStatusCode());
+ assertEquals(200, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (httpClient != null) {
+ httpClient.close();
+ }
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ private HttpPost getHttpPost(String url) {
+ HttpPost httpPost = new HttpPost(url);
+ httpPost.addHeader("Content-type", "application/json; charset=utf-8");
+ httpPost.setHeader("Accept", "application/json");
+ String authorization = getAuthorization("root", "root");
+ httpPost.setHeader("Authorization", authorization);
+ return httpPost;
+ }
+
+ private HttpPost getHttpPost_1(String url) {
+ HttpPost httpPost = new HttpPost(url);
+ httpPost.addHeader("Content-type", "application/json; charset=utf-8");
+ httpPost.setHeader("Accept", "application/json");
+ String authorization = getAuthorization("root1", "root1");
+ httpPost.setHeader("Authorization", authorization);
+ return httpPost;
+ }
+
+ public void nonQuery(CloseableHttpClient httpClient, String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(200, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void errorNonQuery(CloseableHttpClient httpClient, String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(700, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void rightInsertRecords(CloseableHttpClient httpClient, String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(200, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void errorInsertRecords(CloseableHttpClient httpClient, String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(509, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void rightInsertTablet(CloseableHttpClient httpClient, String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(200, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void rightInsertTablet(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/insertTablet");
+ String json =
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"isAligned\":false,\"deviceId\":\"root.sg25\"}";
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(200, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void errorInsertTablet(String json, HttpPost httpPost) {
+ CloseableHttpResponse response = null;
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ try {
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(606, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ @Test
+ public void errorInsertTablet() {
+ CloseableHttpResponse response = null;
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/insertTablet");
+ String json =
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"isAligned\":fal [...]
+ httpPost.setEntity(new StringEntity(json, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(413, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void perData(CloseableHttpClient httpClient) {
+ HttpPost httpPost2 = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/nonQuery");
+ HttpPost httpPostV2 = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/nonQuery");
+
+ nonQuery(httpClient, "{\"sql\":\"CREATE USER `root1` 'root1'\"}", httpPost2);
+ nonQuery(
+ httpClient, "{\"sql\":\"GRANT USER `root1` PRIVILEGES WRITE on root.**\"}", httpPostV2);
+ }
+
+ @Test
+ public void insertAndQuery() {
+
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ //
+
+ rightInsertTablet(httpClient);
+ query(httpClient);
+ queryGroupByLevel(httpClient);
+ queryRowLimit(httpClient);
+ queryShowChildPaths(httpClient);
+ queryShowNodes(httpClient);
+ showAllTTL(httpClient);
+ showStorageGroup(httpClient);
+ showFunctions(httpClient);
+ showTimeseries(httpClient);
+
+ showLastTimeseries(httpClient);
+ countTimeseries(httpClient);
+ countNodes(httpClient);
+ showDevices(httpClient);
+
+ showDevicesWithStroage(httpClient);
+ listUser(httpClient);
+ selectCount(httpClient);
+ selectLast(httpClient);
+
+ queryV2(httpClient);
+ queryGroupByLevelV2(httpClient);
+ queryRowLimitV2(httpClient);
+ queryShowChildPathsV2(httpClient);
+ queryShowNodesV2(httpClient);
+ showAllTTLV2(httpClient);
+ showStorageGroupV2(httpClient);
+ showFunctionsV2(httpClient);
+ showTimeseriesV2(httpClient);
+
+ showLastTimeseriesV2(httpClient);
+ countTimeseriesV2(httpClient);
+ countNodesV2(httpClient);
+ showDevicesV2(httpClient);
+
+ showDevicesWithStroageV2(httpClient);
+ listUserV2(httpClient);
+ selectCountV2(httpClient);
+ selectLastV2(httpClient);
+ perData(httpClient);
+ List<String> insertTablet_right_json_list = new ArrayList<>();
+ List<String> insertTablet_error_json_list = new ArrayList<>();
+
+ List<String> insertRecords_right_json_list = new ArrayList<>();
+ List<String> insertRecords_error_json_list = new ArrayList<>();
+
+ List<String> nonQuery_right_json_list = new ArrayList<>();
+ List<String> nonQuery_error_json_list = new ArrayList<>();
+
+ List<String> insertTablet_right_json_list_v2 = new ArrayList<>();
+ List<String> insertTablet_error_json_list_v2 = new ArrayList<>();
+
+ List<String> insertRecords_right_json_list_v2 = new ArrayList<>();
+ List<String> insertRecords_error_json_list_v2 = new ArrayList<>();
+
+ List<String> nonQuery_right_json_list_v2 = new ArrayList<>();
+ List<String> nonQuery_error_json_list_v2 = new ArrayList<>();
+
+ for (int i = 0; i <= 1; i++) {
+ boolean isAligned = false;
+ if (i == 0) {
+ isAligned = true;
+ }
+ insertTablet_right_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"isAligned\":\""
+ + isAligned
+ + "\",\"deviceId\":\"root.sg21"
+ + i
+ + "\"}");
+ insertTablet_right_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"isAligned\":\""
+ + isAligned
+ + "\",\"deviceId\":\"root.sg22"
+ + i
+ + "\"}");
+ insertTablet_right_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"isAligned\":\""
+ + isAligned
+ + "\",\"deviceId\":\"root.`sg23"
+ + i
+ + "`\"}");
+ insertTablet_right_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"isAligned\":\""
+ + isAligned
+ + "\",\"deviceId\":\"root.`sg24"
+ + i
+ + "`\"}");
+ insertTablet_error_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"isAligned\":"
+ + isAligned
+ + ",\"deviceId\":\"root.sg25"
+ + i
+ + "\"}");
+ insertTablet_error_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"isAligned\":"
+ + isAligned
+ + ",\"deviceId\":\"root.sg26"
+ + i
+ + "\"}");
+ insertTablet_error_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"123123\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"isAligned\":"
+ + isAligned
+ + ",\"deviceId\":\"root.`sg26"
+ + i
+ + "`\"}");
+ insertTablet_error_json_list.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`1231231`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"dataTypes\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"isAligned\":"
+ + isAligned
+ + ",\"deviceId\":\"root.`3333a"
+ + i
+ + "`\"}");
+ insertRecords_right_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"s33\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.s3\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"`s33`\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.s3\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"s33\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.`s3`\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"`s33`\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.`s11`\",\"root.s11\",\"root.s1\",\"root.`s3`\"],\"isAligned\":"
+ + isAligned
+ + "}");
+
+ insertRecords_error_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"root\",\"442\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.time\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"time\",\"123123\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.time\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"time\",\"12321\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.111\",\"root.`s11`\",\"root.s1\",\"root.s3\"],\"isAligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurementsList\":[[\"`root`\",\"1111\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"dataTypesList\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"valuesList\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"deviceIds\":[\"root.s11\",\"root.s11\",\"root.time\",\"root.`s3`\"],\"isAligned\":"
+ + isAligned
+ + "}");
+
+ nonQuery_right_json_list.add("{\"sql\":\"insert into root.aa(time,`bb`) values(111,1)\"}");
+ nonQuery_right_json_list.add("{\"sql\":\"insert into root.`aa`(time,bb) values(111,1)\"}");
+ nonQuery_right_json_list.add("{\"sql\":\"insert into root.`aa`(time,`bb`) values(111,1)\"}");
+ nonQuery_right_json_list.add("{\"sql\":\"insert into root.aa(time,bb) values(111,1)\"}");
+
+ nonQuery_error_json_list.add("{\"sql\":\"insert into root.time(time,1`bb`) values(111,1)\"}");
+ nonQuery_error_json_list.add(
+ "{\"sql\":\"insert into root.time.`aa`(time,bb) values(111,1)\"}");
+ nonQuery_error_json_list.add(
+ "{\"sql\":\"insert into root.time.`aa`(time,`bb`) values(111,1)\"}");
+ nonQuery_error_json_list.add("{\"sql\":\"insert into root.aa(time,root) values(111,1)\"}");
+
+ insertTablet_right_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"is_aligned\":\""
+ + isAligned
+ + "\",\"device\":\"root.sg21"
+ + i
+ + "\"}");
+ insertTablet_right_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.sg22"
+ + i
+ + "\"}");
+ insertTablet_right_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.`sg23"
+ + i
+ + "`\"}");
+ insertTablet_right_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[11,2],[1635000012345555,1635000012345556],[1.41,null],[null,false],[null,3.5555]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.`sg24"
+ + i
+ + "`\"}");
+ insertTablet_error_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"s3\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.sg25"
+ + i
+ + "\"}");
+ insertTablet_error_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`s3`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.sg26"
+ + i
+ + "\"}");
+ insertTablet_error_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"123123\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.`sg26"
+ + i
+ + "`\"}");
+ insertTablet_error_json_list_v2.add(
+ "{\"timestamps\":[1635232143960,1635232153960],\"measurements\":[\"`1231231`\",\"s4\",\"s5\",\"s6\",\"s7\",\"s8\"],\"data_types\":[\"TEXT\",\"INT32\",\"INT64\",\"FLOAT\",\"BOOLEAN\",\"DOUBLE\"],\"values\":[[\"2aa\",\"\"],[111111112312312442352545452323123,2],[16,15],[1.41,null],[null,false],[null,3.55555555555555555555555555555555555555555555312234235345123127318927461482308478123645555555555555555555555555555555555555555555531223423534512312731892746148230847812364]],\"is_aligned\":"
+ + isAligned
+ + ",\"device\":\"root.`3333a"
+ + i
+ + "`\"}");
+ insertRecords_right_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"s33\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.s3\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"`s33`\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.s3\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"s33\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.`s3`\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_right_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"`s33`\",\"s44\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[1,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.`s11`\",\"root.s11\",\"root.s1\",\"root.`s3`\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+
+ insertRecords_error_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"root\",\"442\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.time\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"time\",\"123123\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.s1\",\"root.time\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"time\",\"12321\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.111\",\"root.`s11`\",\"root.s1\",\"root.s3\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+ insertRecords_error_json_list_v2.add(
+ "{\"timestamps\":[1635232113960,1635232151960,1123,10],\"measurements_list\":[[\"`root`\",\"1111\"],[\"s55\",\"s66\"],[\"s77\",\"s88\"],[\"s771\",\"s881\"]],\"data_types_list\":[[\"INT32\",\"INT64\"],[\"float\",\"double\"],[\"float\",\"double\"],[\"boolean\",\"text\"]],\"values_list\":[[true,11],[2.1,2],[4,6],[false,\"cccccc\"]],\"devices\":[\"root.s11\",\"root.s11\",\"root.time\",\"root.`s3`\"],\"is_aligned\":"
+ + isAligned
+ + "}");
+
+ nonQuery_right_json_list_v2.add("{\"sql\":\"insert into root.aa(time,`bb`) values(111,1)\"}");
+ nonQuery_right_json_list_v2.add("{\"sql\":\"insert into root.`aa`(time,bb) values(111,1)\"}");
+ nonQuery_right_json_list_v2.add(
+ "{\"sql\":\"insert into root.`aa`(time,`bb`) values(111,1)\"}");
+ nonQuery_right_json_list_v2.add("{\"sql\":\"insert into root.aa(time,bb) values(111,1)\"}");
+
+ nonQuery_error_json_list_v2.add(
+ "{\"sql\":\"insert into root.time(time,1`bb`) values(111,1)\"}");
+ nonQuery_error_json_list_v2.add(
+ "{\"sql\":\"insert into root.time.`aa`(time,bb) values(111,1)\"}");
+ nonQuery_error_json_list_v2.add(
+ "{\"sql\":\"insert into root.time.`aa`(time,`bb`) values(111,1)\"}");
+ nonQuery_error_json_list_v2.add("{\"sql\":\"insert into root.aa(time,root) values(111,1)\"}");
+ }
+
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/insertTablet");
+ HttpPost httpPost1 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v1/insertTablet");
+
+ List<HttpPost> httpPosts = new ArrayList<>();
+ httpPosts.add(httpPost);
+ httpPosts.add(httpPost1);
+ for (HttpPost hp : httpPosts) {
+ for (String json : insertTablet_right_json_list) {
+ rightInsertTablet(httpClient, json, hp);
+ }
+ for (String json : insertTablet_error_json_list) {
+ errorInsertTablet(json, hp);
+ }
+ }
+
+ HttpPost httpPost3 = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/nonQuery");
+ HttpPost httpPost4 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v1/nonQuery");
+ List<HttpPost> httpPosts1 = new ArrayList<>();
+ httpPosts1.add(httpPost3);
+ httpPosts1.add(httpPost4);
+
+ for (HttpPost hp : httpPosts1) {
+ for (String json : nonQuery_right_json_list) {
+ nonQuery(httpClient, json, hp);
+ }
+ for (String json : nonQuery_error_json_list) {
+ errorNonQuery(httpClient, json, hp);
+ }
+ }
+
+ HttpPost httpPost5 = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/insertRecords");
+ HttpPost httpPost6 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v1/insertRecords");
+ List<HttpPost> httpPosts2 = new ArrayList<>();
+ httpPosts2.add(httpPost5);
+ httpPosts2.add(httpPost6);
+
+ for (HttpPost hp : httpPosts2) {
+ for (String json : insertRecords_right_json_list) {
+ rightInsertRecords(httpClient, json, hp);
+ }
+
+ for (String json : insertRecords_error_json_list) {
+ errorInsertRecords(httpClient, json, hp);
+ }
+ }
+
+ HttpPost httpPostV2 = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/insertTablet");
+ HttpPost httpPost1V2 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v2/insertTablet");
+
+ List<HttpPost> httpPostsV2 = new ArrayList<>();
+ httpPostsV2.add(httpPostV2);
+ httpPostsV2.add(httpPost1V2);
+ for (HttpPost hp : httpPostsV2) {
+ for (String json : insertTablet_right_json_list_v2) {
+ rightInsertTablet(httpClient, json, hp);
+ }
+ for (String json : insertTablet_error_json_list_v2) {
+ errorInsertTablet(json, hp);
+ }
+ }
+
+ HttpPost httpPost3V2 = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/nonQuery");
+ HttpPost httpPost4V2 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v2/nonQuery");
+ List<HttpPost> httpPosts1V2 = new ArrayList<>();
+ httpPosts1V2.add(httpPost3V2);
+ httpPosts1V2.add(httpPost4V2);
+
+ for (HttpPost hp : httpPosts1V2) {
+ for (String json : nonQuery_right_json_list_v2) {
+ nonQuery(httpClient, json, hp);
+ }
+ for (String json : nonQuery_error_json_list_v2) {
+ errorNonQuery(httpClient, json, hp);
+ }
+ }
+
+ HttpPost httpPost5V2 = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/insertRecords");
+ HttpPost httpPost6V2 = getHttpPost_1("http://127.0.0.1:" + port + "/rest/v2/insertRecords");
+ List<HttpPost> httpPosts2V2 = new ArrayList<>();
+ httpPosts2V2.add(httpPost5V2);
+ httpPosts2V2.add(httpPost6V2);
+
+ for (HttpPost hp : httpPosts2V2) {
+ for (String json : insertRecords_right_json_list_v2) {
+ rightInsertRecords(httpClient, json, hp);
+ }
+
+ for (String json : insertRecords_error_json_list_v2) {
+ errorInsertRecords(httpClient, json, hp);
+ }
+ }
+
+ try {
+ httpClient.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+
+ @Test
+ public void queryWithUnsetAuthorization() {
+ CloseableHttpResponse response = null;
+ try {
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ HttpPost httpPost = new HttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ httpPost.addHeader("Content-type", "application/json; charset=utf-8");
+ httpPost.setHeader("Accept", "application/json");
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ Assert.assertEquals(401, response.getStatusLine().getStatusCode());
+ String message = EntityUtils.toString(response.getEntity(), "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(603, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ @Test
+ public void queryWithWrongAuthorization() {
+ CloseableHttpResponse response = null;
+ try {
+ CloseableHttpClient httpClient = HttpClientBuilder.create().build();
+ HttpPost httpPost = new HttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ httpPost.addHeader("Content-type", "application/json; charset=utf-8");
+ httpPost.setHeader("Accept", "application/json");
+ String authorization = getAuthorization("abc", "def");
+ httpPost.setHeader("Authorization", authorization);
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ Assert.assertEquals(401, response.getStatusLine().getStatusCode());
+ String message = EntityUtils.toString(response.getEntity(), "utf-8");
+ JsonObject result = JsonParser.parseString(message).getAsJsonObject();
+ assertEquals(600, Integer.parseInt(result.get("code").toString()));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void query(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ List<Long> expressionsResult = (List<Long>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> expressions =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s6");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ add("root.sg25.s4 + 1");
+ add("root.sg25.s4 + 1");
+ }
+ };
+ List<Object> timestamps =
+ new ArrayList<Object>() {
+ {
+ add(1635232143960l);
+ add(1635232153960l);
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("2aa");
+ add("");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(11);
+ add(2);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add(1635000012345555l);
+ add(1635000012345556l);
+ }
+ };
+
+ List<Object> values4 =
+ new ArrayList<Object>() {
+ {
+ add(1.41);
+ add(null);
+ }
+ };
+ List<Object> values5 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(false);
+ }
+ };
+ List<Object> values6 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(3.5555);
+ }
+ };
+
+ Assert.assertEquals(expressions, expressionsResult);
+ Assert.assertEquals(timestamps, timestampsResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ Assert.assertEquals(values4, valuesResult.get(3));
+ Assert.assertEquals(values5, valuesResult.get(4));
+ Assert.assertEquals(values6, valuesResult.get(5));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void queryGroupByLevel(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ String sql =
+ "{\"sql\":\"select count(s4) from root.sg25 group by([1635232143960,1635232153960),1s),level=1\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ List<Long> expressionsResult = (List<Long>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ Assert.assertTrue(timestampsResult.size() == 10);
+ Assert.assertTrue(valuesResult.size() == 1);
+ Assert.assertTrue("count(root.sg25.s4)".equals(expressionsResult.get(0)));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void queryRowLimit(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\",\"rowLimit\":1}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ assertTrue(message.contains("row size exceeded the given max row size"));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public Map queryMetaData(CloseableHttpClient httpClient, String sql) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v1/query");
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ return map;
+
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ return null;
+ }
+
+ public void queryShowChildPaths(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show child paths root\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("ChildPaths");
+ add("NodeTypes");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void queryShowNodes(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show child nodes root\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("ChildNodes");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showAllTTL(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show all ttl\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Database");
+ add("TTL");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ }
+
+ public void showStorageGroup(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"SHOW DATABASES root.*\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Database");
+ add("TTL");
+ add("SchemaReplicationFactor");
+ add("DataReplicationFactor");
+ add("TimePartitionInterval");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showFunctions(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show functions\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ assertEquals(3, columnNamesResult.size());
+ assertEquals(3, valuesResult.size());
+ }
+
+ public void showTimeseries(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show timeseries\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Alias");
+ add("Database");
+ add("DataType");
+ add("Encoding");
+ add("Compression");
+ add("Tags");
+ add("Attributes");
+ add("Deadband");
+ add("DeadbandParameters");
+ add("ViewType");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s6");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void showLastTimeseries(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"SHOW LATEST TIMESERIES\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Alias");
+ add("Database");
+ add("DataType");
+ add("Encoding");
+ add("Compression");
+ add("Tags");
+ add("Attributes");
+ add("Deadband");
+ add("DeadbandParameters");
+ add("ViewType");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ add("root.sg25.s6");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void countTimeseries(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"COUNT TIMESERIES root.** GROUP BY LEVEL=1\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Column");
+ add("count(timeseries)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(6);
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ }
+
+ public void countNodes(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"count nodes root.** level=2\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("count(nodes)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add(6);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showDevices(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show devices\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Device");
+ add("IsAligned");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Boolean> values2 =
+ new ArrayList<Boolean>() {
+ {
+ add(false);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showDevicesWithStroage(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show devices with database\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Device");
+ add("Database");
+ add("IsAligned");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("false");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void listUser(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"list user\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("columnNames");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("user");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void selectCount(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"select count(s3) from root.** group by level = 1\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("count(root.sg25.s3)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add(2);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void selectLast(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"select last s4 from root.sg25\"}";
+ Map map = queryMetaData(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Value");
+ add("DataType");
+ }
+ };
+ List<Long> timestamps =
+ new ArrayList<Long>() {
+ {
+ add(1635232153960l);
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s4");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add("2");
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("INT32");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(timestamps, timestampsResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2.get(0), valuesResult.get(1).get(0));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void queryV2(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/query");
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ List<Long> expressionsResult = (List<Long>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> expressions =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s6");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ add("root.sg25.s4 + 1");
+ add("root.sg25.s4 + 1");
+ }
+ };
+ List<Object> timestamps =
+ new ArrayList<Object>() {
+ {
+ add(1635232143960l);
+ add(1635232153960l);
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("2aa");
+ add("");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(11);
+ add(2);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add(1635000012345555l);
+ add(1635000012345556l);
+ }
+ };
+
+ List<Object> values4 =
+ new ArrayList<Object>() {
+ {
+ add(1.41);
+ add(null);
+ }
+ };
+ List<Object> values5 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(false);
+ }
+ };
+ List<Object> values6 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(3.5555);
+ }
+ };
+
+ Assert.assertEquals(expressions, expressionsResult);
+ Assert.assertEquals(timestamps, timestampsResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ Assert.assertEquals(values4, valuesResult.get(3));
+ Assert.assertEquals(values5, valuesResult.get(4));
+ Assert.assertEquals(values6, valuesResult.get(5));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void queryGroupByLevelV2(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/query");
+ String sql =
+ "{\"sql\":\"select count(s4) from root.sg25 group by([1635232143960,1635232153960),1s),level=1\"}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ List<Long> expressionsResult = (List<Long>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ Assert.assertTrue(timestampsResult.size() == 10);
+ Assert.assertTrue(valuesResult.size() == 1);
+ Assert.assertTrue("count(root.sg25.s4)".equals(expressionsResult.get(0)));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public void queryRowLimitV2(CloseableHttpClient httpClient) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/query");
+ String sql = "{\"sql\":\"select *,s4+1,s4+1 from root.sg25\",\"row_limit\":1}";
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ assertTrue(message.contains("row size exceeded the given max row size"));
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ }
+
+ public Map queryMetaDataV2(CloseableHttpClient httpClient, String sql) {
+ CloseableHttpResponse response = null;
+ try {
+ HttpPost httpPost = getHttpPost("http://127.0.0.1:" + port + "/rest/v2/query");
+ httpPost.setEntity(new StringEntity(sql, Charset.defaultCharset()));
+ response = httpClient.execute(httpPost);
+ HttpEntity responseEntity = response.getEntity();
+ String message = EntityUtils.toString(responseEntity, "utf-8");
+ ObjectMapper mapper = new ObjectMapper();
+ Map map = mapper.readValue(message, Map.class);
+ return map;
+
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ try {
+ if (response != null) {
+ response.close();
+ }
+ } catch (IOException e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ }
+ }
+ return null;
+ }
+
+ public void queryShowChildPathsV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show child paths root\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("ChildPaths");
+ add("NodeTypes");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void queryShowNodesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show child nodes root\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("ChildNodes");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showAllTTLV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show all ttl\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Database");
+ add("TTL");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ }
+
+ public void showStorageGroupV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"SHOW DATABASES root.*\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Database");
+ add("TTL");
+ add("SchemaReplicationFactor");
+ add("DataReplicationFactor");
+ add("TimePartitionInterval");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showFunctionsV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show functions\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ assertEquals(3, columnNamesResult.size());
+ assertEquals(3, valuesResult.size());
+ }
+
+ public void showTimeseriesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show timeseries\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Alias");
+ add("Database");
+ add("DataType");
+ add("Encoding");
+ add("Compression");
+ add("Tags");
+ add("Attributes");
+ add("Deadband");
+ add("DeadbandParameters");
+ add("ViewType");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s6");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ // Assert.assertEquals(values1, valuesResult.get(0));
+ // Assert.assertEquals(values2, valuesResult.get(1));
+ // Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void showLastTimeseriesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"SHOW LATEST TIMESERIES\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Alias");
+ add("Database");
+ add("DataType");
+ add("Encoding");
+ add("Compression");
+ add("Tags");
+ add("Attributes");
+ add("Deadband");
+ add("DeadbandParameters");
+ add("ViewType");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s3");
+ add("root.sg25.s4");
+ add("root.sg25.s5");
+ add("root.sg25.s7");
+ add("root.sg25.s8");
+ add("root.sg25.s6");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ add(null);
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ add("root.sg25");
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void countTimeseriesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"COUNT TIMESERIES root.** GROUP BY LEVEL=1\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Column");
+ add("count(timeseries)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add(6);
+ }
+ };
+
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ }
+
+ public void countNodesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"count nodes root.** level=2\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("count(nodes)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add(6);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void showDevicesV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show devices\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Device");
+ add("IsAligned");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Boolean> values2 =
+ new ArrayList<Boolean>() {
+ {
+ add(false);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ // Assert.assertEquals(values2, valuesResult.get(1));
+ }
+
+ public void showDevicesWithStroageV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"show devices with database\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Device");
+ add("Database");
+ add("IsAligned");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25");
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("false");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2, valuesResult.get(1));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+
+ public void listUserV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"list user\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("column_names");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("user");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void selectCountV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"select count(s3) from root.** group by level = 1\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("count(root.sg25.s3)");
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add(2);
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ }
+
+ public void selectLastV2(CloseableHttpClient httpClient) {
+ String sql = "{\"sql\":\"select last s4 from root.sg25\"}";
+ Map map = queryMetaDataV2(httpClient, sql);
+ List<String> columnNamesResult = (List<String>) map.get("expressions");
+ List<List<Object>> valuesResult = (List<List<Object>>) map.get("values");
+ List<Long> timestampsResult = (List<Long>) map.get("timestamps");
+ Assert.assertTrue(map.size() > 0);
+ List<Object> columnNames =
+ new ArrayList<Object>() {
+ {
+ add("Timeseries");
+ add("Value");
+ add("DataType");
+ }
+ };
+ List<Long> timestamps =
+ new ArrayList<Long>() {
+ {
+ add(1635232153960l);
+ }
+ };
+ List<Object> values1 =
+ new ArrayList<Object>() {
+ {
+ add("root.sg25.s4");
+ }
+ };
+ List<Object> values2 =
+ new ArrayList<Object>() {
+ {
+ add("2");
+ }
+ };
+ List<Object> values3 =
+ new ArrayList<Object>() {
+ {
+ add("INT32");
+ }
+ };
+ Assert.assertEquals(columnNames, columnNamesResult);
+ Assert.assertEquals(timestamps, timestampsResult);
+ Assert.assertEquals(values1, valuesResult.get(0));
+ Assert.assertEquals(values2.get(0), valuesResult.get(1).get(0));
+ Assert.assertEquals(values3, valuesResult.get(2));
+ }
+}
[iotdb] 01/02: it test
Posted by mi...@apache.org.
This is an automated email from the ASF dual-hosted git repository.
miao pushed a commit to branch rel/1.2_rest_it
in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 5bb36b68ac50f181cbbaf8f5e36e556758ed3e45
Author: Cloudwise_Luke <28...@qq.com>
AuthorDate: Wed Sep 27 10:34:48 2023 +0800
it test
---
integration-test/pom.xml | 28 ++++++++++++++++++++++
.../iotdb/it/env/cluster/DataNodeWrapper.java | 16 +++++++++++++
.../org/apache/iotdb/it/env/cluster/EnvUtils.java | 2 +-
3 files changed, 45 insertions(+), 1 deletion(-)
diff --git a/integration-test/pom.xml b/integration-test/pom.xml
index e05fbd30c9a..c9b6163dd15 100644
--- a/integration-test/pom.xml
+++ b/integration-test/pom.xml
@@ -326,6 +326,34 @@
<groupId>org.apache.logging.log4j</groupId>
<artifactId>log4j-slf4j-impl</artifactId>
</exclusion>
+ <exclusion>
+ <groupId>com.sun.jersey</groupId>
+ <artifactId>jersey-core</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.core</groupId>
+ <artifactId>jersey-client</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.core</groupId>
+ <artifactId>jersey-common</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.core</groupId>
+ <artifactId>jersey-server</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.containers</groupId>
+ <artifactId>jersey-container-servlet</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.containers</groupId>
+ <artifactId>jersey-container-servlet-core</artifactId>
+ </exclusion>
+ <exclusion>
+ <groupId>org.glassfish.jersey.inject</groupId>
+ <artifactId>jersey-hk2</artifactId>
+ </exclusion>
</exclusions>
</dependency>
<dependency>
diff --git a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/DataNodeWrapper.java b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/DataNodeWrapper.java
index c9e828c4795..d65db7a170b 100644
--- a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/DataNodeWrapper.java
+++ b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/DataNodeWrapper.java
@@ -33,6 +33,8 @@ public class DataNodeWrapper extends AbstractNodeWrapper {
private final int schemaRegionConsensusPort;
private final int mqttPort;
+ private final int restServicePort;
+
private final String defaultNodePropertiesFile =
EnvUtils.getFilePathFromSysVar("DefaultDataNodeProperties");
private final String defaultCommonPropertiesFile =
@@ -46,7 +48,9 @@ public class DataNodeWrapper extends AbstractNodeWrapper {
this.internalPort = portList[2];
this.dataRegionConsensusPort = portList[3];
this.schemaRegionConsensusPort = portList[4];
+
this.mqttPort = portList[5];
+ this.restServicePort = portList[10] + 6000;
// Initialize mutable properties
reloadMutableFields();
@@ -55,6 +59,11 @@ public class DataNodeWrapper extends AbstractNodeWrapper {
// Override mqtt properties of super class
immutableCommonProperties.setProperty("mqtt_host", super.getIp());
immutableCommonProperties.setProperty("mqtt_port", String.valueOf(this.mqttPort));
+ immutableCommonProperties.setProperty("rest_service_port", String.valueOf(restServicePort));
+ immutableCommonProperties.setProperty("enable_rest_service", "true");
+ immutableCommonProperties.setProperty("enable_swagger", "false");
+ immutableCommonProperties.setProperty("rest_query_default_row_size_limit", "10000");
+ immutableCommonProperties.setProperty("cache_expire_in_seconds", "28800");
immutableNodeProperties.setProperty(IoTDBConstant.DN_TARGET_CONFIG_NODE_LIST, targetConfigNode);
immutableNodeProperties.setProperty("dn_system_dir", MppBaseConfig.NULL_VALUE);
@@ -140,6 +149,9 @@ public class DataNodeWrapper extends AbstractNodeWrapper {
mutableCommonProperties.put("max_tsblock_size_in_bytes", "1024");
mutableCommonProperties.put("page_size_in_byte", "1024");
+ mutableCommonProperties.put("enable_rest_service", "true");
+ mutableCommonProperties.put("rest_service_port", String.valueOf(this.restServicePort));
+
mutableNodeProperties.put("dn_join_cluster_retry_interval_ms", "1000");
mutableNodeProperties.put("dn_connection_timeout_ms", "30000");
mutableNodeProperties.put("dn_metric_internal_reporter_type", "MEMORY");
@@ -207,6 +219,10 @@ public class DataNodeWrapper extends AbstractNodeWrapper {
return dataRegionConsensusPort;
}
+ public int getRestServicePort() {
+ return restServicePort;
+ }
+
public int getSchemaRegionConsensusPort() {
return schemaRegionConsensusPort;
}
diff --git a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/EnvUtils.java b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/EnvUtils.java
index 48c08e69c2e..4cbef9981a5 100644
--- a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/EnvUtils.java
+++ b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/EnvUtils.java
@@ -49,7 +49,7 @@ public class EnvUtils {
continue;
}
List<Integer> requiredPorts =
- IntStream.rangeClosed(randomPortStart, randomPortStart + 9)
+ IntStream.rangeClosed(randomPortStart, randomPortStart + 10)
.boxed()
.collect(Collectors.toList());
if (checkPortsAvailable(requiredPorts)) {