fix: 修复 reader 中没有判断状态码的错误

1. 修复 reader 中没有判断状态码的错误
2. 修复 writer 中 RandFileWriterStd 错误类型不是 FileWriterError 的问题

Author: share121

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "fast-pull"
-version = "2.2.1"
+version = "2.2.2"
 edition = "2024"
 license = "MIT"
 authors = ["share121 <[email protected]>"]

Cargo.toml

@@ -93,7 +93,7 @@ pub struct RandFileWriterStd {
     buffer_size: usize,
 }
 impl RandFileWriterStd {
-    pub async fn new(file: File, size: u64, buffer_size: usize) -> Result<Self, io::Error> {
+    pub async fn new(file: File, size: u64, buffer_size: usize) -> Result<Self, FileWriterError> {
         file.set_len(size).await?;
         Ok(Self {
             buffer: BufWriter::with_capacity(buffer_size, file),
@@ -105,7 +105,7 @@ impl RandFileWriterStd {
     }
 }
 impl RandWriter for RandFileWriterStd {
-    type Error = io::Error;
+    type Error = FileWriterError;
     async fn write(&mut self, range: ProgressEntry, bytes: Bytes) -> Result<(), Self::Error> {
         let pos = self.cache.partition_point(|(i, _)| i < &range.start);
         self.cache_size += bytes.len();

src/file/writer.rs

@@ -83,6 +83,10 @@ impl Stream for ReqwestStream {
                 return self.poll_next(cx);
             }
             ResponseState::Ready(resp) => {
+                if let Err(e) = resp.error_for_status_ref() {
+                    self.resp = ResponseState::None;
+                    return Poll::Ready(Some(Err(e)));
+                }
                 let mut chunk = pin!(resp.chunk());
                 match chunk.try_poll_unpin(cx) {
                     Poll::Ready(Ok(Some(chunk))) => chunk_global = Ok(chunk),