Rails：在每個請求上切換連接，但保留連接池

Question

在我們的Rails應用程序中，我們需要根據請求的子域（每個國家/地區的不同數據庫）使用不同的數據庫。

現在我們正在做類似於this question中推薦的內容。也就是說，在每個請求上調用ActiveRecord::Base.establish_connection。

但是it seemsActiveRecord::Base.establish_connection刪除當前連接池並在每次調用時建立一個新連接。

我做了這個快速基準測試，看看是否有呼叫establish_connection每次和在已經建立的連接之間的任何顯著差異：

require 'benchmark/ips' 

$config = Rails.configuration.database_configuration[Rails.env] 
$db1_config = $config.dup.update('database' => 'db1') 
$db2_config = $config.dup.update('database' => 'db2') 

# Method 1: call establish_connection on each "request". 
Benchmark.ips do |r| 
    r.report('establish_connection:') do 
    # Simulate two requests, one for each DB. 
    ActiveRecord::Base.establish_connection($db1_config) 
    MyModel.count # A little query to force the DB connection to establish. 
    ActiveRecord::Base.establish_connection($db2_config) 
    MyModel.count 
    end 
end 

# Method 2: Have different subclasses of my models, one for each DB, and 
# call establish_connection only once 
class MyModelDb1 < MyModel 
    establish_connection($db1_config) 
end 

class MyModelDb2 < MyModel 
    establish_connection($db2_config) 
end 

Benchmark.ips do |r| 
    r.report('different models:') do 
    MyModelDb1.count 
    MyModelDb2.count 
    end 
end 

我rails runner運行此腳本，並指向一個本地的MySQL一些幾千條數據庫上的記錄和結果似乎表明，實際上這兩種方法之間存在很大差異（一個數量級）（順便說一句，我不確定基準是否有效，或者我搞砸了，因此結果是誤導）：

Calculating ------------------------------------- 
establish_connection: 8 i/100ms 
------------------------------------------------- 
establish_connection: 117.9 (±26.3%) i/s -  544 in 5.001575s 
Calculating ------------------------------------- 
    different models: 119 i/100ms 
------------------------------------------------- 
    different models: 1299.4 (±22.1%) i/s -  6188 in 5.039483s 

所以，基本上，我想知道是否有辦法維護每個子域的連接池，然後重新使用這些連接，而不是在每個請求上建立新的連接。由於有許多模型，因此我的模型的子類不適用於每個子域。我只是想改變所有的模型（ActiveRecord::Base）連接

Answer 1

嗯，我一直在深入挖掘這一點，並設法讓一些工作。

讀完tenderlove's post關於ActiveRecord中的連接管理，它解釋了類層次結構如何與連接管理不必要地耦合，我明白了爲什麼要做的事情不像我期望的那樣直截了當。

我最終什麼事做了繼承的ActiveRecord的ConnectionHandler和使用，需要在我的模型層次結構（一些擺弄的ConnectionHandler code頂新的連接處理程序，以瞭解它的內部工作原理，所以當然這種解決方案可能是非常綁定到我使用的Rails版本（3.2））。喜歡的東西：

# A model class that connects to a different DB depending on the subdomain 
# we're in 
class ModelBase < ActiveRecord::Base 
    self.abstract_class = true 
    self.connection_handler = CustomConnectionHandler.new 
end 

# ... 

class CustomConnectionHandler < ActiveRecord::ConnectionAdapters::ConnectionHandler 
    def initialize 
    super 
    @pools_by_subdomain = {} 
    end 

    # Override the behaviour of ActiveRecord's ConnectionHandler to return a 
    # connection pool for the current domain. 
    def retrieve_connection_pool(klass) 
    # Get current subdomain somehow (Maybe store it in a class variable on 
    # each request or whatever) 
    subdomain = @@subdomain 
    @pools_by_subdomain[subdomain] ||= create_pool(subdomain) 
    end 

    private 
    def create_pool(subdomain) 
    conf = Rails.configuration.database_configuration[Rails.env].dup 
    # The name of the DB for that subdomain... 
    conf.update!('database' => "db_#{subdomain}") 
    resolver = ActiveRecord::Base::ConnectionSpecification::Resolver.new(conf, nil) 
    # Call ConnectionHandler#establish_connection, which receives a key 
    # (in this case the subdomain) for the new connection pool 
    establish_connection(subdomain, resolver.spec) 
    end 
end 

這還需要一些測試，以檢查是否有其實是一個性能增益，但我的本地麒麟服務器上運行的最初測試表明有。

Answer 2

據我所知Rails不維護請求之間它的數據庫連接池，除非你使用多線程ENV。像Sidekiq。但是如果您在生產服務器上使用Passenger或Unicorn，它將爲每個Rails實例創建一個新的數據庫連接。

因此基本上使用數據庫連接池是無用的，這意味着在每個請求上創建新的數據庫連接不應該成爲問題。