write_exclusive (#1)

splitgraph · Dec 4, 2024 · e7008f3 · e7008f3
1 parent 4fba3f4
commit e7008f3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/crates/iceberg/src/io/file_io.rs b/crates/iceberg/src/io/file_io.rs
@@ -344,6 +344,16 @@ impl OutputFile {
         writer.close().await
     }
 
+    /// Create a new output file with given bytes.
+    /// Error out if the file already exists
+    pub async fn write_exclusive(&self, bs: Bytes) -> crate::Result<()> {
+        self.op
+            .write_with(&self.path[self.relative_path_pos..], bs)
+            .if_none_match("*")
+            .await?;
+        Ok(())
+    }
+
     /// Creates output file for continues writing.
     ///
     /// # Notes