try DML patch

milenkovicm · Jan 11, 2025 · deb9519 · deb9519
1 parent ae54d70
commit deb9519
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -26,10 +26,10 @@ arrow-flight = { version = "53", features = ["flight-sql-experimental"] }
 clap = { version = "4.5", features = ["derive", "cargo"] }
 configure_me = { version = "0.4.0" }
 configure_me_codegen = { version = "0.4.4" }
-datafusion = "44.0.0"
-datafusion-cli = "44.0.0"
-datafusion-proto = "44.0.0"
-datafusion-proto-common = "44.0.0"
+datafusion = { path = "../arrow-datafusion-fork/datafusion/core" }
+datafusion-cli = "44"
+datafusion-proto = { path = "../arrow-datafusion-fork/datafusion/proto" }
+datafusion-proto-common = { path = "../arrow-datafusion-fork/datafusion/proto-common" }
 object_store = "0.11"
 prost = "0.13"
 prost-types = "0.13"

diff --git a/ballista/core/proto/datafusion.proto b/ballista/core/proto/datafusion.proto
@@ -61,6 +61,7 @@ message LogicalPlanNode {
     UnnestNode unnest = 30;
     RecursiveQueryNode recursive_query = 31;
     CteWorkTableScanNode cte_work_table_scan = 32;
+    DmlNode dml = 33;
   }
 }
 
@@ -264,6 +265,21 @@ message CopyToNode {
   repeated string partition_by = 7;
 }
 
+message DmlNode{
+   enum Type {
+    UPDATE = 0;
+    DELETE = 1;
+    INSERT_APPEND = 2;
+    INSERT_OVERWRITE = 3;
+    INSERT_REPLACE = 4;
+    CTAS = 5;
+  }
+  Type dml_type = 1;
+  LogicalPlanNode input = 2;
+  TableReference table_name = 3;
+  datafusion_common.DfSchema schema = 4;
+}
+
 message UnnestNode {
   LogicalPlanNode input = 1;
   repeated datafusion_common.Column exec_columns = 2;
@@ -1255,4 +1271,4 @@ message RecursiveQueryNode {
 message CteWorkTableScanNode {
     string name = 1;
     datafusion_common.Schema schema = 2;
-}
+}