This is an automated email from the ASF dual-hosted git repository.
yiguolei pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/master by this push:
new 1e02de1846b [feature](nestedType)add nested data type to create table
(#26129)
1e02de1846b is described below
commit 1e02de1846b3d1fe564daaf648cb24b6c6f95ad6
Author: zfr95 <87513668+zfr9...@users.noreply.github.com>
AuthorDate: Tue Oct 31 13:35:29 2023 +0800
[feature](nestedType)add nested data type to create table (#26129)
---
.../plugins_create_table_nested_type.groovy | 92 ++++++++++++++++++++++
.../datatype_p0/nested_types/create_table.groovy | 24 ++++++
2 files changed, 116 insertions(+)
diff --git a/regression-test/plugins/plugins_create_table_nested_type.groovy
b/regression-test/plugins/plugins_create_table_nested_type.groovy
new file mode 100644
index 00000000000..9c830795875
--- /dev/null
+++ b/regression-test/plugins/plugins_create_table_nested_type.groovy
@@ -0,0 +1,92 @@
+// 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.
+
+import org.apache.doris.regression.suite.Suite
+
+// create table with nested data type, now default complex data include array,
map, struct
+Suite.metaClass.create_table_with_nested_type = { int maxDepth, String tbName
/* param */ ->
+ Suite suite = delegate as Suite
+
+ maxDepth = maxDepth > 9 ? 9 : maxDepth
+
+ def dataTypeArr = ["boolean", "tinyint(4)", "smallint(6)", "int(11)",
"bigint(20)", "largeint(40)", "float",
+ "double", "decimal(20, 3)", "decimalv3(20, 3)", "date",
"datetime", "datev2", "datetimev2(0)",
+ "char(15)", "varchar(100)", "text"]
+ def colNameArr = ["c_bool", "c_tinyint", "c_smallint", "c_int",
"c_bigint", "c_largeint", "c_float",
+ "c_double", "c_decimal", "c_decimalv3", "c_date",
"c_datetime", "c_datev2", "c_datetimev2",
+ "c_char", "c_varchar", "c_string"]
+ def complexDataTypeArr = ["array", "map", "struct"]
+// def tbName = "test"
+ def colCount = 1
+ def memo = new String[20]
+ def r = new Random()
+
+ def getDataType
+ getDataType = { dataType, level ->
+ if (memo[level] != null) {
+ return memo[level];
+ }
+
+ StringBuilder res = new StringBuilder();
+ def data_r = r.nextInt(3);
+
+ if (level == 1) {
+ if (data_r == 0) {
+ res.append(complexDataTypeArr[data_r]+"<"+dataType+">");
+ } else if (data_r == 1) {
+ res.append(complexDataTypeArr[data_r]+"<"+dataType+","
+dataType+">");
+ } else if (data_r == 2) {
+ res.append(complexDataTypeArr[data_r]+"<col_"+colCount+":" +
dataType +">");
+ colCount++;
+ }
+ } else {
+ level--;
+ if (data_r == 0) {
+
res.append(complexDataTypeArr[data_r]+"<"+getDataType(dataType, level)+">");
+ } else if (data_r == 1) {
+// String str = getDataType(dataType, level);
+
res.append(complexDataTypeArr[data_r]+"<"+getDataType(dataType, level)+","
+getDataType(dataType, level)+">")
+ } else if (data_r == 2) {
+ res.append(complexDataTypeArr[data_r]+"<col_"+colCount+":" +
getDataType(dataType, level) +">");
+ colCount++;
+ }
+ }
+ memo[level] = res.toString()
+ return memo[level];
+ }
+
+ def stmt = "CREATE TABLE IF NOT EXISTS " + tbName + "(\n" +
+ "`k1` bigint(11) NULL,\n"
+ String strTmp = "`" + colNameArr[0] + "` " + getDataType(dataTypeArr[0],
maxDepth) + " NULL,\n";
+ stmt += strTmp
+ for (int i = 1; i < dataTypeArr.size(); i++) {
+ String changeDataType = strTmp.replaceAll(colNameArr[0], colNameArr[i])
+ changeDataType = changeDataType.replaceAll(dataTypeArr[0],
dataTypeArr[i])
+ stmt += changeDataType
+ }
+ stmt = stmt.substring(0, stmt.length()-2)
+ stmt += ") ENGINE=OLAP\n" +
+ "DUPLICATE KEY(`k1`)\n" +
+ "COMMENT 'OLAP'\n" +
+ "DISTRIBUTED BY HASH(`k1`) BUCKETS 10\n" +
+ "PROPERTIES(\"replication_num\" = \"1\");"
+ logger.info(stmt)
+ return stmt
+}
+
+logger.info("Added 'create_table_with_nested_type' function to Suite")
+
diff --git
a/regression-test/suites/datatype_p0/nested_types/create_table.groovy
b/regression-test/suites/datatype_p0/nested_types/create_table.groovy
new file mode 100644
index 00000000000..8fe270cb33f
--- /dev/null
+++ b/regression-test/suites/datatype_p0/nested_types/create_table.groovy
@@ -0,0 +1,24 @@
+// 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.
+
+suite("create_table_with_nested_type") {
+ // register testPlugin function in
${DORIS_HOME}/regression-test/plugins/plugins_create_table_nested_type.groovy
+ // input the nested depth and table name, the output is table schema
+ // run "bash run-regression-test.sh --run -s
create_table_with_nested_type" and watch the logger
+ def result = create_table_with_nested_type(1, "test")
+ logger.info(result)
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org
