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

async_backtrace

Browse source
This commit is contained in:
Igor Katson 2024-08-19 12:35:32 +01:00
parent 79e206d5a7
commit 7cda2c9807
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
5 changed files with 131 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -353,6 +353,7 @@ impl TorrentStateLive {
Ok(())
}
#[async_backtrace::framed]
async fn task_manage_incoming_peer(
self: Arc<Self>,
checked_peer: CheckedIncomingConnection,
@ -413,6 +414,7 @@ impl TorrentStateLive {
Ok(())
}
#[async_backtrace::framed]
async fn task_manage_outgoing_peer(
self: Arc<Self>,
addr: SocketAddr,
@ -449,12 +451,12 @@ impl TorrentStateLive {
state.meta.spawner,
state.meta.connector.clone(),
);
let requester = handler
let requester = async_backtrace::frame!(handler
.task_peer_chunk_requester()
.instrument(error_span!("chunk_requester"));
let conn_manager = peer_connection
.instrument(error_span!("chunk_requester")));
let conn_manager = async_backtrace::frame!(peer_connection
.manage_peer_outgoing(rx, state.have_broadcast_tx.subscribe())
.instrument(error_span!("peer_connection"));
.instrument(error_span!("peer_connection")));
handler
.counters
@ -1213,7 +1215,7 @@ impl PeerHandler {
}
loop {
self.wait_for_unchoke().await;
async_backtrace::frame!(self.wait_for_unchoke()).await;
if self.state.is_finished_and_dont_need_peers() {
debug!("nothing left to do, disconnecting peer");
@ -1232,7 +1234,7 @@ impl PeerHandler {
Some(next) => next,
None => {
debug!("no pieces to request");
tokio::time::sleep(Duration::from_secs(10)).await;
async_backtrace::frame!(tokio::time::sleep(Duration::from_secs(10))).await;
continue;
}
};
@ -1266,7 +1268,12 @@ impl PeerHandler {
};
loop {
match timeout(Duration::from_secs(10), self.requests_sem.acquire()).await {
match async_backtrace::frame!(timeout(
Duration::from_secs(10),
async_backtrace::frame!(self.requests_sem.acquire())
))
.await
{
Ok(acq) => break acq?.forget(),
Err(_) => continue,
};