From 8131ba04826ea841711e1a0a5076e99a21f24097 Mon Sep 17 00:00:00 2001 From: Igor Katson Date: Thu, 5 Dec 2024 21:47:37 +0000 Subject: [PATCH] Store initial_peers in state --- crates/librqbit/src/session.rs | 3 ++- crates/librqbit/src/torrent_state/mod.rs | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/crates/librqbit/src/session.rs b/crates/librqbit/src/session.rs index a1104d5..5f6b2c6 100644 --- a/crates/librqbit/src/session.rs +++ b/crates/librqbit/src/session.rs @@ -1179,6 +1179,7 @@ impl Session { output_folder, disk_write_queue: self.disk_write_tx.clone(), ratelimits: opts.ratelimits, + initial_peers: opts.initial_peers.clone().unwrap_or_default(), #[cfg(feature = "disable-upload")] _disable_upload: self._disable_upload, }, @@ -1391,7 +1392,7 @@ impl Session { handle.shared().trackers.clone().into_iter().collect(), true, handle.shared().options.force_tracker_interval, - Default::default(), + handle.shared().options.initial_peers.clone(), )?; handle.start(peer_rx, false)?; self.try_update_persistence_metadata(handle).await; diff --git a/crates/librqbit/src/torrent_state/mod.rs b/crates/librqbit/src/torrent_state/mod.rs index 4c16106..91aec07 100644 --- a/crates/librqbit/src/torrent_state/mod.rs +++ b/crates/librqbit/src/torrent_state/mod.rs @@ -6,6 +6,7 @@ mod streaming; pub mod utils; use std::collections::HashSet; +use std::net::SocketAddr; use std::path::PathBuf; use std::sync::atomic::Ordering; use std::sync::Arc; @@ -105,6 +106,7 @@ pub(crate) struct ManagedTorrentOptions { pub output_folder: PathBuf, pub disk_write_queue: Option, pub ratelimits: LimitsConfig, + pub initial_peers: Vec, #[cfg(feature = "disable-upload")] pub _disable_upload: bool, }