Question

我正在尝试将一个activerecord连接传递给更新我的数据库和几个表的gem。到目前为止我没有成功。我需要多个线程连接到数据库以添加数据。

调用MyGem gem的主要代码

database_yml = File.expand_path('database.yml', File.dirname(__FILE__))
db_config = YAML.load_file(database_yml)
connection = ActiveRecord::Base.establish_connection(db_config)

puts "Command array: #{options_array}"
threads = options_array.length
options_array.pmap(threads) do |option|
  MyGem::RoboCopy.run_and_parse_robocopy option, connection
end

这称为Gem方法

module MyGem
  class RoboCopy
    def self.run_and_parse_robocopy(options, db_connection)
      ActiveRecord::Base.connection db_connection
      statistic = RobocopyData.create output_file: options[:output_file], file_gen_start: Time.now.strftime("%Y-%m-%d %H:%M:%S"), status: 'Started'
      #More Code
    end
  end
end

这会导致以下错误：

ArgumentError：调用connection的参数数量错误（1表示0）

我试过

ActiveRecord::Base.connection = db_connection

但得到了这个错误：

NoMethodError：未定义的方法`connection =＆＃39; for ActiveRecord :: Base：Class

我错过了什么？

谢谢

Answer 1

不是传递连接对象本身，而是传递db_config，然后在gem中建立新连接：

在您的主要代码中：

MyGem::RoboCopy.run_and_parse_robocopy option, db_config

在你的宝石中：

module MyGem
  class RoboCopy
    def self.run_and_parse_robocopy(options, db_config)
      ActiveRecord::Base.establish_connection db_config
      # ...
    end
  end
end

如何将activerecord连接传递给ruby中的自定义gem

1 个答案: