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

Flush messages before disconnecting (lame)

Browse source
This commit is contained in:
Igor Katson 2023-12-05 22:22:04 +00:00
parent cc92afcdec
commit b224ed2397
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1159,7 +1159,8 @@ impl PeerHandler {
let handle = self.addr; let handle = self.addr;
self.wait_for_bitfield().await; self.wait_for_bitfield().await;
// TODO: this check needs to happen more often // TODO: this check needs to happen more often, we need to update our
// interested state with the other side, for now we send it only once.
if self.state.is_finished() { if self.state.is_finished() {
self.tx self.tx
.send(WriterRequest::Message(MessageOwned::NotInterested))?; .send(WriterRequest::Message(MessageOwned::NotInterested))?;
@ -1174,6 +1175,8 @@ impl PeerHandler {
{ {
debug!("both peer and us have full torrent, disconnecting"); debug!("both peer and us have full torrent, disconnecting");
self.tx.send(WriterRequest::Disconnect)?; self.tx.send(WriterRequest::Disconnect)?;
// Sleep a bit to ensure this gets written to the network by manage_peer
tokio::time::sleep(Duration::from_millis(100)).await;
return Ok(()); return Ok(());
} }
} else { } else {