leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add --disable-dht-persistence option to run multiple rqbit instances

Browse source
This commit is contained in:
Igor Katson 2021-10-08 23:54:51 +01:00
parent 18b99c91a1
commit e97377e239
1 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

20
crates/rqbit/src/main.rs
View file

@ -110,6 +110,12 @@ struct Opts {
#[clap(long = "disable-dht")] #[clap(long = "disable-dht")]
disable_dht: bool, disable_dht: bool,
/// Set this to disable DHT reading and storing it's state.
/// For now this is a useful workaround if you want to launch multiple rqbit instances,
/// otherwise DHT port will conflict.
#[clap(long = "disable-dht-persistence")]
disable_dht_persistence: bool,
/// The connect timeout, e.g. 1s, 1.5s, 100ms etc. /// The connect timeout, e.g. 1s, 1.5s, 100ms etc.
#[clap(long = "peer-connect-timeout")] #[clap(long = "peer-connect-timeout")]
peer_connect_timeout: Option<ParsedDuration>, peer_connect_timeout: Option<ParsedDuration>,
@ -119,7 +125,7 @@ fn compute_only_files<ByteBuf: AsRef<[u8]>>(
torrent: &TorrentMetaV1Info<ByteBuf>, torrent: &TorrentMetaV1Info<ByteBuf>,
filename_re: &str, filename_re: &str,
) -> anyhow::Result<Vec<usize>> { ) -> anyhow::Result<Vec<usize>> {
let filename_re = regex::Regex::new(&filename_re).context("filename regex is incorrect")?; let filename_re = regex::Regex::new(filename_re).context("filename regex is incorrect")?;
let mut only_files = Vec::new(); let mut only_files = Vec::new();
for (idx, (filename, _)) in torrent.iter_filenames_and_lengths()?.enumerate() { for (idx, (filename, _)) in torrent.iter_filenames_and_lengths()?.enumerate() {
let full_path = filename let full_path = filename
@ -191,11 +197,13 @@ async fn async_main(opts: Opts, spawner: BlockingSpawner) -> anyhow::Result<()>
let dht = if opts.disable_dht { let dht = if opts.disable_dht {
None None
} else { } else {
Some( let dht = if opts.disable_dht_persistence {
PersistentDht::create(None) Dht::new().await
.await } else {
.context("error initializing DHT")?, PersistentDht::create(None).await
) }
.context("error initializing DHT")?;
Some(dht)
}; };
let peer_opts = PeerConnectionOptions { let peer_opts = PeerConnectionOptions {