diff --git a/src/messages.rs b/src/messages.rs index 45a827c..8e49653 100644 --- a/src/messages.rs +++ b/src/messages.rs @@ -136,9 +136,10 @@ pub async fn startup(stream: &mut TcpStream, user: &str, database: &str) -> Resu match stream.write_all(&startup).await { Ok(_) => Ok(()), - Err(_) => { + Err(err) => { return Err(Error::SocketError(format!( - "Error writing startup to server socket" + "Error writing startup to server socket - Error: {:?}", + err ))) } } @@ -454,7 +455,12 @@ where { match stream.write_all(&buf).await { Ok(_) => Ok(()), - Err(_) => return Err(Error::SocketError(format!("Error writing to socket"))), + Err(err) => { + return Err(Error::SocketError(format!( + "Error writing to socket - Error: {:?}", + err + ))) + } } } @@ -465,7 +471,12 @@ where { match stream.write_all(buf).await { Ok(_) => Ok(()), - Err(_) => return Err(Error::SocketError(format!("Error writing to socket"))), + Err(err) => { + return Err(Error::SocketError(format!( + "Error writing to socket - Error: {:?}", + err + ))) + } } } @@ -476,19 +487,20 @@ where { let code = match stream.read_u8().await { Ok(code) => code, - Err(_) => { + Err(err) => { return Err(Error::SocketError(format!( - "Error reading message code from socket" + "Error reading message code from socket - Error {:?}", + err ))) } }; let len = match stream.read_i32().await { Ok(len) => len, - Err(_) => { + Err(err) => { return Err(Error::SocketError(format!( - "Error reading message len from socket, code: {:?}", - code + "Error reading message len from socket - Code: {:?}, Error: {:?}", + code, err ))) } }; @@ -509,10 +521,10 @@ where .await { Ok(_) => (), - Err(_) => { + Err(err) => { return Err(Error::SocketError(format!( - "Error reading message from socket, code: {:?}", - code + "Error reading message from socket - Code: {:?}, Error: {:?}", + code, err ))) } }; diff --git a/tests/ruby/load_balancing_spec.rb b/tests/ruby/load_balancing_spec.rb index 8be066d..fccf0a8 100644 --- a/tests/ruby/load_balancing_spec.rb +++ b/tests/ruby/load_balancing_spec.rb @@ -93,7 +93,7 @@ describe "Least Outstanding Queries Load Balancing" do threads = Array.new(slow_query_count) do Thread.new do conn = PG.connect(processes.pgcat.connection_string("sharded_db", "sharding_user")) - conn.async_exec("SELECT pg_sleep(1)") + conn.async_exec("BEGIN") end end