feat: support truncate table statement

4paradigm · Oct 8, 2023 · e5a777e · e5a777e
1 parent 9190ecf
commit e5a777e
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 0 deletions.
diff --git a/cases/plan/cmd.yaml b/cases/plan/cmd.yaml
@@ -649,6 +649,22 @@ cases:
           +-cmd_type: drop function
           +-if_exists: true
           +-args: [func1]
+  - id: truncate_stmt
+    desc: truncate
+    sql: TRUNCATE TABLE t1;
+    expect:
+      node_tree_str: |
+        +-node[CMD]
+          +-cmd_type: truncate table
+          +-args: [t1]
+  - id: truncate_stmt_db
+    desc: truncate
+    sql: TRUNCATE TABLE db1.t1;
+    expect:
+      node_tree_str: |
+        +-node[CMD]
+          +-cmd_type: truncate table
+          +-args: [db1, t1]
   - id: exit_stmt
     desc: exit statement
     sql: EXIT;

diff --git a/hybridse/include/node/node_enum.h b/hybridse/include/node/node_enum.h
@@ -283,6 +283,7 @@ enum CmdType {
     kCmdShowFunctions,
     kCmdDropFunction,
     kCmdShowJobLog,
+    kCmdTruncate,
     kCmdFake,  // not a real cmd, for testing purpose only
     kLastCmd = kCmdFake,
 };

diff --git a/hybridse/src/node/sql_node.cc b/hybridse/src/node/sql_node.cc
@@ -75,6 +75,7 @@ static absl::flat_hash_map<CmdType, absl::string_view> CreateCmdTypeNamesMap() {
         {CmdType::kCmdDropFunction, "drop function"},
         {CmdType::kCmdShowFunctions, "show functions"},
         {CmdType::kCmdShowJobLog, "show joblog"},
+        {CmdType::kCmdTruncate, "truncate table"},
     };
     for (auto kind = 0; kind < CmdType::kLastCmd; ++kind) {
         DCHECK(map.find(static_cast<CmdType>(kind)) != map.end());

diff --git a/hybridse/src/planv2/ast_node_converter.cc b/hybridse/src/planv2/ast_node_converter.cc
@@ -611,6 +611,16 @@ base::Status ConvertStatement(const zetasql::ASTStatement* statement, node::Node
             *output = node;
             break;
         }
+        case zetasql::AST_TRUNCATE_STATEMENT: {
+            const zetasql::ASTTruncateStatement* truncate_statement =
+                statement->GetAsOrNull<zetasql::ASTTruncateStatement>();
+            std::vector<std::string> names;
+            CHECK_STATUS(AstPathExpressionToStringList(truncate_statement->target_path(), names));
+            auto node =
+                dynamic_cast<node::CmdNode*>(node_manager->MakeCmdNode(node::CmdType::kCmdTruncate, names));
+            *output = node;
+            break;
+        }
         case zetasql::AST_DROP_FUNCTION_STATEMENT: {
             const zetasql::ASTDropFunctionStatement* drop_fun_statement =
                 statement->GetAsOrNull<zetasql::ASTDropFunctionStatement>();