Function query_named_rows

nodedb-cluster-tests/tests/cluster_array.rs:43–74

Run `sql` on `client` and return every row's columns as a `HashMap` keyed by column name. Used for `SELECT *` over TVFs that expand into multiple pgwire fields (one per array dimension/attribute).

(
    client: &tokio_postgres::Client,
    sql: &str,
)

/// keyed by column name.  Used for `SELECT *` over TVFs that expand into
/// multiple pgwire fields (one per array dimension/attribute).
async fn query_named_rows(
    client: &tokio_postgres::Client,
    sql: &str,
) -> Vec<std::collections::HashMap<String, String>> {
    let msgs = client.simple_query(sql).await.unwrap_or_else(|e| {
        let detail = if let Some(db) = e.as_db_error() {
            format!(
                "SQLSTATE={} severity={} msg={}",
                db.code().code(),
                db.severity(),
                db.message()
            )
        } else {
            format!("{e:?}")
        };
        panic!("query failed: {detail}\n  sql: {sql}")
    });
    msgs.into_iter()
        .filter_map(|m| {
            if let tokio_postgres::SimpleQueryMessage::Row(r) = m {
                let names: Vec<String> = r.columns().iter().map(|c| c.name().to_string()).collect();
                let mut map = std::collections::HashMap::with_capacity(names.len());
                for (i, name) in names.into_iter().enumerate() {
                    map.insert(name, r.get(i).unwrap_or("").to_string());
                }
                Some(map)
            } else {
                None
            }
        })
        .collect()
}

/// Spin up a 3-node cluster with a pre-populated genome array.
///