fix: small changes

This commit is contained in:
Jindřich Moravec 2024-01-25 23:23:18 +01:00
parent 88fb13325a
commit 4fca7ce12b
3 changed files with 8 additions and 4 deletions

View file

@ -44,7 +44,6 @@ async fn main() -> anyhow::Result<()> {
loop {
let state = state.clone();
let tokens = tokens.clone();
let (pid, key) = random_pid_key();
let (socket, _) = listener.accept().await?;
println!("New client connected: {}", socket.peer_addr()?);
@ -156,6 +155,11 @@ where
while let Some(row) = rows.next() {
writer.write_table_row(&row).await?;
sent_rows += 1;
if token.is_canceled() {
token.reset();
break;
}
}
writer.write_command_complete(CompleteStatus::Select(sent_rows)).await?;