girineko/hbb_common
1
0
Fork 0
mirror of https://github.com/GiriNeko/hbb_common.git synced 2025-12-16 21:37:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #143 from YinMo19/main

Browse source 
[enhance] add from method
This commit is contained in:
RustDesk 2025-04-27 17:58:50 +08:00 • committed by GitHub
commit 3afaf64944
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/stream.rs
View file

@ -1,6 +1,7 @@
use crate::{config, tcp, websocket, ResultType}; use crate::{config, tcp, websocket, ResultType};
use sodiumoxide::crypto::secretbox::Key; use sodiumoxide::crypto::secretbox::Key;
use std::net::SocketAddr; use std::net::SocketAddr;
use tokio::net::TcpStream;
// support Websocket and tcp. // support Websocket and tcp.
pub enum Stream { pub enum Stream {
@ -107,4 +108,9 @@ impl Stream {
Self::Tcp(tcp) => tcp.local_addr(), Self::Tcp(tcp) => tcp.local_addr(),
} }
} }
#[inline]
pub fn from(stream: TcpStream, stream_addr: SocketAddr) -> Self {
Self::Tcp(tcp::FramedStream::from(stream, stream_addr))
}
} }