rails_multisite_discourse 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 992f8e0cc674d6bdc7f6e252940596f006de0c64
+  data.tar.gz: 0246b02a3da70e424823b77a5c5c69ff2b166885
+SHA512:
+  metadata.gz: f06bfa63fd47d83ebb99a4a0f5e111c3a238c926386b060101a0990255368a9e8983e70dbbf83096965ad77ee016d56941600f7dd49574219cb69c91eb95a556
+  data.tar.gz: 9f91b582b0adc7ec741119ae33158bcfc267737670cf947ed0c41cda68f3a709b3e4d33c6f8457ce6ce4347a9b89107e4dfd77eb9845c3025363c0b821f77da1

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Sam Saffron
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,54 @@
+# Rails Multisite
+
+This gem provides multi-db support for Rails applications.
+
+Using its middleware you can partition your app so each hostname has its own db.
+
+It provides a series of helper for working with multiple database, and some additional rails tasks for working with them.
+
+It was extracted from Discourse. http://discourse.org
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rails_multisite'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rails_multisite
+
+## Usage
+
+Configuration requires a file called: `config/multisite.yml` that specifies connection specs for all dbs.
+
+
+### Exectue a query on each connection
+
+```
+RailsMultisite::ConnectionManagement.each_connection do |db|
+  # run query in context of db
+  # eg: User.find(1)
+end
+```
+
+```
+RailsMultisite::ConnectionManagement.each_connection(threads: 5) do |db|
+  # run query in context of db, will do so in a thread pool of 5 threads
+  # if any query fails an exception will be raised
+  # eg: User.find(1)
+end
+```
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/lib/rails_multisite.rb

@@ -0,0 +1,3 @@
+require "rails_multisite/version"
+require "rails_multisite/railtie"
+require "rails_multisite/connection_management"

data/lib/rails_multisite/connection_management.rb

