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

Add per piece locks

Browse source
This commit is contained in:
Igor Katson 2024-05-03 10:34:39 +01:00
parent e53f5a7f23
commit ca9afa8535
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
crates/librqbit/src/torrent_state/live/mod.rs
View file

@ -161,6 +161,8 @@ pub struct TorrentStateLive {
pub(crate) files: FileStorage, pub(crate) files: FileStorage,
per_piece_locks: Vec<RwLock<()>>,
stats: AtomicStats, stats: AtomicStats,
lengths: Lengths, lengths: Lengths,
@ -212,6 +214,7 @@ impl TorrentStateLive {
peers: Default::default(), peers: Default::default(),
locked: RwLock::new(TorrentStateLocked { locked: RwLock::new(TorrentStateLocked {
chunks: Some(paused.chunk_tracker), chunks: Some(paused.chunk_tracker),
// TODO: move under per_piece_locks
inflight_pieces: Default::default(), inflight_pieces: Default::default(),
file_priorities, file_priorities,
fatal_errors_tx: Some(fatal_errors_tx), fatal_errors_tx: Some(fatal_errors_tx),
@ -229,6 +232,9 @@ impl TorrentStateLive {
up_speed_estimator, up_speed_estimator,
cancellation_token, cancellation_token,
streams: paused.streams, streams: paused.streams,
per_piece_locks: (0..lengths.total_pieces())
.map(|_| RwLock::new(()))
.collect(),
}); });
state.spawn( state.spawn(