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

Timing on_received_message

Browse source
This commit is contained in:
Igor Katson 2023-11-19 20:21:26 +00:00
parent 124c6057be
commit 17d40824b4
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 29 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

56
crates/librqbit/src/torrent_state.rs
View file

@ -857,33 +857,35 @@ struct PeerHandler {
impl PeerConnectionHandler for PeerHandler { impl PeerConnectionHandler for PeerHandler {
fn on_received_message(&self, message: Message<ByteBuf<'_>>) -> anyhow::Result<()> { fn on_received_message(&self, message: Message<ByteBuf<'_>>) -> anyhow::Result<()> {
match message { timeit("on_received_message", || {
Message::Request(request) => { match message {
self.on_download_request(self.addr, request) Message::Request(request) => {
.context("on_download_request")?; self.on_download_request(self.addr, request)
} .context("on_download_request")?;
Message::Bitfield(b) => self }
.on_bitfield(self.addr, b.clone_to_owned()) Message::Bitfield(b) => self
.context("on_bitfield")?, .on_bitfield(self.addr, b.clone_to_owned())
Message::Choke => self.on_i_am_choked(self.addr), .context("on_bitfield")?,
Message::Unchoke => self.on_i_am_unchoked(self.addr), Message::Choke => self.on_i_am_choked(self.addr),
Message::Interested => self.on_peer_interested(self.addr), Message::Unchoke => self.on_i_am_unchoked(self.addr),
Message::Piece(piece) => { Message::Interested => self.on_peer_interested(self.addr),
self.on_received_piece(self.addr, piece) Message::Piece(piece) => {
.context("on_received_piece")?; self.on_received_piece(self.addr, piece)
} .context("on_received_piece")?;
Message::KeepAlive => { }
debug!("keepalive received"); Message::KeepAlive => {
} debug!("keepalive received");
Message::Have(h) => self.on_have(self.addr, h), }
Message::NotInterested => { Message::Have(h) => self.on_have(self.addr, h),
info!("received \"not interested\", but we don't care yet") Message::NotInterested => {
} info!("received \"not interested\", but we don't care yet")
message => { }
warn!("received unsupported message {:?}, ignoring", message); message => {
} warn!("received unsupported message {:?}, ignoring", message);
} }
Ok(()) };
Ok(())
})
} }
fn get_have_bytes(&self) -> u64 { fn get_have_bytes(&self) -> u64 {