Function run_execute

src/testdrive/src/action/postgres/execute.rs:32–65

(
    mut cmd: BuiltinCommand,
    state: &State,
)

}

pub async fn run_execute(
    mut cmd: BuiltinCommand,
    state: &State,
) -> Result<ControlFlow, anyhow::Error> {
    let connection = cmd.args.string("connection")?;
    let background = cmd.args.opt_bool("background")?.unwrap_or(false);
    cmd.args.done()?;

    match (connection.starts_with("postgres://"), background) {
        (true, true) => {
            let (client_inner, _) = postgres_client(&connection, state.default_timeout).await?;
            task::spawn(|| "postgres-execute", async move {
                match execute_input(cmd, &client_inner).await {
                    Ok(_) => {}
                    Err(e) => println!("Error in backgrounded postgres-execute query: {e}"),
                }
            });
        }
        (false, true) => bail!("cannot use 'background' arg with referenced connection"),
        (true, false) => {
            let (client_inner, _) = postgres_client(&connection, state.default_timeout).await?;
            execute_input(cmd, &client_inner).await?;
        }
        (false, false) => {
            let client = state
                .postgres_clients
                .get(&connection)
                .ok_or_else(|| anyhow!("connection '{}' not found", &connection))?;
            execute_input(cmd, client).await?;
        }
    }

    Ok(ControlFlow::Continue)
}