rails_multisite 2.0.6 → 2.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2dcf0982fa73681b77ead88c24501dd8a8dba972b5bd23ecbc3779095dc53168
-  data.tar.gz: 985413ce023cebd3e215b9a961547fa4e2643667dacce3136469236d35ad6ccb
+  metadata.gz: 6cf1529e713938afd1edfb04e2e52f8199f86901f3278821d8514c9bc5855b27
+  data.tar.gz: b5d16cf92694fe94445a141ebc0e8bf2817b665d9789f3ba0ac79ef7af369ed4
 SHA512:
-  metadata.gz: 4a459dc64ff1a9396637e74547fc72c4619c5314a919ef906b54cd94c17d5bdb9042daab13fc04e1be85c1e367030933ac8d40585ba83827cd3f3418155e3545
-  data.tar.gz: e4f77a94d8e91f38c31142175835eb16884d7e5b5f6bcfa4c55c621980f310c48bea4123fd930b50a69f2cb7e0a887154fd2c8210d6a811acc5faffeeb50cb26
+  metadata.gz: 7ecfa664b8367ebd065ecfee14ab1f52ab613aef1c7d404c2944cbedcc93c45899ef53544ca6915260aca0cd5ed6823e6b1bf032569f69fce3001bdaccda12bf
+  data.tar.gz: d3bb77f2e4906ae9736fba551c40972be4a9bd15777447cb5eaaf45d8dc0b0ecf9fa23d1c2baf09dd4b20304776b2b8f7c929aa2173846d64219c73885be6ccf

data/README.md

@@ -35,7 +35,6 @@ mlp:
   host: dbhost
   pool: 5
   timeout: 5000
-  db_id: 1  # ensure db_id is unique for each site
   host_names:
     - discourse.equestria.com
     - discourse.equestria.internal
@@ -48,7 +47,6 @@ drwho:
   host: dbhost
   pool: 5
   timeout: 5000
-  db_id: 2 # ensure db_id is unique for each site
   host_names:
     - discuss.tardis.gallifrey
 ```
@@ -92,7 +90,7 @@ db_two:
 To get a Rails console that is connected to `some_database_1` database:
 
 ```
