mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-13 06:20:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

ping every 30 seconds

Browse source
This commit is contained in:
r58Playz 2024-09-23 17:33:16 -07:00
parent fdd641c67f
commit 09b15e3c43
4 changed files with 59 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
wisp/src/inner.rs
View file

@ -23,6 +23,8 @@ pub(crate) enum WsEvent {
u16,
oneshot::Sender<Result<MuxStream, WispError>>,
),
SendPing(Payload<'static>, oneshot::Sender<Result<(), WispError>>),
SendPong(Payload<'static>),
WispMessage(Option<Packet<'static>>, Option<Frame<'static>>),
EndFut(Option<CloseReason>),
}
@ -234,6 +236,8 @@ impl<R: WebSocketRead + Send> MuxInner<R> {
let (mut frame, optional_frame) = msg?;
if frame.opcode == OpCode::Close {
return Ok(None);
} else if frame.opcode == OpCode::Ping {
return Ok(Some(WsEvent::SendPong(frame.payload)));
}
if let Some(ref extra_frame) = optional_frame {
@ -308,6 +312,12 @@ impl<R: WebSocketRead + Send> MuxInner<R> {
let _ = channel.send(Err(WispError::InvalidStreamId));
}
}
WsEvent::SendPing(payload, channel) => {
let _ = channel.send(self.tx.write_frame(Frame::new(OpCode::Ping, payload, true)).await);
}
WsEvent::SendPong(payload) => {
self.tx.write_frame(Frame::new(OpCode::Pong, payload, true)).await?;
}
WsEvent::EndFut(x) => {
if let Some(reason) = x {
let _ = self