[Enhancement] Support alter table add roll up in shared data (#53153)

Signed-off-by: smartlxh <[email protected]>

fe/fe-core/src/main/java/com/starrocks/alter/MaterializedViewHandler.java

@@ -947,9 +947,6 @@ private void getAlterJobV2Infos(Database db, List<List<Comparable>> rollupJobInf
     @Override
     public ShowResultSet process(List<AlterClause> alterClauses, Database db, OlapTable olapTable)
             throws DdlException, AnalysisException, MetaNotFoundException {
-        if (olapTable.isCloudNativeTable()) {
-            throw new DdlException("Does not support add rollup on lake table");
-        }
         if (olapTable.existTempPartitions()) {
             throw new DdlException("Can not alter table when there are temp partitions in table");
         }

test/sql/test_lake_rollup/R/test_lake_rollup

@@ -0,0 +1,38 @@
+-- name: test_lake_rollup @cloud
+CREATE TABLE tbl1 (
+    k1 date,
+    k2 int,
+    v1 int sum
+) PARTITION BY RANGE(k1) (
+    PARTITION p1 values [('2020-01-01'),('2020-02-01')),
+    PARTITION p2 values [('2020-02-01'),('2020-03-01')))
+    DISTRIBUTED BY HASH(k2) BUCKETS 3
+    PROPERTIES('replication_num' = '1');
+-- result:
+-- !result
+
+INSERT INTO tbl1
+VALUES
+    ("2020-01-12",4,100),
+    ("2020-01-11",5,100),
+    ("2020-01-11",4,100);
+-- result:
+-- !result
+
+ALTER TABLE tbl1 ADD ROLLUP rollup1 (k1, v1) FROM tbl1;
+-- result:
+-- !result
+function: wait_alter_table_finish("ROLLUP")
+-- result:
+None
+-- !result
+
+INSERT INTO tbl1 VALUES("2020-01-11",6,100);
+-- result:
+-- !result
+
+SELECT v1 FROM rollup1 [_SYNC_MV_] ORDER BY v1;
+-- result:
+100
+300
+-- !result

test/sql/test_lake_rollup/T/test_lake_rollup

@@ -0,0 +1,23 @@
+-- name: test_lake_rollup @cloud
+CREATE TABLE tbl1 (
+    k1 date,
+    k2 int,
+    v1 int sum
+) PARTITION BY RANGE(k1) (
+    PARTITION p1 values [('2020-01-01'),('2020-02-01')),
+    PARTITION p2 values [('2020-02-01'),('2020-03-01')))
+    DISTRIBUTED BY HASH(k2) BUCKETS 3
+    PROPERTIES('replication_num' = '1');
+
+INSERT INTO tbl1
+VALUES
+    ("2020-01-12",4,100),
+    ("2020-01-11",5,100),
+    ("2020-01-11",4,100);
+
+ALTER TABLE tbl1 ADD ROLLUP rollup1 (k1, v1) FROM tbl1;
+function: wait_alter_table_finish("ROLLUP")
+
+INSERT INTO tbl1 VALUES("2020-01-11",6,100);
+
+SELECT v1 FROM rollup1 [_SYNC_MV_] ORDER BY v1;

test/sql/test_lake_sync_mv/R/test_lake_sync_mv

@@ -1,4 +1,4 @@
--- name: test_invalid_json @cloud
+-- name: test_lake_sync_mv @cloud
 CREATE TABLE tbl1 (
     k1 date,
     k2 int,
@@ -8,13 +8,31 @@ CREATE TABLE tbl1 (
     PARTITION p2 values [('2020-02-01'),('2020-03-01')))
     DISTRIBUTED BY HASH(k2) BUCKETS 3
     PROPERTIES('replication_num' = '1');
+-- result:
+-- !result
 
 INSERT INTO tbl1
 VALUES
     ("2020-01-12",4,100),
     ("2020-01-11",5,100),
     ("2020-01-11",4,100);
+-- result:
+-- !result
 
 CREATE MATERIALIZED VIEW sync_mv AS SELECT k1, SUM(v1) FROM tbl1 GROUP BY k1;
+-- result:
+-- !result
+function: wait_materialized_view_finish()
+-- result:
+None
+-- !result
 
-INSERT INTO tbl1 VALUES("2020-01-11",6,100);
+INSERT INTO tbl1 VALUES("2020-01-11",6,100);
+-- result:
+-- !result
+
+SELECT mv_sum_v1 FROM sync_mv [_SYNC_MV_] ORDER BY k1;
+-- result:
+300
+100
+-- !result

test/sql/test_lake_sync_mv/T/test_lake_sync_mv

@@ -1,4 +1,4 @@
--- name: test_invalid_json @cloud
+-- name: test_lake_sync_mv @cloud
 CREATE TABLE tbl1 (
     k1 date,
     k2 int,
@@ -8,21 +8,16 @@ CREATE TABLE tbl1 (
     PARTITION p2 values [('2020-02-01'),('2020-03-01')))
     DISTRIBUTED BY HASH(k2) BUCKETS 3
     PROPERTIES('replication_num' = '1');
--- result:
--- !result
 
 INSERT INTO tbl1
 VALUES
     ("2020-01-12",4,100),
     ("2020-01-11",5,100),
     ("2020-01-11",4,100);
--- result:
--- !result
 
 CREATE MATERIALIZED VIEW sync_mv AS SELECT k1, SUM(v1) FROM tbl1 GROUP BY k1;
--- result:
--- !result
+function: wait_materialized_view_finish()
 
 INSERT INTO tbl1 VALUES("2020-01-11",6,100);
--- result:
--- !result
+
+SELECT mv_sum_v1 FROM sync_mv [_SYNC_MV_] ORDER BY k1;