基于URL参数实现Rails应用动态连接MSSQL数据库的方案咨询

针对你的需求，以下是无需手动配置每个数据库、通过URL路径动态切换连接的具体实现步骤，适配Rails 6.1和MSSQL环境：

一、改造数据库配置模板 #

先在config/database.yml中保留通用默认配置模板，不用写死每个数据库：

default: &default
  adapter: sqlserver
  host: <通用数据库服务器地址>
  username: <只读用户名>
  password: <只读密码>
  driver: tinytds
  timeout: 5000
  pool: <%= ENV.fetch("RAILS_MAX_THREADS") { 5 } %>
  readonly: true # 只读API场景开启只读模式，拦截写操作

development:
  <<: *default
  database: default_dev # 开发环境默认占位库

production:
  <<: *default
  database: default_prod # 生产环境默认占位库

仅保留通用参数，具体数据库名称将从URL动态获取。

二、创建动态切换数据库的中间件 #

在app/middleware/dynamic_db_switcher.rb中编写中间件，从URL路径提取数据库标识并动态建立ActiveRecord连接：

class DynamicDbSwitcher
  def initialize(app)
    @app = app
  end

  def call(env)
    request = Rack::Request.new(env)
    # 从URL路径提取数据库名，比如 /db1/api/users 提取出 db1
    db_name_match = request.path.match(/^\/([a-z0-9_-]+)\//)
    
    if db_name_match
      db_identifier = db_name_match[1]
      # 验证数据库标识合法性，防止非法访问或注入
      unless valid_db_identifier?(db_identifier)
        return [403, {"Content-Type" => "application/json"}, [{error: "Invalid database identifier"}.to_json]]
      end

      # 基于默认配置构建动态数据库连接参数
      db_config = Rails.configuration.database_configuration[Rails.env].dup
      db_config["database"] = db_identifier

      # 为目标数据库建立连接
      ActiveRecord::Base.establish_connection(db_config)
      # 兼容复合主键gem的连接切换
      CompositePrimaryKeys::Base.establish_connection(db_config) if defined?(CompositePrimaryKeys)
    else
      # 无合法路径标识时，连接默认库
      ActiveRecord::Base.establish_connection(Rails.env.to_sym)
      CompositePrimaryKeys::Base.establish_connection(Rails.env.to_sym) if defined?(CompositePrimaryKeys)
    end

    @app.call(env)
  ensure
    # 请求结束后清理连接，避免连接池耗尽
    ActiveRecord::Base.clear_active_connections!
  end

  private

  def valid_db_identifier?(identifier)
    # 可根据实际允许的数据库名规则调整验证逻辑，比如白名单校验
    identifier.match?(/\A[a-z0-9_-]+\z/i)
  end
end

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

module YourAppName
  class Application < Rails::Application
    # ... 其他现有配置
    config.middleware.use DynamicDbSwitcher
  end
end

三、调整路由结构 #

修改config/routes.rb，让所有API路由带上数据库标识前缀：

Rails.application.routes.draw do
  scope "/:db" do
    # 定义你的API路由，示例：
    resources :users, only: [:index, :show]
    resources :orders, only: [:index]
    # 其他API资源...
  end

  # 默认路由，返回指引或错误
  root to: proc { [400, {"Content-Type" => "application/json"}, [{error: "Please specify database path, e.g. /db1"}.to_json]] }
end

四、模型层适配 #

由于所有数据库结构一致，模型无需额外修改，直接继承对应基类即可，中间件已完成全局连接切换：

# app/models/user.rb
class User < ActiveRecord::Base
  self.primary_key = :user_id # 复合主键场景使用CompositePrimaryKeys的写法
end

五、MSSQL适配注意事项 #

1. 编码处理：若遇中文乱码，在默认配置中添加encoding: UTF-8或client_charset: UTF-8。
2. 连接池优化：动态切换多数据库时，确保pool参数设置合理，避免连接耗尽；Rails 6.1会基于不同连接配置创建独立连接池。
3. 只读权限保障：确保数据库用户仅拥有只读权限，配合配置中的readonly: true双重拦截写操作。

六、可选：升级到Rails 7+优化实现 #

若允许升级Rails到7+，可利用connected_to方法更优雅地切换连接，比如在控制器中实现：

class ApiController < ApplicationController
  before_action :switch_database

  private

  def switch_database
    db_identifier = params[:db]
    return head :forbidden unless valid_db_identifier?(db_identifier)

    db_config = Rails.configuration.database_configuration[Rails.env].dup
    db_config["database"] = db_identifier

    # 后续控制器操作将在该连接上下文执行
    ActiveRecord::Base.connected_to(database: db_config) do
      yield if block_given?
    end
  end

升级后需同步将activerecord-sqlserver-adapter升级到对应7.x版本。

七、安全与性能优化 #

• 白名单校验：建议维护允许访问的数据库名白名单，替代仅正则验证，进一步降低非法访问风险。
• 缓存隔离：使用Rails缓存时，需在缓存键中加入数据库标识，避免不同库的数据混淆。
• 日志区分：修改日志格式，加入当前数据库标识，方便问题排查。

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