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