actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-23 23:51:06 +01:00
Code Issues Releases Wiki Activity

Reduce copies in ping/pong

Browse Source
This commit is contained in:
asonix 2024-11-04 18:02:41 -06:00
parent b3c5934b00
commit a66f20a106
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
actix-ws/examples/chat.rs
View File

@ -82,7 +82,7 @@ async fn ws(
loop {
interval.tick().await;
if session2.ping(b"").await.is_err() {
if session2.ping(&b""[..]).await.is_err() {
break;
}
@ -97,7 +97,7 @@ async fn ws(
while let Some(Ok(msg)) = stream.recv().await {
match msg {
AggregatedMessage::Ping(bytes) => {
if session.pong(&bytes).await.is_err() {
if session.pong(bytes).await.is_err() {
return;
}
}

8
actix-ws/src/session.rs
View File

@ -103,11 +103,11 @@ impl Session {
/// }
/// # }
/// ```
pub async fn ping(&mut self, msg: &[u8]) -> Result<(), Closed> {
pub async fn ping(&mut self, msg: impl Into<Bytes>) -> Result<(), Closed> {
self.pre_check();
if let Some(inner) = self.inner.as_mut() {
inner
.send(Message::Ping(Bytes::copy_from_slice(msg)))
.send(Message::Ping(msg.into()))
.await
.map_err(|_| Closed)
} else {
@ -127,11 +127,11 @@ impl Session {
/// _ => (),
/// }
/// # }
pub async fn pong(&mut self, msg: &[u8]) -> Result<(), Closed> {
pub async fn pong(&mut self, msg: impl Into<Bytes>) -> Result<(), Closed> {
self.pre_check();
if let Some(inner) = self.inner.as_mut() {
inner
.send(Message::Pong(Bytes::copy_from_slice(msg)))
.send(Message::Pong(msg.into()))
.await
.map_err(|_| Closed)
} else {