@@ -0,0 +1,236 @@
+module RailsMultisite
+  class ConnectionManagement
+    CONFIG_FILE = 'config/multisite.yml'
+    DEFAULT = 'default'.freeze
+
+    def self.has_db?(db)
+      return true if db == DEFAULT
+      (defined? @@db_spec_cache) && @@db_spec_cache && @@db_spec_cache[db]
+    end
+
+    def self.rails4?
+      !!(Rails.version =~ /^4/)
+    end
+
+    def self.establish_connection(opts)
+      if opts[:db] == DEFAULT && (!defined?(@@default_spec) || !@@default_spec)
+        # don't do anything .. handled implicitly
+      else
+        spec = connection_spec(opts) || @@default_spec
+        handler = nil
+        if spec != @@default_spec
+          handler = @@connection_handlers[spec]
+          unless handler
+            handler = ActiveRecord::ConnectionAdapters::ConnectionHandler.new
+            handler.establish_connection(ActiveRecord::Base, spec)
+            @@connection_handlers[spec] = handler
+          end
+        else
+          handler = @@default_connection_handler
+          if !@@established_default
+            handler.establish_connection(ActiveRecord::Base, spec)
+            @@established_default = true
+          end
+        end
+
+        ActiveRecord::Base.connection_handler = handler
+      end
+    end
+
+    def self.with_hostname(hostname)
+
+      unless defined? @@db_spec_cache
+        # just fake it for non multisite
+        yield hostname
+        return
+      end
+
+      old = current_hostname
+      connected = ActiveRecord::Base.connection_pool.connected?
+
+      establish_connection(:host => hostname) unless connected && hostname == old
+      rval = yield hostname
+
+      unless connected && hostname == old
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+
+        establish_connection(:host => old)
+        ActiveRecord::Base.connection_handler.clear_active_connections! unless connected
+      end
+
+      rval
+    end
+
+    def self.with_connection(db = "default")
+      old = current_db
+      connected = ActiveRecord::Base.connection_pool.connected?
+
+      establish_connection(:db => db) unless connected && db == old
+      rval = yield db
+
+      unless connected && db == old
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+
+        establish_connection(:db => old)
+        ActiveRecord::Base.connection_handler.clear_active_connections! unless connected
+      end
+
+      rval
+    end
+
+    def self.each_connection(opts=nil, &blk)
+
+      old = current_db
+      connected = ActiveRecord::Base.connection_pool.connected?
+
+      queue = nil
+      threads = nil
+
+      if (opts && (threads = opts[:threads]))
+        queue = Queue.new
+        all_dbs.each{|db| queue << db}
+      end
+
+      errors = nil
+
+      if queue
+        threads.times.map do
+          Thread.new do
+
+            while true
+              begin
+                db = queue.deq(true)
+              rescue ThreadError
+                db = nil
+              end
+
+              break unless db
+
+              establish_connection(:db => db)
+              # no choice but to rescue, should probably log
+
+              begin
+                blk.call(db)
+              rescue => e
+                (errors ||= []) << e
+              end
+              ActiveRecord::Base.connection_handler.clear_active_connections!
+            end
+          end
+        end.map(&:join)
+      else
+        all_dbs.each do |db|
+          establish_connection(:db => db)
+          blk.call(db)
+          ActiveRecord::Base.connection_handler.clear_active_connections!
+        end
+      end
+
+      if errors && errors.length > 0
+        raise StandardError, "Failed to run queries #{errors.inspect}"
+      end
+
+
+    ensure
+      establish_connection(:db => old)
+      ActiveRecord::Base.connection_handler.clear_active_connections! unless connected
+    end
+
+    def self.all_dbs
+      ["default"] +
+        if defined?(@@db_spec_cache) && @@db_spec_cache
+          @@db_spec_cache.keys.to_a
+        else
+          []
+        end
+    end
+
+    def self.current_db
+      ActiveRecord::Base.connection_pool.spec.config[:db_key] || "default"
+    end
+
+    def self.config_filename=(config_filename)
+      @@config_filename = config_filename
+    end
+
+    def self.config_filename
+      @@config_filename ||= File.absolute_path(Rails.root.to_s + "/" + RailsMultisite::ConnectionManagement::CONFIG_FILE)
+    end
+
+    def self.current_hostname
+      config = ActiveRecord::Base.connection_pool.spec.config
+      config[:host_names].nil? ? config[:host] : config[:host_names].first
+    end
+
+    def self.clear_settings!
+      @@db_spec_cache = nil
+      @@host_spec_cache = nil
+      @@default_spec = nil
+    end
+
+    def self.load_settings!
+      spec_klass = ActiveRecord::ConnectionAdapters::ConnectionSpecification
+      configs = YAML::load(File.open(self.config_filename))
+      configs.each do |k,v|
+        raise ArgumentError.new("Please do not name any db default!") if k == "default"
+        v[:db_key] = k
+      end
+
+      @@db_spec_cache = Hash[*configs.map do |k, data|
+        [k, spec_klass::Resolver.new(configs).spec(k)]
+      end.flatten]
+
+      @@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]
+        end
+      end
+
+      @@default_spec = spec_klass::Resolver.new(ActiveRecord::Base.configurations).spec(Rails.env)
+      ActiveRecord::Base.configurations[Rails.env]["host_names"].each do |host|
+        @@host_spec_cache[host] = @@default_spec
+      end
+
+      @@default_connection_handler = ActiveRecord::Base.connection_handler
+
+      @@connection_handlers = {}
+      @@established_default = false
+    end
+
+
+    def initialize(app, config = nil)
+      @app = app
+    end
+
+    def self.host(env)
+      request = Rack::Request.new(env)
+      request['__ws'] || request.host
+    end
+
+    def call(env)
+      host = self.class.host(env)
+      begin
+
+        #TODO: add a callback so users can simply go to a domain to register it, or something
+        return [404, {}, ["not found"]] unless @@host_spec_cache[host]
+
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+        self.class.establish_connection(:host => host)
+        @app.call(env)
+      ensure
+        ActiveRecord::Base.connection_handler.clear_active_connections!
+      end
+    end
+
+    def self.connection_spec(opts)
+      if opts[:host]
+        @@host_spec_cache[opts[:host]]
+      else
+        @@db_spec_cache[opts[:db]]
+      end
+    end
+
+  end
+end

data/lib/rails_multisite/railtie.rb

@@ -0,0 +1,23 @@
+module RailsMultisite
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      Dir[File.join(File.dirname(__FILE__),'../tasks/*.rake')].each { |f| load f }
+    end
+
+    initializer "RailsMultisite.init" do |app|
+      Rails.configuration.multisite = false
+      if File.exists?(ConnectionManagement.config_filename)
+        Rails.configuration.multisite = true
+        ConnectionManagement.load_settings!
+        app.middleware.insert_after(ActiveRecord::ConnectionAdapters::ConnectionManagement, RailsMultisite::ConnectionManagement)
+        app.middleware.delete(ActiveRecord::ConnectionAdapters::ConnectionManagement)
+        if ENV['RAILS_DB']
+          ConnectionManagement.establish_connection(:db => ENV['RAILS_DB'])
+        end
+      end
+    end
+
+
+  end
+end
+

data/lib/rails_multisite/version.rb

@@ -0,0 +1,3 @@
+module RailsMultisite
+  VERSION = "1.0.0"
+end

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: rails_multisite_discourse
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Sam Saffron
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-10-12 00:00:00.000000000 Z
+dependencies: []
+description: Multi tenancy support for Rails
+email:
+- sam.saffron@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/rails_multisite.rb
+- lib/rails_multisite/connection_management.rb
+- lib/rails_multisite/railtie.rb
+- lib/rails_multisite/version.rb
+homepage: ''
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Multi tenancy support for Rails
+test_files: []