wip: add max memory size

ikatson · Oct 5, 2024 · 0734368 · 0734368
1 parent a8bdc5f
commit 0734368
Show file tree

Hide file tree

Showing 17 changed files with 281 additions and 107 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/dht/src/dht.rs b/crates/dht/src/dht.rs
@@ -402,9 +402,8 @@ impl<C: RecursiveRequestCallbacks> RecursiveRequest<C> {
             self.callbacks.on_request_start(self, id, addr);
         }
 
-        let response = self.dht.request(self.request.clone(), addr).await.map(|r| {
+        let response = self.dht.request(self.request.clone(), addr).await.inspect(|r| {
             self.mark_node_responded(addr, &r);
-            r
         });
         if let Some(id) = id {
             self.callbacks.on_request_end(self, id, addr, &response);

diff --git a/crates/librqbit/src/api.rs b/crates/librqbit/src/api.rs
@@ -19,6 +19,7 @@ use crate::{
         peer::stats::snapshot::{PeerStatsFilter, PeerStatsSnapshot},
         FileStream, ManagedTorrentHandle,
     },
+    StreamOptions,
 };
 
 #[cfg(feature = "tracing-subscriber-utils")]
@@ -423,7 +424,7 @@ impl Api {
 
     pub fn api_stream(&self, idx: TorrentIdOrHash, file_id: usize) -> Result<FileStream> {
         let mgr = self.mgr_handle(idx)?;
-        Ok(mgr.stream(file_id)?)
+        Ok(mgr.stream(file_id, StreamOptions::default())?)
     }
 }
 

diff --git a/crates/librqbit/src/lib.rs b/crates/librqbit/src/lib.rs
@@ -61,6 +61,7 @@ pub mod session_stats;
 mod spawn_utils;
 pub mod storage;
 mod stream_connect;
+mod stream_ops;
 mod torrent_state;
 #[cfg(feature = "tracing-subscriber-utils")]
 pub mod tracing_subscriber_config_utils;
@@ -80,6 +81,7 @@ pub use session::{
     SessionPersistenceConfig, SUPPORTED_SCHEMES,
 };
 pub use spawn_utils::spawn as librqbit_spawn;
+pub use stream_ops::StreamOptions;
 pub use torrent_state::{
     ManagedTorrent, ManagedTorrentShared, ManagedTorrentState, TorrentStats, TorrentStatsState,
 };