Rust中使用MySQL source命令执行外部SQL文件报错求助

问题原因 #

你用source命令执行外部SQL文件报错，是因为**source是MySQL命令行客户端的内置命令，并非MySQL服务器支持的标准SQL语句**，直接通过数据库连接执行会触发语法错误。

解决方案 #

核心思路是：读取外部SQL文件的内容，将其分割为独立的SQL语句，再逐个执行这些语句。

修改后的代码示例 #

async fn test_db(
    user_name: &str,
    password: &str,
    database: &str,
    authorization_script_path: &str,
) -> std::result::Result<(), Box<dyn std::error::Error>> {
    let connection_string = format!("mysql://{}:{}@localhost:3306", user_name, password);
    let pool = Pool::new(&connection_string)?;
    let mut connection = pool.get_conn().await?;

    // 创建目标数据库
    let create_db_query = format!("CREATE DATABASE IF NOT EXISTS {}", database);
    connection.query_drop(create_db_query).await?;

    // 切换到目标数据库
    let use_db_query = format!("USE {}", database);
    connection.query_drop(use_db_query).await?;

    // -------------------------- 替换原source命令的逻辑 --------------------------
    // 异步读取SQL文件内容
    let sql_content = tokio::fs::read_to_string(authorization_script_path).await?;

    // 分割SQL语句（简单处理，过滤空内容和单行注释）
    let statements = sql_content.split(';').filter(|s| !s.trim().is_empty());

    for stmt in statements {
        let trimmed_stmt = stmt.trim();
        if !trimmed_stmt.starts_with("--") && !trimmed_stmt.is_empty() {
            connection.query_drop(trimmed_stmt).await?;
        }
    }
    // --------------------------------------------------------------------------

    Ok(())
}

补充说明 #

1. 依赖调整：确保Cargo.toml中添加了异步文件操作所需的tokio依赖（启用fs特性），以及你正在使用的MySQL异步客户端库（如mysql_async或sqlx）。
2. 复杂场景适配：如果你的SQL文件包含带分号的字符串、存储过程等复杂语句，简单按分号分割会出错，建议使用sqlparser库来精准解析SQL语句，避免分割逻辑出错。

内容的提问来源于stack exchange，提问作者OddlyGhost