[Lake] Support export table (#9307)

StarRocks · Aug 2, 2022 · c6d468f · c6d468f
1 parent a272ac8
commit c6d468f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/analysis/ExportStmt.java b/fe/fe-core/src/main/java/com/starrocks/analysis/ExportStmt.java
@@ -224,6 +224,7 @@ private void checkTable(GlobalStateMgr globalStateMgr) throws AnalysisException
             switch (tblType) {
                 case MYSQL:
                 case OLAP:
+                case LAKE:
                     break;
                 case BROKER:
                 case SCHEMA:

diff --git a/fe/fe-core/src/main/java/com/starrocks/load/ExportJob.java b/fe/fe-core/src/main/java/com/starrocks/load/ExportJob.java
@@ -208,6 +208,11 @@ public void setJob(ExportStmt stmt) throws UserException {
         }
 
         this.sql = stmt.toSql();
+
+        // TODO Support snapshot of lake table
+        if (exportTable.isLakeTable()) {
+            this.state = JobState.EXPORTING;
+        }
     }
 
     private void genExecFragment(ExportStmt stmt) throws UserException {
@@ -306,6 +311,7 @@ private ScanNode genScanNode() throws UserException {
         ScanNode scanNode = null;
         switch (exportTable.getType()) {
             case OLAP:
+            case LAKE:
                 scanNode = new OlapScanNode(new PlanNodeId(0), exportTupleDesc, "OlapScanNodeForExport");
                 scanNode.setColumnFilters(Maps.newHashMap());
                 ((OlapScanNode) scanNode).setIsPreAggregation(false, "This an export operation");
@@ -336,6 +342,7 @@ private PlanFragment genPlanFragment(Table.TableType type, ScanNode scanNode, in
         PlanFragment fragment = null;
         switch (exportTable.getType()) {
             case OLAP:
+            case LAKE:
                 fragment = new PlanFragment(
                         new PlanFragmentId(nextId.getAndIncrement()), scanNode, DataPartition.RANDOM);
                 break;