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

Disk write queue

Browse source
This commit is contained in:
Igor Katson 2024-05-03 12:47:57 +01:00
parent e658b960e3
commit 1744f0c101
5 changed files with 43 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
crates/librqbit/src/peer_connection.rs
View file

@ -19,6 +19,7 @@ use tracing::trace;
use crate::{read_buf::ReadBuf, spawn_utils::BlockingSpawner};
#[async_trait::async_trait]
pub trait PeerConnectionHandler {
fn on_connected(&self, _connection_time: Duration) {}
fn get_have_bytes(&self) -> u64;
@ -28,7 +29,7 @@ pub trait PeerConnectionHandler {
&self,
extended_handshake: &ExtendedHandshake<ByteBuf>,
) -> anyhow::Result<()>;
fn on_received_message(&self, msg: Message<ByteBuf<'_>>) -> anyhow::Result<()>;
async fn on_received_message(&self, msg: Message<ByteBuf<'_>>) -> anyhow::Result<()>;
fn on_uploaded_bytes(&self, bytes: u32);
fn read_chunk(&self, chunk: &ChunkInfo, buf: &mut [u8]) -> anyhow::Result<()>;
}
@ -360,6 +361,7 @@ impl<H: PeerConnectionHandler> PeerConnection<H> {
} else {
self.handler
.on_received_message(message)
.await
.context("error in handler.on_received_message()")?;
}
}