使用 ActiveRecord 连接动态创建 Ruby 类

Question

我正在尝试动态创建一个类并将几个数据库连接之一分配给每个类。

我正在使用随时间变化的两个或三个数据库之间的任何地方，因此，我不愿将每个连接字符串存储在一个单独的类中并从它继承而不是 ActiveRecord::Base。

以下会引发错误“RuntimeError: Anonymous class is not allowed.”，但我不确定如何解决它或是否有更好的替代方案。

  class ClassFactory
    def self.create_class(new_class, table, db_connection)

      c = Class.new(ActiveRecord::Base) do
        db = db_connection
        self.table_name = table
        establish_connection(:adapter => db.database_type, :host => db.host, :database => db.database, :username => db.username, :password => db.password).connection
      end

      Module.const_set new_class, c
    end
  end

Answer 1

根据ActiveRecord::ConnectionHandling#establish_connection's source，您无法从name 不返回真实值的类建立连接。

现在，您正在使用const_set 将类分配给一个常量，这将为它命名。但是您需要在调用establish_connection 之前这样做：

class ClassFactory
  def self.create_class(new_class, table, db_connection)
    c = Class.new(ActiveRecord::Base) do
      db = db_connection
      self.table_name = table
    end

    Module.const_set new_class, c
    c.establish_connection(:adapter => db.database_type, :host => db.host, :database => db.database, :username => db.username, :password => db.password).connection
  end
end

另外，你真的想要Module.const_set(...)吗？这将产生一个名为Module::Foo 的类。可能你只想要Object.const_set(...)（它只是给Foo）？甚至const_set(...)，所以你得到ClassFactory::Foo？

Answer 2

您可以为模型动态设置establish_connection：

数据库.yml

development:
  adapter: mysql
  username: root
  password: 
  database: example_development

oracle_development:
  adapter: oracle
  username: root
  password: 
  database: example_oracle_development

并且在任何代码位置，您都可以更改模型的数据库连接：

User.establish_connection "oracle_#{RAILS_ENV}".to_sym

您还可以动态创建模型类：

class ClassFactory
  def self.create_class(new_class, table, connection_name)

    Object.const_set(new_class, Class.new(ActiveRecord::Base) {})
    new_class.constantize.table_name = table
    new_class.constantize.establish_connection(connection_name)

  end
end

ClassFactory.create_class('NewUser', 'users', :development)

之后，NewUser 类就可以使用了。

此版本适用于 Rails 5.0 和 3.2。