avantgardnerio commented on code in PR #4902:
URL: https://github.com/apache/arrow-datafusion/pull/4902#discussion_r1070047063
##########
datafusion/expr/src/logical_plan/plan.rs:
##########
@@ -1490,6 +1500,38 @@ pub struct CreateExternalTable {
pub options: HashMap<String, String>,
}
+#[derive(Clone)]
+pub enum WriteOp {
+ Insert,
+ Delete,
+ Update,
+ Ctas,
+}
+
+impl Display for WriteOp {
+ fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
+ match self {
+ WriteOp::Insert => write!(f, "Insert"),
+ WriteOp::Delete => write!(f, "Delete"),
+ WriteOp::Update => write!(f, "Update"),
+ WriteOp::Ctas => write!(f, "Ctas"),
+ }
+ }
+}
+
+/// The operator that modifies the content of a database (adapted from
substrait WriteRel)
+#[derive(Clone)]
+pub struct WriteRel {
+ /// The table name
+ pub table_name: OwnedTableReference,
+ /// The schema of the table (must align with Rel input)
+ pub table_schema: DFSchemaRef,
+ /// The type of operation to perform
+ pub op: WriteOp,
+ /// The relation that determines the tuples to add/remove/modify the
schema must match with table_schema
+ pub input: Arc<LogicalPlan>,
+}
Review Comment:
I only brought over the most core parts of the Substrait model, and adapted
them as needed to DataFusion equivalents.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
