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

more tracing spans

Browse source
This commit is contained in:
Igor Katson 2024-08-19 12:09:40 +01:00
parent 6e92eec167
commit 79e206d5a7
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -78,7 +78,7 @@ use tokio::{
time::timeout, time::timeout,
}; };
use tokio_util::sync::CancellationToken; use tokio_util::sync::CancellationToken;
use tracing::{debug, error, error_span, info, trace, warn}; use tracing::{debug, error, error_span, info, trace, warn, Instrument};
use crate::{ use crate::{
chunk_tracker::{ChunkMarkingResult, ChunkTracker, HaveNeededSelected}, chunk_tracker::{ChunkMarkingResult, ChunkTracker, HaveNeededSelected},
@ -449,7 +449,12 @@ impl TorrentStateLive {
state.meta.spawner, state.meta.spawner,
state.meta.connector.clone(), state.meta.connector.clone(),
); );
let requester = handler.task_peer_chunk_requester(); let requester = handler
.task_peer_chunk_requester()
.instrument(error_span!("chunk_requester"));
let conn_manager = peer_connection
.manage_peer_outgoing(rx, state.have_broadcast_tx.subscribe())
.instrument(error_span!("peer_connection"));
handler handler
.counters .counters
@ -457,7 +462,7 @@ impl TorrentStateLive {
.fetch_add(1, Ordering::Relaxed); .fetch_add(1, Ordering::Relaxed);
let res = tokio::select! { let res = tokio::select! {
r = requester => {r} r = requester => {r}
r = peer_connection.manage_peer_outgoing(rx, state.have_broadcast_tx.subscribe()) => {r} r = conn_manager => {r}
}; };
match res { match res {
@ -926,7 +931,7 @@ impl PeerHandler {
self.counters.errors.fetch_add(1, Ordering::Relaxed); self.counters.errors.fetch_add(1, Ordering::Relaxed);
if self.state.is_finished_and_dont_need_peers() { if self.state.is_finished_and_dont_need_peers() {
trace!("torrent finished, not re-queueing"); debug!("torrent finished, not re-queueing");
pe.value_mut().state.set(PeerState::NotNeeded, pstats); pe.value_mut().state.set(PeerState::NotNeeded, pstats);
return Ok(()); return Ok(());
} }