implement PutPart methods for Box<dyn ObjectStore>

ArroyoSystems · Oct 21, 2023 · 2769380 · 2769380
1 parent 6062e6b
commit 2769380
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/object_store/src/lib.rs b/object_store/src/lib.rs
@@ -549,6 +549,24 @@ macro_rules! as_ref_impl {
                 self.as_ref().put_multipart(location).await
             }
 
+            #[cfg(feature = "cloud")]
+            async fn initiate_multipart_upload(
+                &self,
+                location: &Path,
+            ) -> Result<(MultipartId, Box<dyn PutPart>)> {
+                self.as_ref().initiate_multipart_upload(location).await
+            }
+
+            /// Get a multi-part upload that allows writing data in chunks
+            #[cfg(feature = "cloud")]
+            async fn get_put_part(
+                &self,
+                location: &Path,
+                multipart_id: &MultipartId,
+            ) -> Result<Box<dyn PutPart>> {
+                self.as_ref().get_put_part(location, multipart_id).await
+            }
+
             async fn abort_multipart(
                 &self,
                 location: &Path,