@@ -5,7 +5,7 @@ impl TcpRequest {
55 & mut self ,
66 stream : & mut TcpStream ,
77 data : & [ u8 ] ,
8- ) -> Result < BoxResponseTrait , Error > {
8+ ) -> Result < BoxResponseTrait , RequestError > {
99 let mut data_vec: Vec < u8 > = data. into ( ) ;
1010 data_vec. extend_from_slice ( SPLIT_REQUEST_BYTES ) ;
1111 stream
@@ -15,7 +15,7 @@ impl TcpRequest {
1515 self . read_response ( stream)
1616 }
1717
18- fn read_response ( & mut self , stream : & mut TcpStream ) -> Result < BoxResponseTrait , Error > {
18+ fn read_response ( & mut self , stream : & mut TcpStream ) -> Result < BoxResponseTrait , RequestError > {
1919 let cfg_buffer_size: usize = self
2020 . get_config ( )
2121 . read ( )
@@ -36,22 +36,22 @@ impl TcpRequest {
3636 ) ) ;
3737 }
3838
39- fn get_connection_stream ( & self , host : String , port : usize ) -> Result < TcpStream , Error > {
39+ fn get_connection_stream ( & self , host : String , port : usize ) -> Result < TcpStream , RequestError > {
4040 let host_port: ( String , u16 ) = ( host. clone ( ) , port as u16 ) ;
4141 let cfg_timeout: u64 = self
4242 . get_config ( )
4343 . read ( )
4444 . map_or ( DEFAULT_TIMEOUT , |data| data. timeout ) ;
4545 let timeout: Duration = Duration :: from_millis ( cfg_timeout) ;
46- let tcp_stream: TcpStream =
47- TcpStream :: connect ( host_port . clone ( ) ) . map_err ( |_| Error :: TcpStreamConnectError ) ?;
46+ let tcp_stream: TcpStream = TcpStream :: connect ( host_port . clone ( ) )
47+ . map_err ( |_| RequestError :: TcpStreamConnectError ) ?;
4848 tcp_stream
4949 . set_read_timeout ( Some ( timeout) )
50- . map_err ( |_| Error :: SetReadTimeoutError ) ?;
50+ . map_err ( |_| RequestError :: SetReadTimeoutError ) ?;
5151 tcp_stream
5252 . set_write_timeout ( Some ( timeout) )
53- . map_err ( |_| Error :: SetWriteTimeoutError ) ?;
54- let stream: Result < TcpStream , Error > = Ok ( tcp_stream) ;
53+ . map_err ( |_| RequestError :: SetWriteTimeoutError ) ?;
54+ let stream: Result < TcpStream , RequestError > = Ok ( tcp_stream) ;
5555 stream
5656 }
5757}
@@ -68,8 +68,8 @@ impl RequestTrait for TcpRequest {
6868 let port: usize = cfg_timeout. get_port ( ) . clone ( ) ;
6969 let mut stream: TcpStream = self
7070 . get_connection_stream ( host, port)
71- . map_err ( |_| Error :: TcpStreamConnectError ) ?;
72- let res: Result < BoxResponseTrait , Error > = self . send_request ( & mut stream, data) ;
71+ . map_err ( |_| RequestError :: TcpStreamConnectError ) ?;
72+ let res: Result < BoxResponseTrait , RequestError > = self . send_request ( & mut stream, data) ;
7373 res
7474 }
7575}
0 commit comments