Function remap_gets

src/clusterd-test-driver/src/script.rs:310–330

Rewrite every global `Get` in `expr` from the catalog's assigned id back to the script's id, looked up by the object's name.

(
    expr: &mut MirRelationExpr,
    catalog: &TestCatalog,
    name_to_id: &BTreeMap<String, GlobalId>,
)

/// Rewrite every global `Get` in `expr` from the catalog's assigned id back to
/// the script's id, looked up by the object's name.
fn remap_gets(
    expr: &mut MirRelationExpr,
    catalog: &TestCatalog,
    name_to_id: &BTreeMap<String, GlobalId>,
) -> anyhow::Result<()> {
    expr.try_visit_mut_post::<_, anyhow::Error>(&mut |e| {
        if let MirRelationExpr::Get {
            id: Id::Global(g), ..
        } = e
        {
            let name = catalog
                .get_source_name(g)
                .ok_or_else(|| anyhow::anyhow!("get of unknown catalog object {g}"))?;
            let id = name_to_id
                .get(name)
                .ok_or_else(|| anyhow::anyhow!("get of unregistered object {name}"))?;
            *g = *id;
        }
        Ok(())
    })
}

/// A command read from the script stream.
///