Extend Writer to run arbitrary actions

rust/src/writer.rs

@@ -4,7 +4,7 @@
 //! Unlike the transaction API on DeltaTable, this higher level writer will also write out the
 //! parquet files
 
-use crate::action::Txn;
+use crate::action::Action;
 use crate::DeltaTableError;
 use arrow::record_batch::RecordBatch;
 use log::*;
@@ -24,7 +24,7 @@ pub struct BufferedJsonWriter {
     buffer: HashMap<WriterPartition, Vec<Value>>,
     schema: arrow::datatypes::SchemaRef,
     partitions: Vec<String>,
-    txns: Vec<Txn>,
+    actions: Vec<Action>,
 }
 
 impl BufferedJsonWriter {
@@ -42,7 +42,7 @@ impl BufferedJsonWriter {
             schema,
             buffer: HashMap::new(),
             partitions: metadata.partition_columns,
-            txns: vec![],
+            actions: vec![],
         })
     }
 
@@ -52,8 +52,8 @@ impl BufferedJsonWriter {
     }
 
     /// Add a txn action to the buffer
-    pub fn record_txn(&mut self, txn: Txn) {
-        self.txns.push(txn);
+    pub fn record_action(&mut self, action: Action) {
+        self.actions.push(action);
     }
 
     /// Write a new Value into the buffer
@@ -127,12 +127,7 @@ impl BufferedJsonWriter {
             }
         }
 
-        dtx.add_actions(
-            self.txns
-                .drain(0..)
-                .map(crate::action::Action::txn)
-                .collect(),
-        );
+        dtx.add_actions(self.actions.drain(0..).collect());
 
         dtx.commit(None).await?;
         self.buffer.clear();