Function query_message

nodedb/tests/pgwire_tls_e2e.rs:145–156

Build a pgwire simple Query message.

(sql: &str)

/// Build a pgwire simple Query message.
fn query_message(sql: &str) -> Vec<u8> {
    let body = {
        let mut b = sql.as_bytes().to_vec();
        b.push(0); // null terminator
        b
    };
    let msg_len = (4u32 + body.len() as u32).to_be_bytes();
    let mut msg = vec![b'Q'];
    msg.extend_from_slice(&msg_len);
    msg.extend_from_slice(&body);
    msg
}

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