mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-08-31 00:09:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf(recv): add try_recv fast path in recv_packet_from_chan (#2426)

Browse Source 
Add a try_recv() fast path to recv_packet_from_chan(): if a packet is
immediately available, return it without parking the task. Only when
the channel is empty do we fall back to recv().await.

This benefits all callers of recv_packet_from_chan() including
start_peer_recv, virtual_nic, foreign_network_manager, and instance.
This commit is contained in:
fanyang
2026-08-19 00:16:38 +08:00
committed by GitHub
parent f64af13424
commit 57eb6908f4
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier-core/src/peers/mod.rs
+9 -5
View File
@@ -21,7 +21,7 @@ pub(crate) mod test_support;
mod tests;
use crate::packet::ZCPacket;
use tokio::sync::mpsc::error::{SendError, TrySendError};
use tokio::sync::mpsc::error::{SendError, TryRecvError, TrySendError};
pub type PacketRecvChan = tokio::sync::mpsc::Sender<ZCPacket>;
pub type PacketRecvChanReceiver = tokio::sync::mpsc::Receiver<ZCPacket>;
@@ -44,10 +44,14 @@ pub(crate) async fn send_packet_to_chan(
pub async fn recv_packet_from_chan(
packet_recv_chan_receiver: &mut PacketRecvChanReceiver,
) -> Result<ZCPacket, anyhow::Error> {
packet_recv_chan_receiver
.recv()
.await
.ok_or(anyhow::anyhow!("recv_packet_from_chan failed"))
match packet_recv_chan_receiver.try_recv() {
Ok(packet) => Ok(packet),
Err(TryRecvError::Empty) => packet_recv_chan_receiver
.recv()
.await
.ok_or(anyhow::anyhow!("recv_packet_from_chan failed")),
Err(TryRecvError::Disconnected) => Err(anyhow::anyhow!("recv_packet_from_chan failed")),
}
}
#[async_trait::async_trait]