如何在MySQL中使用变量调用脚本及指定数据库？

核心问题原因 #

你遇到的报错本质是：source和use属于MySQL客户端命令，不是SQL执行语句，MySQL客户端会直接解析这些命令，不会替换其中的用户变量（@xxx），所以才会出现找不到@sname文件、识别不了@db数据库的情况。

分场景解决方案 #

1. 用变量指定要执行的脚本

无法直接在MySQL客户端内用用户变量传递给source，但可以通过系统脚本（Shell/批处理）或者MySQL Shell来实现：

• Shell脚本示例（Linux/macOS）：
  先创建环境配置文件settings_dev.sql：

  SET @db_name = 'devdb';
  SET @script_name = 'crt_db_small.sql';
  

  再写部署脚本deploy.sh：

  #!/bin/bash
  # 从配置文件读取变量
  DB_NAME=$(mysql -u root -p -Nse "source settings_dev.sql; SELECT @db_name;")
  SCRIPT_NAME=$(mysql -u root -p -Nse "source settings_dev.sql; SELECT @script_name;")
  
  # 执行数据库切换与脚本
  mysql -u root -p << EOF
  -- 用动态SQL切换数据库
  SET @sql = CONCAT('USE ', '$DB_NAME');
  PREPARE stmt FROM @sql;
  EXECUTE stmt;
  DEALLOCATE PREPARE stmt;
  -- 执行指定脚本
  source $SCRIPT_NAME;
  EOF
  

• MySQL Shell脚本示例：
  用JavaScript模式编写配置文件settings_dev.js：

  var dbConfig = {
      dbName: 'devdb',
      scriptPath: './crt_db_small.sql'
  };
  

  部署脚本deploy.js：

  // 加载配置
  shell.source('./settings_dev.js');
  // 连接数据库
  var session = shell.connect('root@localhost');
  // 切换数据库
  session.sql(`USE ${dbConfig.dbName}`);
  // 执行指定脚本
  shell.source(dbConfig.scriptPath);
  

2. 用变量切换数据库

不能直接use @db，但可以用动态SQL实现：

SET @db := 'devdb';
-- 构造并执行切换数据库的SQL
SET @sql = CONCAT('USE ', @db);
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;

跨环境部署实现方案 #

完全可以实现你想要的"无需修改脚本即可部署到任意环境"的需求，流程如下：

1. 为每个环境创建独立的配置文件：比如settings_dev.sql、settings_prod.sql，里面定义该环境的数据库名、要执行的脚本路径等变量；
2. 编写通用的部署脚本（Shell/MySQL Shell脚本），该脚本负责读取对应环境的配置文件，然后通过系统变量传递给source命令，通过动态SQL处理数据库切换；
3. 部署时只需指定要加载的配置文件即可，无需修改核心部署逻辑。

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