Function read_pg_message

nodedb/tests/pgwire_tls_e2e.rs:160–171

Read a single pgwire message from the stream. Returns `(type_byte, payload)` where payload excludes the 4-byte length field.

(r: &mut R)

/// Read a single pgwire message from the stream.
/// Returns `(type_byte, payload)` where payload excludes the 4-byte length field.
async fn read_pg_message<R: AsyncReadExt + Unpin>(r: &mut R) -> (u8, Vec<u8>) {
    let mut header = [0u8; 5];
    r.read_exact(&mut header).await.expect("read msg header");
    let msg_type = header[0];
    let length = u32::from_be_bytes(header[1..5].try_into().unwrap()) as usize;
    let payload_len = length.saturating_sub(4);
    let mut payload = vec![0u8; payload_len];
    if payload_len > 0 {
        r.read_exact(&mut payload).await.expect("read msg payload");
    }
    (msg_type, payload)
}

// ── Test ──────────────────────────────────────────────────────────────────