Function read_until_ready

tests/benchmark_batch_operations.rs:263–279

(stream: &mut TcpStream)

}

async fn read_until_ready(stream: &mut TcpStream) {
    loop {
        let mut msg_type = [0u8; 1];
        stream.read_exact(&mut msg_type).await.unwrap();
        
        let mut len_buf = [0u8; 4];
        stream.read_exact(&mut len_buf).await.unwrap();
        let len = u32::from_be_bytes(len_buf) as usize - 4;
        
        let mut data = vec![0u8; len];
        stream.read_exact(&mut data).await.unwrap();
        
        if msg_type[0] == b'Z' { // ReadyForQuery
            break;
        }
    }
}