[incubator-opendal] 13/31: Fix sftp

[xu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

xuanwo pushed a commit to branch poll-write
in repository https://gitbox.apache.org/repos/asf/incubator-opendal.git

commit 8ad74fb8c43941db0fb454b7937407718e7da58d
Author: Xuanwo <gi...@xuanwo.io>
AuthorDate: Fri Sep 8 16:13:47 2023 +0800

    Fix sftp
    
    Signed-off-by: Xuanwo <gi...@xuanwo.io>
---
 core/src/services/sftp/writer.rs | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/core/src/services/sftp/writer.rs b/core/src/services/sftp/writer.rs
index cfe51c7f3..fb1ab0a3e 100644
--- a/core/src/services/sftp/writer.rs
+++ b/core/src/services/sftp/writer.rs
@@ -16,7 +16,10 @@
 // under the License.
 
 use async_trait::async_trait;
+use futures::AsyncWrite;
 use openssh_sftp_client::file::File;
+use std::pin::Pin;
+use std::task::{Context, Poll};
 
 use crate::raw::oio;
 use crate::*;
@@ -34,19 +37,16 @@ impl SftpWriter {
 #[async_trait]
 impl oio::Write for SftpWriter {
     fn poll_write(&mut self, cx: &mut Context<'_>, bs: &dyn oio::WriteBuf) -> Poll<Result<usize>> {
-        let size = self.file.write(bs.chunk()).await?;
-
-        Ok(size)
+        Pin::new(&mut self.file)
+            .poll_write(cx, bs.chunk())
+            .map_err(Error::from)
     }
 
     fn poll_abort(&mut self, cx: &mut Context<'_>) -> Poll<Result<()>> {
-        Err(Error::new(
-            ErrorKind::Unsupported,
-            "SFTP does not support aborting writes",
-        ))
+        Poll::Ready(Ok(()))
     }
 
     fn poll_close(&mut self, cx: &mut Context<'_>) -> Poll<Result<()>> {
-        Ok(())
+        Pin::new(&mut self.file).poll_flush(cx).map_err(Error::from)
     }
 }