From b66e12ccd8823ffeaff23e2b4e5426cb15f59c99 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Mon, 15 Jul 2024 17:52:33 +0200 Subject: set websocket nonblocking --- light-client/src/network.rs | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'light-client/src') diff --git a/light-client/src/network.rs b/light-client/src/network.rs index f2c42518..3eb3f50a 100644 --- a/light-client/src/network.rs +++ b/light-client/src/network.rs @@ -59,9 +59,14 @@ impl Network { }); let stream = TcpStream::connect((host, port))?; stream.set_nodelay(true).unwrap(); - // stream.set_nonblocking(true).unwrap(); - let (sock, _) = client_tls_with_config(request, stream, None, None).unwrap(); + let (mut sock, _) = client_tls_with_config(request, stream, None, None).unwrap(); + + match sock.get_mut() { + MaybeTlsStream::Plain(s) => s.set_nonblocking(true).unwrap(), + MaybeTlsStream::Rustls(s) => s.sock.set_nonblocking(true).unwrap(), + _ => todo!(), + }; Ok(Self { sock, -- cgit v1.2.3-70-g09d2