http: allocate dynamically the space for HTTP response

hch12907 · Nov 19, 2023 · ea41e82 · ea41e82
1 parent c62d5f9
commit ea41e82
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/http/curl_backend.rs b/src/http/curl_backend.rs
@@ -106,7 +106,7 @@ impl Request {
         // UNWRAP-SAFETY: HTTP is supported.
         self.curl.http_headers(self.header_list).unwrap();
 
-        let mut response = Vec::with_capacity(8192);
+        let mut response = Vec::with_capacity(1024);
         let mut transfer = self.curl.transfer();
 
         transfer

diff --git a/src/http/mod.rs b/src/http/mod.rs
@@ -34,9 +34,9 @@ impl Response {
             .map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))
     }
 
-    pub fn into_string(mut self) -> Result<String, io::Error> {
-        let mut vec = vec![0; 1024 * 1024];
-        let read = self.reader.read(&mut vec)?;
+    pub fn into_string(self) -> Result<String, io::Error> {
+        let mut vec = vec![0; 1024];
+        let read = self.reader.take(2 * 1024 * 1024).read(&mut vec)?;
         vec.resize(read, 0);
         String::from_utf8(vec).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))
     }