andras/vault-link
andras/vault-link
Watch
1
0
Fork
You've already forked vault-link
 0
Code Issues Pull requests 18 Projects Releases 55 Packages Wiki Activity Actions 33

Timeout on websocket send

Browse source
This commit is contained in:
Andras Schmelczer 2026-09-22 08:09:19 +01:00
commit c3eb2dbf67
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
sync-server/src/app_state/websocket/utils.rs
View file

@ -9,6 +9,7 @@ use crate::{
errors::{SyncServerError, server_error, unauthenticated_error},
server::auth::auth,
};
use crate::consts::WEBSOCKET_SEND_TIMEOUT;
pub struct AuthenticatedWebSocketHandshake {
pub handshake: WebSocketHandshake,
@ -49,9 +50,13 @@ pub async fn send_update_over_websocket(
.context("Failed to serialize update")
.map_err(server_error)?;
sender
.send(Message::Text(serialized_update))
.await
.context("Failed to send message over websocket")
.map_err(server_error)
tokio::time::timeout(
WEBSOCKET_SEND_TIMEOUT,
sender.send(Message::Text(serialized_update)),
)
.await
.context("WebSocket send deadline expired")
.map_err(server_error)?
.context("Failed to send message over websocket")
.map_err(server_error)
}