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

Add a struct for session stats + bump the counters

Browse source
This commit is contained in:
Igor Katson 2024-08-21 11:22:37 +01:00
parent ee2ad7138e
commit 67afdb0aa5
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
8 changed files with 96 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

6
crates/librqbit/src/session.rs
View file

@ -16,6 +16,7 @@ use crate::{
peer_connection::PeerConnectionOptions,
read_buf::ReadBuf,
session_persistence::{json::JsonSessionPersistenceStore, SessionPersistenceStore},
session_stats::atomic::AtomicSessionStats,
spawn_utils::BlockingSpawner,
storage::{
filesystem::FilesystemStorageFactory, BoxStorageFactory, StorageFactoryExt, TorrentStorage,
@ -116,6 +117,8 @@ pub struct Session {
root_span: Option<Span>,
stats: Arc<AtomicSessionStats>,
// This is stored for all tasks to stop when session is dropped.
_cancellation_token_drop_guard: DropGuard,
}
@ -602,6 +605,7 @@ impl Session {
reqwest_client,
connector: stream_connector,
root_span: opts.root_span,
stats: Default::default(),
concurrent_initialize_semaphore: Arc::new(tokio::sync::Semaphore::new(opts.concurrent_init_limit.unwrap_or(3)))
});
@ -1147,6 +1151,7 @@ impl Session {
self.cancellation_token.child_token(),
self.concurrent_initialize_semaphore.clone(),
self.bitv_factory.clone(),
self.stats.clone(),
)
.context("error starting torrent")?;
}
@ -1303,6 +1308,7 @@ impl Session {
self.cancellation_token.child_token(),
self.concurrent_initialize_semaphore.clone(),
self.bitv_factory.clone(),
self.stats.clone(),
)?;
self.try_update_persistence_metadata(handle).await;
Ok(())