[Enhancement](ctas) Support create auto range partition table with CTAS command

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/commands/CreateTableCommand.java

@@ -109,6 +109,7 @@ public void run(ConnectContext ctx, StmtExecutor executor) throws Exception {
         if (slots.size() != ctasCols.size()) {
             throw new AnalysisException("ctas column size is not equal to the query's");
         }
+        String autoRangePartitionName = getAutoRangePartitionNameOrNull();
         ImmutableList.Builder<ColumnDefinition> columnsOfQuery = ImmutableList.builder();
         for (int i = 0; i < slots.size(); i++) {
             Slot s = slots.get(i);
@@ -154,8 +155,14 @@ public void run(ConnectContext ctx, StmtExecutor executor) throws Exception {
                     }
                 }
             }
-            // if the column is an expression, we set it to nullable, otherwise according to the nullable of the slot.
-            columnsOfQuery.add(new ColumnDefinition(s.getName(), dataType, !s.isColumnFromTable() || s.nullable()));
+            if (autoRangePartitionName != null && autoRangePartitionName.equalsIgnoreCase(s.getName())) {
+                // for auto range partition column, it must be not nullable. so keep its origin.
+                columnsOfQuery.add(new ColumnDefinition(s.getName(), dataType, s.nullable()));
+            } else {
+                // if the column is an expression, we set it to nullable, otherwise according to the nullable of the
+                // slot.
+                columnsOfQuery.add(new ColumnDefinition(s.getName(), dataType, !s.isColumnFromTable() || s.nullable()));
+            }
         }
         List<String> qualifierTableName = RelationUtil.getQualifierName(ctx, createTableInfo.getTableNameParts());
         createTableInfo.validateCreateTableAsSelect(qualifierTableName, columnsOfQuery.build(), ctx);
@@ -199,6 +206,20 @@ void handleFallbackFailedCtas(ConnectContext ctx) {
         }
     }
 
+    private String getAutoRangePartitionNameOrNull() {
+        try {
+            if (createTableInfo.getPartitionTableInfo().isAutoPartition()
+                    && createTableInfo.getPartitionTableInfo().getPartitionType().equalsIgnoreCase("RANGE")) {
+                // should collect first before use them.
+                createTableInfo.getPartitionTableInfo().extractPartitionColumns();
+                return createTableInfo.getPartitionTableInfo().getIdentifierPartitionColumns().get(0);
+            }
+        } catch (Exception e) {
+            return null;
+        }
+        return null;
+    }
+
     public boolean isCtasCommand() {
         return ctasQuery.isPresent();
     }

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/commands/info/PartitionTableInfo.java

@@ -329,6 +329,7 @@ public void extractPartitionColumns() throws AnalysisException {
         try {
             identifierPartitionColumns = PartitionDesc.getColNamesFromExpr(exprs,
                     partitionType.equalsIgnoreCase(PartitionType.LIST.name()), isAutoPartition);
+            // check of it will be done in validatePartitionInfo later.
         } catch (Exception e) {
             throw new AnalysisException(e.getMessage(), e.getCause());
         }
@@ -337,4 +338,8 @@ public void extractPartitionColumns() throws AnalysisException {
     public boolean inIdentifierPartitions(String columnName) {
         return identifierPartitionColumns != null && identifierPartitionColumns.contains(columnName);
     }
+
+    public List<String> getIdentifierPartitionColumns() {
+        return identifierPartitionColumns;
+    }
 }

regression-test/suites/nereids_p0/create_table/test_ctas_auto_partition.groovy

@@ -0,0 +1,94 @@
+// 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("test_ctas_auto_partition") {
+    sql "drop table if exists test_partition__dbt_tmp3"
+    test {
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select nullable(current_date()) as today;
+        """
+        exception "AUTO RANGE PARTITION doesn't support NULL column"
+    }
+
+    test {
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select cast(null as date) as today;
+        """
+        exception "AUTO RANGE PARTITION doesn't support NULL column"
+    }
+
+    test {
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select 1 as today;
+        """
+        exception "partition expr date_trunc is illegal!"
+    }
+
+    test{
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day'), date_trunc(`today1`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select 1 as today;
+        """
+        exception "auto create partition only support one slotRef in function expr"
+    }
+
+    test{
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select current_date() as today, current_date() + 1 as today;
+        """
+        exception "Duplicate column name 'today'"
+    }
+
+    test{
+        sql """
+            create table `test_partition__dbt_tmp3`
+            AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+            PROPERTIES ("replication_num" = "1" )
+            as
+            select current_date() as today2, current_date() + 1 as today1;
+        """
+        exception "partition key today is not exists"
+    }
+
+    sql """
+        create table `test_partition__dbt_tmp3`
+        AUTO PARTITION BY RANGE (date_trunc(`today`, 'day')) ()
+        PROPERTIES ("replication_num" = "1" )
+        as
+        select 1 as id, 'cyril' as name, 18 as age, 100 as score, current_date() as today;
+    """
+    assertEquals((sql "select count() from test_partition__dbt_tmp3")[0][0], 1)
+}