refactor(bindings/python): simplify async writer aexit (#4128)

Signed-off-by: suyanhanx <[email protected]>
apache · Feb 1, 2024 · f2cc712 · f2cc712
1 parent f4a3027
commit f2cc712
Showing 1 changed file with 2 additions and 12 deletions.
diff --git a/bindings/python/src/file.rs b/bindings/python/src/file.rs
@@ -386,22 +386,12 @@ impl AsyncFile {
     }
 
     fn __aexit__<'a>(
-        &self,
+        &'a mut self,
         py: Python<'a>,
         _exc_type: &'a PyAny,
         _exc_value: &'a PyAny,
         _traceback: &'a PyAny,
     ) -> PyResult<&'a PyAny> {
-        let state = self.0.clone();
-        future_into_py(py, async move {
-            let mut state = state.lock().await;
-            if let AsyncFileState::Writer(w) = &mut *state {
-                w.close()
-                    .await
-                    .map_err(|err| PyIOError::new_err(err.to_string()))?;
-            }
-            *state = AsyncFileState::Closed;
-            Ok(())
-        })
+        self.close(py)
     }
 }