如何让Rails模型动态访问不同的数据库？

核心思路 #

基于Active Record的动态连接切换能力，结合请求上下文识别区域标识，将请求路由到对应区域的独立数据库，实现数据隔离且无需修改模型结构或部署多实例。

一、配置数据库连接模板 #

在config/database.yml中定义基础连接模板，后续动态替换区域对应的数据库名：

default: &default
  adapter: mysql2
  encoding: utf8mb4
  pool: <%= ENV.fetch("RAILS_MAX_THREADS") { 5 } %>
  username: <%= ENV['DB_USERNAME'] %>
  password: <%= ENV['DB_PASSWORD'] %>
  host: <%= ENV['DB_HOST'] %>

# 仅作为模板，无需实际创建该数据库
region_template:
  <<: *default
  database: app_db_#{region} # 动态占位符，后续替换为实际区域名

二、通过中间件处理区域识别与连接切换 #

创建请求中间件，从请求中提取区域标识（支持HTTP头、子域名、请求参数等方式），并为当前请求线程切换到对应数据库：

# app/middleware/region_db_middleware.rb
class RegionDbMiddleware
  def initialize(app)
    @app = app
  end

  def call(env)
    request = Rack::Request.new(env)
    # 从请求头获取区域标识，可根据实际场景改为子域名/请求参数
    region = request.headers['X-Region'] || 'default'
    
    # 校验区域合法性，防止恶意请求
    valid_regions = ENV['VALID_REGIONS']&.split(',') || ['default']
    unless valid_regions.include?(region)
      return [400, {'Content-Type' => 'application/json'}, [{error: 'Invalid region'}.to_json]]
    end

    # 生成当前区域的数据库配置
    db_config = Rails.configuration.database_configuration['region_template'].dup
    db_config['database'] = "app_db_#{region}"
    
    # 切换全局Active Record连接
    ActiveRecord::Base.establish_connection(db_config)
    
    # 处理请求
    response = @app.call(env)
    
    # 请求结束后清除连接，避免线程污染
    ActiveRecord::Base.clear_active_connections!
    
    response
  end
end

在config/application.rb中注册中间件：

config.middleware.use RegionDbMiddleware

三、模型层优化（可选） #

创建抽象基类模型，让所有业务模型继承该类，后续可更灵活地控制连接范围：

# app/models/region_base.rb
class RegionBase < ActiveRecord::Base
  self.abstract_class = true
end

# 业务模型示例
class User < RegionBase
  # 模型逻辑
end

此时中间件中只需切换基类的连接：

# 替换中间件中的连接代码
RegionBase.establish_connection(db_config)

四、关键注意事项 #

• 线程安全：确保establish_connection和clear_active_connections!在请求周期内执行，避免线程间连接串用。
• 连接池配置：根据区域数量调整RAILS_MAX_THREADS和数据库最大连接数，防止连接耗尽。
• 数据库初始化：提前批量创建各区域数据库并同步schema，可通过rails db:clone或自定义脚本实现。
• 缓存隔离：缓存键需添加区域前缀，避免跨区域缓存污染，示例：

  Rails.cache.fetch("user_#{user_id}_#{region}") do
    # 缓存逻辑
  end
  

• 异步任务处理：异步任务（如Sidekiq）需传递区域标识，执行时切换对应数据库：

  # 触发任务时传递区域
  UserSyncJob.perform_async(user_id, region)
  
  # 任务内切换连接
  class UserSyncJob < ApplicationJob
    def perform(user_id, region)
      db_config = Rails.configuration.database_configuration['region_template'].dup
      db_config['database'] = "app_db_#{region}"
      RegionBase.establish_connection(db_config)
      
      user = User.find(user_id)
      # 任务逻辑
      
      RegionBase.clear_active_connections!
    end
  end
  

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