rustdesk/hbb_common
1
0
Fork 0
mirror of https://github.com/rustdesk/hbb_common.git synced 2026-04-10 01:46:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

[enhance] add websocket NOT a config feature.

Browse Source
This commit is contained in:
YinMo19
2025-04-18 02:53:39 +08:00
parent d00dd60e4c
commit 58103659e7
2 changed files with 116 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/socket_client.rs
View File

@@ -2,7 +2,7 @@ use crate::{
config::{Config, NetworkType},
tcp::FramedStream,
udp::FramedSocket,
websocket, ResultType,
websocket, ResultType, Stream,
};
use anyhow::Context;
use std::net::SocketAddr;
@@ -113,11 +113,11 @@ pub async fn connect_tcp_local<
target: T,
local: Option<SocketAddr>,
ms_timeout: u64,
) -> ResultType<crate::Stream> {
) -> ResultType<Stream> {
#[cfg(feature = "websocket")]
{
let url = format!("ws://{}", target);
websocket::WsFramedStream::new(url, local, None, ms_timeout).await
Ok(Stream::WebSocket(websocket::WsFramedStream::new(url, local, None, ms_timeout).await?))
}
#[cfg(not(feature = "websocket"))]
{