如何在Ruby on Rails Rake任务中执行MySQL函数创建脚本？

问题根源 #

你遇到的错误核心原因是：DELIMITER是MySQL客户端（如Workbench、mysql命令行）的专属命令，并非MySQL服务器可解析的SQL语句。ActiveRecord的数据库连接直接和服务器通信，无法识别DELIMITER指令，所以直接执行带DELIMITER的脚本必然报错。

另外你尝试的逐行拆分方式也有问题：正则/;$/匹配的是行尾的分号，但原脚本里的分号位置和函数体结构会导致拆分后的语句不完整或包含无效内容。

可行解决方案 #

方案1：调用系统mysql命令执行脚本（推荐）

这种方式和你用rails dbconsole < lib/sql/create-func.sql的原理一致，借助MySQL客户端处理DELIMITER，避免ActiveRecord的限制。

在Rake任务中添加以下代码：

namespace :db do
  desc "Create MySQL functions and stored procedures"
  task create_db_objects: :environment do
    # 获取当前环境的数据库配置
    db_config = Rails.configuration.database_configuration[Rails.env]
    
    # 构造mysql命令
    mysql_cmd = [
      "mysql",
      "-u#{db_config['username']}",
      db_config['password'].present? ? "-p#{db_config['password']}" : "",
      "-h#{db_config['host'] || 'localhost'}",
      "#{db_config['database']}",
      "< #{Rails.root}/lib/sql/create-func.sql"
    ].reject(&:empty?).join(' ')
    
    # 执行命令
    system(mysql_cmd)
    
    if $?.success?
      puts "数据库函数/存储过程创建成功"
    else
      puts "数据库对象创建失败，请检查脚本或配置"
      exit 1
    end
  end
end

注意：如果数据库密码包含特殊字符，建议用--defaults-extra-file参数读取配置文件，避免命令行暴露密码。

方案2：修改SQL脚本，适配ActiveRecord执行

去掉脚本中的DELIMITER和USE指令（ActiveRecord已经连接到当前环境的数据库），并将函数定义作为单个语句执行：

修改后的SQL脚本（lib/sql/create-func-clean.sql）：

DROP function IF EXISTS prov_id;

CREATE DEFINER=`root`@`localhost` FUNCTION prov_id(prov char(2)) RETURNS bigint
    READS SQL DATA
    DETERMINISTIC
BEGIN
    DECLARE pvid BIGINT;
    SELECT p.id INTO pvid FROM provinces AS p WHERE p.Prov = prov;
RETURN pvid;
END;

然后在Rake任务中直接读取并执行整个函数创建语句（注意不要拆分，因为函数体包含内部分号）：

namespace :db do
  desc "Create MySQL functions via ActiveRecord"
  task create_functions: :environment do
    # 读取清理后的SQL脚本
    sql = File.read("#{Rails.root}/lib/sql/create-func-clean.sql")
    
    # 分两次执行：先删除旧函数，再创建新函数
    sql.split(/;\s*\n/).each do |stmt|
      next if stmt.strip.empty?
      ActiveRecord::Base.connection.execute(stmt)
    end
    
    puts "函数创建成功"
  end
end

原理：ActiveRecord的execute支持包含内部分号的单个SQL语句（比如函数/存储过程定义），只要整个语句是完整的。

为什么原来的方法失效？ #

1. DELIMITER命令不被服务器识别：ActiveRecord直接和MySQL服务器通信，服务器无法解析客户端专属的DELIMITER指令，导致语法错误。
2. 拆分逻辑错误：原脚本中的split('/;$/')正则匹配的是行尾分号，但函数体内部的分号会被误拆分，导致执行不完整的SQL语句。

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