如何在 Heroku 中为一个 rails 3.1 应用程序使用多个数据库？

Question

我的 Rails 3.1 应用程序连接到 2 个数据库，一个是默认数据库，另一个是 Amazon RDS MYSQL 实例。

当前的 database.yml 包含两个生产数据库连接。需要从第二个数据库中提取的模型只需使用

establish_connection "production_on_amazon"

不幸的是，Heroku 覆盖了您的 database.yml，并且似乎只包含一个数据库连接。有谁知道我如何添加或配置我的第二个？

运行“heroku config”我可以看到列出了 2 个数据库，但似乎无法配置为连接到这两个数据库。也许以某种方式将我的默认设置为 Heroku 上的 SHARED_DATABASE_URL db，并将备用设置为指向 Amazon 的 DATABASE_URL...

Answer 1

Heroku 将始终将您的应用连接到他们为您创建的生产数据库。如果您想建立一个额外的连接，您需要在您的代码中手动执行此操作，并创建一个代码可以用作连接字符串的 ENV var。

database.yml 的生产段中的任何内容都被 Heroku 分箱并替换。

Answer 2

关于尼尔的回答，这是一种方法。不是开箱即用的解决方案，但可能会给你一个想法...... /lib/active_record_extensions.rb

module ActiveRecordExtensions
  class Shard < ActiveRecord::Base
    #need to switch to the shard database connection from heroku config 
    primary_database_url = ENV['PRIMARY_DATABASE_URL']

    if(!primary_database_url.nil?)
      parsed_connection_string = primary_database_url.split("://")
      adapter = parsed_connection_string[0]
      parsed_connection_string = parsed_connection_string[1].split(":")
      username = parsed_connection_string[0]
      parsed_connection_string = parsed_connection_string[1].split("@")
      password = parsed_connection_string[0]
      parsed_connection_string = parsed_connection_string[1].split("/")  
      host = parsed_connection_string[0]
      database = parsed_connection_string[1]

      establish_connection(
        :adapter  => adapter,
        :host     => host,
        :username => username,
        :password => password,
        :database => database,
        :port     => 3306,
        :pool     => 5,
        :timeout  => 5000
      )
    else
      self.establish_connection "shard_#{Rails.env}"
    end
  end

  class ShardMigration < ActiveRecord::Migration
    def connection
      ActiveRecord::Shard.connection 
    end
  end
end

所以你的模型应该只扩展 ActiveRecord::Shard 而不是 Base

Answer 3

处理之前的响应，但在配置中结合了 Rails 3 的一些优势并简化了解析...

# config/application.rb
module MyApp
  class Application < Rails::Application
    ... other configs

    config.secondary_database_url = ENV['SECONDARY_DB_URL']
  end
end

我们可能想在开发/测试中覆盖它

# config/environments/development.rb

module MyApp
  class Application < Rails::Application
    ... other configs

    config.secondary_database_url = 'SOME_CONNECTION_STRING'
  end
end    

现在来设置类，我们的模型继承自...

# lib/active_record/secondary.rb 
module ActiveRecord
  class Secondary < ActiveRecord::Base
    self.abstract_class = true

    # prior to AR 3.2.1
    url = URI.parse( MyApp::Application.config.secondary_database_url )
    establish_connection(
      :adapter  => 'mysql',
      :host     => url.host,
      :username => url.userinfo.split(':')[0],
      :password => url.userinfo.split(':')[1],
      :database => url.path[1..-1],
      :port     => url.port || 3306
    )

    # as of AR 3.2.1
    establish_connection(MyApp::Application.config.secondary_database_url)

  end

  class SecondaryMigration < ActiveRecord::Migration
    def connection
      ActiveRecord::Secondary.connection 
    end
  end

end