Fixed codes by Zed

2 files changed, 17 insertions, 5 deletions

diff --git a/crates/utils/tcp_connection/tcp_connection_test/src/example_handle.rs b/crates/utils/tcp_connection/tcp_connection_test/src/example_handle.rs
index cefeef0..2c84bd1 100644
--- a/crates/utils/tcp_connection/tcp_connection_test/src/example_handle.rs
+++ b/crates/utils/tcp_connection/tcp_connection_test/src/example_handle.rs
@@ -1,14 +1,26 @@
-use tcp_connection::handle::{ClientHandle, ServerHandle};
-use tokio::net::TcpStream;
+use tcp_connection::{
+    handle::{ClientHandle, ServerHandle},
+    instance::ConnectionInstance,
+};
 
 pub(crate) struct ExampleClientHandle;
 
 impl ClientHandle<ExampleServerHandle> for ExampleClientHandle {
-    fn process(stream: TcpStream) {}
+    fn process(
+        instance: ConnectionInstance,
+    ) -> impl std::future::Future<Output = ()> + Send + Sync {
+        let _ = instance;
+        async {}
+    }
 }
 
 pub(crate) struct ExampleServerHandle;
 
 impl ServerHandle<ExampleClientHandle> for ExampleServerHandle {
-    fn process(stream: TcpStream) {}
+    fn process(
+        instance: ConnectionInstance,
+    ) -> impl std::future::Future<Output = ()> + Send + Sync {
+        let _ = instance;
+        async {}
+    }
 }
diff --git a/crates/utils/tcp_connection/tcp_connection_test/src/test_tcp_target_build.rs b/crates/utils/tcp_connection/tcp_connection_test/src/test_tcp_target_build.rs
index b375671..c108e42 100644
--- a/crates/utils/tcp_connection/tcp_connection_test/src/test_tcp_target_build.rs
+++ b/crates/utils/tcp_connection/tcp_connection_test/src/test_tcp_target_build.rs
@@ -26,4 +26,4 @@ async fn test_tcp_test_target_build_domain() {
 
     // Test into string
     assert_eq!(target.to_string(), "127.0.0.1:8080");
-}            default_port,
+}