-RAILS_ENV=db_one rails console
+RAILS_DB=db_one rails console
 ```
 
 

data/lib/rails_multisite/connection_management.rb

@@ -4,12 +4,17 @@ module RailsMultisite
   class ConnectionManagement
 
     DEFAULT = 'default'
+    SPEC_KLASS = ActiveRecord::ConnectionAdapters::ConnectionSpecification
 
     def self.default_config_filename
       File.absolute_path(Rails.root.to_s + "/config/multisite.yml")
     end
 
     def self.clear_settings!
+      @instance&.db_spec_cache&.each do |key, spec|
+        @instance.connection_handlers.delete(self.handler_key(spec))
+      end
+
       @instance = nil
     end
 
@@ -35,7 +40,7 @@ module RailsMultisite
     end
 
     def self.reload
-      @instance = new(instance.config_filename)
+      @instance.reload
     end
 
     def self.has_db?(db)
@@ -97,6 +102,13 @@ module RailsMultisite
       config[:host_names].nil? ? config[:host] : config[:host_names].first
     end
 
+    def self.current_db_hostnames
+      spec = @instance.connection_spec(db: self.current_db) if @instance
+      spec ||= ActiveRecord::Base.connection_pool.spec
+      config = spec.config
+      config[:host_names].nil? ? [config[:host]] : config[:host_names]
+    end
+
     def self.connection_spec(opts)
       if @instance
         @instance.connection_spec(opts)
@@ -113,12 +125,42 @@ module RailsMultisite
       end
     end
 
-    attr_reader :config_filename
+    def self.handler_key(spec)
+      @handler_key_suffix ||= begin
+        if ActiveRecord::Base.respond_to?(:writing_role)
+          "_#{ActiveRecord::Base.writing_role}"
+        else
+          ""
+        end
+      end
+
+      :"#{spec.name}#{@handler_key_suffix}"
+    end
+
+    attr_reader :config_filename, :db_spec_cache, :connection_handlers
 
     def initialize(config_filename)
       @config_filename = config_filename
 
-      spec_klass = ActiveRecord::ConnectionAdapters::ConnectionSpecification
+      @connection_handlers = begin
+        if ActiveRecord::Base.respond_to?(:connection_handlers)
+          ActiveRecord::Base.connection_handlers
+        else
+          {}
+        end
+      end
+
+      @db_spec_cache = {}
+      @default_spec = SPEC_KLASS::Resolver.new(ActiveRecord::Base.configurations).spec(Rails.env.to_sym)
+      @default_connection_handler = ActiveRecord::Base.connection_handler
+      @established_default = false
+
+      @reload_mutex = Mutex.new
+
+      load_config!
+    end
+
+    def load_config!
       configs = YAML::load(File.open(@config_filename))
 
       no_prepared_statements = ActiveRecord::Base.configurations[Rails.env]["prepared_statements"] == false
@@ -129,27 +171,52 @@ module RailsMultisite
         v[:prepared_statements] = false if no_prepared_statements
       end
 
-      resolver = spec_klass::Resolver.new(configs)
-      @db_spec_cache = Hash[*configs.map { |k, _| [k, resolver.spec(k.to_sym)] }.flatten]
+      resolve_configs = configs
+
+      # rails 6 needs to use a proper object for the resolver
+      if defined?(ActiveRecord::DatabaseConfigurations)
+        resolve_configs = ActiveRecord::DatabaseConfigurations.new(configs)
+      end
 
-      @host_spec_cache = {}
+      resolver = SPEC_KLASS::Resolver.new(resolve_configs)
 
+      # Build a hash of db name => spec
+      new_db_spec_cache = Hash[*configs.map { |k, _| [k, resolver.spec(k.to_sym)] }.flatten]
+      new_db_spec_cache.each do |k, v|
+        # If spec already existed, use the old version
+        if v&.to_hash == @db_spec_cache[k]&.to_hash
+          new_db_spec_cache[k] = @db_spec_cache[k]
+        end
+      end
+
+      # Build a hash of hostname => spec
+      new_host_spec_cache = {}
       configs.each do |k, v|
         next unless v["host_names"]
         v["host_names"].each do |host|
-          @host_spec_cache[host] = @db_spec_cache[k]
+          new_host_spec_cache[host] = new_db_spec_cache[k]
         end
       end
 
-      @default_spec = spec_klass::Resolver.new(ActiveRecord::Base.configurations).spec(Rails.env.to_sym)
+      # Add the default hostnames as well
       ActiveRecord::Base.configurations[Rails.env]["host_names"].each do |host|
-        @host_spec_cache[host] = @default_spec
+        new_host_spec_cache[host] = @default_spec
       end
 
-      @default_connection_handler = ActiveRecord::Base.connection_handler
+      removed_dbs = @db_spec_cache.keys - new_db_spec_cache.keys
+      removed_specs = @db_spec_cache.values_at(*removed_dbs)
 
-      @connection_handlers = {}
-      @established_default = false
+      @host_spec_cache = new_host_spec_cache
+      @db_spec_cache = new_db_spec_cache
+
+      # Clean up connection handler cache.
+      removed_specs.each { |s| @connection_handlers.delete(handler_key(s)) }
+    end
+
+    def reload
+      @reload_mutex.synchronize do
+        load_config!
+      end
     end
 
     def has_db?(db)
@@ -171,11 +238,11 @@ module RailsMultisite
       spec ||= @default_spec
       handler = nil
       if spec != @default_spec
-        handler = @connection_handlers[spec]
+        handler = @connection_handlers[handler_key(spec)]
         unless handler
           handler = ActiveRecord::ConnectionAdapters::ConnectionHandler.new
           handler_establish_connection(handler, spec)
-          @connection_handlers[spec] = handler
+          @connection_handlers[handler_key(spec)] = handler
         end
       else
         handler = @default_connection_handler
@@ -313,11 +380,11 @@ module RailsMultisite
     private
 
     def handler_establish_connection(handler, spec)
-      if Rails::VERSION::MAJOR >= 5
-        handler.establish_connection(spec.config)
-      else
-        handler.establish_connection(ActiveRecord::Base, spec)
-      end
+      handler.establish_connection(spec.config)
+    end
+
+    def handler_key(spec)
+      self.class.handler_key(spec)
     end
 
   end

data/lib/rails_multisite/middleware.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 module RailsMultisite
   class Middleware
     def initialize(app, config = nil)

data/lib/rails_multisite/railtie.rb

@@ -7,20 +7,28 @@ module RailsMultisite
     end
 
     initializer "RailsMultisite.init" do |app|
-      Rails.configuration.multisite = false
+      app.config.multisite = false
 
       config_file = ConnectionManagement.default_config_filename
       if File.exist?(config_file)
         ConnectionManagement.config_filename = ConnectionManagement.default_config_filename
-        Rails.configuration.multisite = true
+        app.config.multisite = true
         Rails.logger.formatter = RailsMultisite::Formatter.new
-        app.middleware.insert_after(ActionDispatch::Executor, RailsMultisite::Middleware)
-        app.middleware.delete(ActionDispatch::Executor)
+
+        if !skip_middleware?(app.config)
+          app.middleware.insert_after(ActionDispatch::Executor, RailsMultisite::Middleware)
+          app.middleware.delete(ActionDispatch::Executor)
+        end
 
         if ENV['RAILS_DB'].present?
           ConnectionManagement.establish_connection(db: ENV['RAILS_DB'], raise_on_missing: true)
         end
       end
     end
+
+    def skip_middleware?(config)
+      return false if !config.respond_to?(:skip_multisite_middleware)
+      config.skip_multisite_middleware
+    end
   end
 end

data/lib/rails_multisite/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 #
 module RailsMultisite
-  VERSION = "2.0.6"
+  VERSION = "2.2.2"
 end

data/lib/tasks/db.rake

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 desc "migrate all sites in tier"
 task "multisite:migrate" => :environment do
   RailsMultisite::ConnectionManagement.each_connection do |db|

data/lib/tasks/generators.rake

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 desc "generate multisite config file (if missing)"
 task "multisite:generate:config" => :environment do
   filename = RailsMultisite::ConnectionManagement.config_filename
@@ -14,7 +15,6 @@ task "multisite:generate:config" => :environment do
 #   host: localhost
 #   pool: 5
 #   timeout: 5000
-#   db_id: 1           # optionally include other settings you need
 #   host_names:
 #     - www.mysite.com
 #     - www.anothersite.com

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_multisite
 version: !ruby/object:Gem::Version
-  version: 2.0.6
+  version: 2.2.2
 platform: ruby
 authors:
 - Sam Saffron
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-01-23 00:00:00.000000000 Z
+date: 2020-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -16,40 +16,40 @@ dependencies:
     requirements:
     - - ">"
       - !ruby/object:Gem::Version
-        version: '4.2'
+        version: '5.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6'
+        version: '7'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">"
       - !ruby/object:Gem::Version
-        version: '4.2'
+        version: '5.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6'
+        version: '7'
 - !ruby/object:Gem::Dependency
   name: railties
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">"
       - !ruby/object:Gem::Version
-        version: '4.2'
+        version: '5.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6'
+        version: '7'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">"
       - !ruby/object:Gem::Version
-        version: '4.2'
+        version: '5.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6'
+        version: '7'
 description: Multi tenancy support for Rails
 email:
 - sam.saffron@gmail.com
@@ -78,15 +78,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.2.0
+      version: 2.4.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Multi tenancy support for Rails