mz_testdrive/action/mysql/
execute.rs

1// Copyright Materialize, Inc. and contributors. All rights reserved.
2//
3// Use of this software is governed by the Business Source License
4// included in the LICENSE file.
5//
6// As of the Change Date specified in that file, in accordance with
7// the Business Source License, use of this software will be governed
8// by the Apache License, Version 2.0.
9
10use anyhow::{Context, anyhow};
11use mysql_async::prelude::Query;
12
13use crate::action::{ControlFlow, State};
14use crate::parser::BuiltinCommand;
15
16pub async fn run_execute(
17    mut cmd: BuiltinCommand,
18    state: &mut State,
19) -> Result<ControlFlow, anyhow::Error> {
20    let name = cmd.args.string("name")?;
21    cmd.args.done()?;
22
23    let conn = state
24        .mysql_clients
25        .get_mut(&name)
26        .ok_or_else(|| anyhow!("MySQL connection '{}' not found", &name))?;
27
28    for query in cmd.input {
29        println!(">> {}", query);
30        query
31            .run(&mut *conn)
32            .await
33            .context("executing MySQL query")?;
34    }
35
36    Ok(ControlFlow::Continue)
37}