diff --git a/scylla-proxy/src/proxy.rs b/scylla-proxy/src/proxy.rs index 7a79d7a9d1..64e0b4f317 100644 --- a/scylla-proxy/src/proxy.rs +++ b/scylla-proxy/src/proxy.rs @@ -750,20 +750,22 @@ impl Doorkeeper { desired_addr }; - socket.connect(real_addr).await.map(|ok| { + let stream = socket.connect(real_addr).await; + if let Ok(ok) = &stream { info!( "Connected to the cluster from {} at {}, intended shard {}.", ok.local_addr().unwrap(), real_addr, shard_preserving_addr.port() % shards ); - ok - }) + } + stream } else { - TcpStream::connect(real_addr).await.map(|ok| { + let stream = TcpStream::connect(real_addr).await; + if stream.is_ok() { info!("Connected to the cluster at {}.", real_addr); - ok - }) + } + stream } .map_err(|err| DoorkeeperError::NodeConnectionAttempt(real_addr, err))?;