From c01a0595550c1f50da880fefe876bb3db2effd11 Mon Sep 17 00:00:00 2001
From: ssrlive <30760636+ssrlive@users.noreply.github.com>
Date: Mon, 7 Nov 2022 04:11:30 +0800
Subject: [PATCH] Update client.rs

---
 src/client.rs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/client.rs b/src/client.rs
index e092889..8388fc4 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -120,14 +120,14 @@ pub fn connect_with_config<Req: IntoClientRequest>(
 /// you want to use other TLS libraries, use `client` instead.
 ///
 /// [readme]: https://github.com/snapview/tungstenite-rs/#features
-pub fn connect_to_raw_stream<Req: IntoClientRequest>(
+pub fn connect_to_raw_stream<Req: IntoClientRequest, Stream: Read + Write + NoDelay>(
     request: Req,
-    stream: TcpStream,
-) -> Result<(WebSocket<MaybeTlsStream<TcpStream>>, Response)> {
-    fn try_client_handshake(
+    stream: Stream,
+) -> Result<(WebSocket<MaybeTlsStream<Stream>>, Response)> {
+    fn try_client_handshake<Stream: Read + Write + NoDelay>(
         request: Request,
-        mut stream: TcpStream,
-    ) -> Result<(WebSocket<MaybeTlsStream<TcpStream>>, Response)> {
+        mut stream: Stream,
+    ) -> Result<(WebSocket<MaybeTlsStream<Stream>>, Response)> {
         NoDelay::set_nodelay(&mut stream, true)?;
 
         #[cfg(not(any(feature = "native-tls", feature = "__rustls-tls")))]