fix: async fn resumed after initiate part failed (#4013)

Signed-off-by: Xuanwo <[email protected]>
apache · Jan 18, 2024 · 223e810 · 223e810
1 parent d0124ba
commit 223e810
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/src/raw/oio/write/multipart_write.rs b/core/src/raw/oio/write/multipart_write.rs
@@ -272,8 +272,9 @@ where
                 }
                 State::Init(fut) => {
                     let upload_id = ready!(fut.as_mut().poll(cx));
-                    self.upload_id = Some(Arc::new(upload_id?));
+                    // Make sure the future is dropped after it returned ready.
                     self.state = State::Idle;
+                    self.upload_id = Some(Arc::new(upload_id?));
                 }
                 State::Close(_) => {
                     unreachable!("MultipartWriter must not go into State::Close during poll_write")