1 files changed, 4 insertions, 3 deletions
diff --git a/crates/utils/tcp_connection/src/handle.rs b/crates/utils/tcp_connection/src/handle.rs
index ab3a5ff..706d386 100644
--- a/crates/utils/tcp_connection/src/handle.rs
+++ b/crates/utils/tcp_connection/src/handle.rs
@@ -1,9 +1,10 @@
-use tokio::net::TcpStream;
+use crate::instance::ConnectionInstance;
+use std::future::Future;
 
 pub trait ClientHandle<RequestServer> {
-    fn process(stream: TcpStream);
+    fn process(instance: ConnectionInstance) -> impl Future<Output = ()> + Send + Sync;
 }
 
 pub trait ServerHandle<RequestClient> {
-    fn process(stream: TcpStream);
+    fn process(instance: ConnectionInstance) -> impl Future<Output = ()> + Send + Sync;
 }