feat: add compress support

crates/rio/src/hardlimit_reader.rs

@@ -1,12 +1,11 @@
+use crate::compress_index::{Index, TryGetIndex};
+use crate::{EtagResolvable, HashReaderDetector, HashReaderMut, Reader};
+use pin_project_lite::pin_project;
 use std::io::{Error, Result};
 use std::pin::Pin;
 use std::task::{Context, Poll};
 use tokio::io::{AsyncRead, ReadBuf};
 
-use crate::{EtagResolvable, HashReaderDetector, HashReaderMut, Reader};
-
-use pin_project_lite::pin_project;
-
 pin_project! {
     pub struct HardLimitReader {
         #[pin]
@@ -60,10 +59,18 @@ impl HashReaderDetector for HardLimitReader {
     }
 }
 
+impl TryGetIndex for HardLimitReader {
+    fn try_get_index(&self) -> Option<&Index> {
+        self.inner.try_get_index()
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use std::vec;
 
+    use crate::WarpReader;
+
     use super::*;
     use rustfs_utils::read_full;
     use tokio::io::{AsyncReadExt, BufReader};
@@ -72,7 +79,7 @@ mod tests {
     async fn test_hardlimit_reader_normal() {
         let data = b"hello world";
         let reader = BufReader::new(&data[..]);
-        let reader = Box::new(reader);
+        let reader = Box::new(WarpReader::new(reader));
         let hardlimit = HardLimitReader::new(reader, 20);
         let mut r = hardlimit;
         let mut buf = Vec::new();
@@ -85,7 +92,7 @@ mod tests {
     async fn test_hardlimit_reader_exact_limit() {
         let data = b"1234567890";
         let reader = BufReader::new(&data[..]);
-        let reader = Box::new(reader);
+        let reader = Box::new(WarpReader::new(reader));
         let hardlimit = HardLimitReader::new(reader, 10);
         let mut r = hardlimit;
         let mut buf = Vec::new();
@@ -98,7 +105,7 @@ mod tests {
     async fn test_hardlimit_reader_exceed_limit() {
         let data = b"abcdef";
         let reader = BufReader::new(&data[..]);
-        let reader = Box::new(reader);
+        let reader = Box::new(WarpReader::new(reader));
         let hardlimit = HardLimitReader::new(reader, 3);
         let mut r = hardlimit;
         let mut buf = vec![0u8; 10];
@@ -123,7 +130,7 @@ mod tests {
     async fn test_hardlimit_reader_empty() {
         let data = b"";
         let reader = BufReader::new(&data[..]);
-        let reader = Box::new(reader);
+        let reader = Box::new(WarpReader::new(reader));
         let hardlimit = HardLimitReader::new(reader, 5);
         let mut r = hardlimit;
         let mut buf = Vec::new();