Wait for further messages

geldata · Feb 24, 2025 · 2358d95 · 2358d95
1 parent abbd4db
commit 2358d95
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/gel-tokio/src/raw/connection.rs b/gel-tokio/src/raw/connection.rs
@@ -418,9 +418,10 @@ async fn connect4(cfg: &Config, mut stream: gel_stream::RawStream) -> Result<Con
                     "Unexpected authentication response",
                 )));
             }
-            ClientAuthResponse::Waiting | ClientAuthResponse::Complete => {
-                continue;
+            ClientAuthResponse::Complete => {
+                break;
             }
+            ClientAuthResponse::Waiting => {}
             ClientAuthResponse::Continue(message) => {
                 send_messages(
                     &mut stream,
@@ -438,6 +439,7 @@ async fn connect4(cfg: &Config, mut stream: gel_stream::RawStream) -> Result<Con
                 return Err(AuthenticationError::with_source(e));
             }
         }
+        msg = wait_message(&mut stream, &mut in_buf, &proto).await?;
     }
 
     let mut server_params = ServerParams::new();