multidb 3.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 871069e0b89ed728780ae50f13950dc1e8e1c6db
+  data.tar.gz: 56645bb4e9fd29f8f696e6e2534f963ce414dc62
+SHA512:
+  metadata.gz: 67773f48625923f72efa87033d8e988ef36679c58d7ad928c1b79abd13722c54b14590c390c6e1d0c4542cb420199d8a709776ebdfbe2f0ef564086157909b74
+  data.tar.gz: 8ca58cc43cc994807fb72979866defbaf325f861f77b5ab2809c55a08c10b17673b1bead7a26e4482e0aaf85bff395cb5dd64093e9241e7800144670bda76dd2

data/.gitignore

@@ -0,0 +1,12 @@
+# See http://help.github.com/ignore-files/ for more about ignoring files.
+
+# Ignore .lock file as this is a gem
+Gemfile.lock
+
+# Ignore gem
+multidb-*.gem
+
+# Ignore all logfiles and tempfiles.
+testapp_*/coverage
+testapp_*/log/*.log
+testapp_*/tmp

data/.rspec

@@ -0,0 +1,3 @@
+--color
+--drb
+--format documentation

data/.ruby-version

@@ -0,0 +1 @@
+ruby-2.1.0

data/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/LICENSE.md

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2014 Aaron Namba
+
+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,65 @@
+#About MultiDB
+
+MultiDB is a multitenant extension for Rails (or just ActiveRecord) that allows you to isolate each tenant into its own individual database without requiring major changes to your application code.
+
+MultiDB is _not_ meant for systems that may have large numbers of tenants; you wouldn't want to have that many databases. It was designed for a system with 100-200 tenants, with the option to go to 1,000, and an absolute upper limit of 10,000. If you anticipate having more, MultiDB may not be an appropriate solution. (MySQL itself does not specify an upper limit on the number of databases, but it will be constrained by open file limits and, in some cases, directory entry limits.)
+
+To minimize API changes for you, MultiDB patches ActiveRecord, ActionController and their associated rake tasks as needed to enable database switching at appropriate times and add support for three sets of schemas and migrations. ActiveRecord::SessionStore::Session has also been patched.
+
+Internally, MultiDB refers to tenants as organizations. In addition to the organization databases, two additional databases are created: one is just for sessions, and the other is referred to as the master database, which houses the organizations table (tenants), and all other tables not associated with a particular organization. (Yes, it seems like these two additional databases could be combined, but it is a design decision intended to keep all essential data out of the default database. More on this later.)
+
+MultiDB, when used with Rails (ActionController), determines which database to connect to at the beginning of each request by checking for `request.host`, `params[:org_code]`, then `session[:org_code]`. In a test environment, it can will also check the environment variable `RAILS_ORG`. If no organization code is found in any of those places, the sessions database is used (which is one reason it is important that no actual data be stored there).
+
+
+## Compatibility
+
+### Rails & ActiveRecord
+MultiDB 3.2 works with Rails 3.2. A new branch will be created to work with Rails 4.
+
+The concept should work with just ActiveRecord (no Rails), but this use case has not been tested. (Pull requests welcome.)
+
+### Database Adapters
+Warning: MultiDB works only with mysql2 at present. If you are handy with Ruby, please help to add support for more adapters. See the "Contributing" section below.
+
+MultiDB is known to work with [Makara](https://github.com/taskrabbit/makara) in a production environment.
+
+
+##Get Started With MultiDB
+
+Install the gem:
+
+    gem install multidb
+
+Or add it to your Gemfile:
+
+    gem 'multidb'
+
+Then have your Organization class inherit from MultiDB::Organization. This gets you:
+
+    Organization#connect(set_env = false)
+    Organization#create_database
+    Organization#drop_database!
+
+Your table will need to have columns for `code` (string) and `active` (boolean), and to use the request.host feature, you will need organization_hosts as well. Recommended migration:
+
+    create_table "organization_hosts", :force => true do |t|
+      t.integer  "organization_id", :null => false
+      t.string   "host",            :null => false
+      t.datetime "created_at",      :null => false
+      t.datetime "updated_at",      :null => false
+    end
+    add_index "organization_hosts", ["host"], :name => "index_organization_hosts_on_host", :unique => true
+    
+    create_table "organizations", :force => true do |t|
+      t.string   "name"
+      t.string   "code",                         :null => false
+      t.boolean  "active",     :default => true, :null => false
+      t.datetime "created_at",                   :null => false
+      t.datetime "updated_at",                   :null => false
+    end
+    add_index "organizations", ["code"], :name => "index_organizations_on_code", :unique => true
+
+
+## Contributing
+
+Pull requests welcome. Don't forget tests! When adding support for a new adapter, create a new Rails app for your adapter (e.g. `rails _3.2.16_ new testapp_postgresql`), customize database.yml and other files as needed, then add specs to that app.

data/lib/multi_db/action_controller_patches.rb

@@ -0,0 +1,53 @@
+class ActionController::Base
+  prepend_before_filter :connect_to_organization_database
+  
+  # manually establish a connection to the proper database
+  def connect_to_organization_database
+    @org = nil
+    
+    # request is first priority
+    if params[:org_code]
+      if session[:org_code] && session[:org_code] != params[:org_code]
+        reset_session
+      end
+      @org = MultiDB::Organization.active.where(:code => params[:org_code]).first
+    end
+    
+    # try hostname if we don't already have a code in the session
+    if !@org && !session[:org_code] && request && request.host
+      @org ||= MultiDB::Organization.active.where(:code => $1.gsub('-', '_')).first if request.host =~ /^([-\w\d]+)/
+      @org ||= MultiDB::Organization.active.includes(:hosts).where('organization_hosts.host = ?', request.host).first
+    end
+    
+    if @org
+      if session[:org_code] != @org.code
+        session[:org_code] = @org.code
+        session[:org_name] = @org.name
+      end
+    end
+    
+    if session[:org_code]
+      @org ||= MultiDB::Organization.active.where(:code => session[:org_code]).first
+      if @org
+        ActiveRecord::Base.connect_to_organization(session[:org_code], true)
+        return @org
+      end
+    end
+    
+    if Rails.env.test? && ENV['RAILS_ORG']
+      @org ||= MultiDB::Organization.active.where(:code => ENV['RAILS_ORG']).first
+      if @org
+        ActiveRecord::Base.connect_to_organization(session[:org_code], true)
+        return @org
+      end
+    end
+    
+    # if we don't issue an establish_connection by now, connect to default db (sessions)
+    session[:org_code] = session[:org_name] = nil
+    ActiveRecord::Base.connect_to_sessions
+  end
+  
+  def connect_to_master_database
+    ActiveRecord::Base.connect_to_master
+  end
+end

data/lib/multi_db/active_record_patches.rb

@@ -0,0 +1,94 @@
+module ActiveRecord
+  class Base
+    class << self
+      
+      def master_configuration(env = nil)
+        env ||= Rails.env
+        
+        # use master db configuration in config/database.yml if present
+        configurations["master_#{env}"] or Proc.new {
+          c = configurations[env].dup
+          c['database'] += '_master'
+          c
+        }.call
+      end
+      
+      def connect_to_sessions
+        config = configurations[Rails.env || 'development']
+        self.establish_connection(config)
+      end
+      
+      def connect_to_master
+        self.establish_connection(master_configuration)
+      end
+      
+      def connect_to_organization(org = nil, set_env = false)
+        org_code = org.is_a?(MultiDB::Organization) ? org.code : org
+        org_code ||= ENV['RAILS_ORG'] || 'org1'
+        
+        config = ActiveRecord::Base.configurations[Rails.env]
+        self.establish_connection(config.merge('database' => "#{config['database']}_#{org_code}"))
+        
+        begin
+          ActiveRecord::Base.connection.tables
+        rescue Mysql2::Error
+          Rails.logger.error "Couldn't connect to org db for #{org_code}"
+          return false
+        end
+        
+        ENV['RAILS_ORG'] = org_code if set_env
+        org_code
+      end
+    end
+  end
+  
+  class Migration
+    
+    alias_method :migrate_without_multidb, :migrate
+    def migrate(direction)
+      if ENV['RAILS_ORG'] == 'master'
+        Base.connect_to_master
+      elsif ENV['RAILS_ORG'] == 'sessions'
+        Base.connect_to_sessions
+      else
+        Base.connect_to_organization
+      end
+      
+      migrate_without_multidb(direction)
+    end
+    
+  end
+  
+  class Migrator
+    
+    alias_method :initialize_without_multidb, :initialize
+    def initialize(direction, migrations_paths, target_version = nil)
+      if ENV['RAILS_ORG'] == 'master'
+        Base.connect_to_master
+      elsif ENV['RAILS_ORG'] == 'sessions'
+        Base.connect_to_sessions
+      else
+        Base.connect_to_organization
+      end
+      
+      initialize_without_multidb(direction, migrations_paths, target_version)
+    end
+    
+    class << self
+      
+      def migrations_paths
+        @migrations_paths ||= ['db/migrate']
+        # just to not break things if someone uses: migration_path = some_string
+        paths = Array.wrap(@migrations_paths)
+        
+        case ENV['RAILS_ORG']
+        when 'sessions', 'master'
+          paths.map { |path| "#{path}/#{ENV['RAILS_ORG']}" }
+        else
+          paths.map { |path| "#{path}/org" }
+        end
+      end
+      
+    end
+  end
+end

data/lib/multi_db/after_initialize_patches.rb

@@ -0,0 +1,3 @@
+class ActiveRecord::SessionStore::Session < ActiveRecord::Base
+  connect_to_sessions
+end

data/lib/multi_db/engine.rb

@@ -0,0 +1,28 @@
+module MultiDB
+  class Engine < Rails::Engine
+    engine_name "multidb"
+    
+    config.app_root = root
+    config.autoload_paths += Dir["#{config.root}/lib/**/"]
+    
+    ActiveSupport.on_load(:active_record) do
+      require 'multi_db/active_record_patches'
+    end
+    
+    ActiveSupport.on_load(:action_controller) do
+      require 'multi_db/action_controller_patches'
+    end
+    
+    ActiveSupport.on_load(:after_initialize) do
+      require 'multi_db/after_initialize_patches'
+    end
+  end
+end
+
+module MultiDB
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load "multi_db/multidb.rake"
+    end
+  end
+end

data/lib/multi_db/multidb.rake

@@ -0,0 +1,360 @@
+#
+# Based on activerecord-3.2.15/lib/active_record/railties/databases.rake
+# Last updated 2013-11-19 by Aaron Namba
+#
+# AKN: Warning! All code to support db engines other than mysql has been deleted,
+#      since I know it will never receive proper testing.
+#
+
+tasks = Rake.application.instance_variable_get '@tasks'
+tasks.delete 'db:migrate'
+tasks.delete 'db:migrate:up'
+tasks.delete 'db:migrate:down'
+tasks.delete 'db:rollback'
+tasks.delete 'db:forward'
+tasks.delete 'db:schema:dump'
+tasks.delete 'db:schema:load'
+tasks.delete 'db:test:load_schema'
+tasks.delete 'db:test:purge'
+
+
+db_namespace = namespace :db do
+  namespace :create do
+    desc 'Runs create for sessions and master databases.'
+    task :multi => [:load_config] do
+      [ 'sessions', 'master' ].each do |org|
+        ENV['RAILS_ORG'] = org
+        db_namespace[:create].execute
+      end
+    end
+  end
+  
+  namespace :drop do
+    desc 'Runs drop for sessions and master databases.'
+    task :multi => [:load_config] do
+      [ 'sessions', 'master' ].each do |org|
+        ENV['RAILS_ORG'] = org
+        db_namespace[:drop].execute
+      end
+    end
+  end
+  
+  namespace :setup do
+    desc 'Runs setup for sessions and master databases.'
+    task :multi => [:load_config] do
+      [ 'sessions', 'master' ].each do |org|
+        ENV['RAILS_ORG'] = org
+        db_namespace[:create].execute
+        db_namespace['schema:load'].execute
+        db_namespace[:seed].execute
+      end
+    end
+  end
+  
+  namespace :migrate do
+    desc 'Runs migrate for sessions, master and org databases.'
+    task :multi => [:environment, :load_config] do
+      puts "=========================================================="
+      puts "===== SESSIONS"
+      puts "=========================================================="
+      puts
+      ENV['RAILS_ORG'] = 'sessions'
+      db_namespace[:migrate].execute
+      puts
+      
+      puts "=========================================================="
+      puts "===== MASTER"
+      puts "=========================================================="
+      puts
+      ENV['RAILS_ORG'] = 'master'
+      db_namespace[:migrate].execute
+      puts
+      
+      puts "=========================================================="
+      puts "===== ORGANIZATIONS"
+      puts "=========================================================="
+      puts
+      ENV['RAILS_ORG'] = nil
+      db_namespace[:migrate].execute
+      puts
+    end
+    
+    # desc 'Runs the "up" for a given migration VERSION.'
+    task :up => [:environment, :load_config] do
+      version = ENV['VERSION'] ? ENV['VERSION'].to_i : nil
+      raise 'VERSION is required' unless version
+      
+      if ENV['RAILS_ORG'] == 'sessions' || ENV['RAILS_ORG'] == 'all'
+        ActiveRecord::Base.connect_to_sessions
+        ActiveRecord::Migrator.run(:up, ActiveRecord::Migrator.migrations_paths, version)
+        db_namespace['_dump'].invoke
+      end
+      
+      if ENV['RAILS_ORG'] == 'master' || ENV['RAILS_ORG'] == 'all'
+        ActiveRecord::Base.connect_to_master
+        ActiveRecord::Migrator.run(:up, ActiveRecord::Migrator.migrations_paths, version)
+        db_namespace['_dump'].invoke
+      end
+      
+      if ENV['RAILS_ORG'] == 'all' || ![ 'sessions', 'master' ].include?(ENV['RAILS_ORG'])
+        schema_dumped = false
+        MultiDB::Organization.active.each do |org|
+          ActiveRecord::Base.connect_to_organization(org, true)
+          ActiveRecord::Migrator.run(:up, ActiveRecord::Migrator.migrations_paths, version)
+          db_namespace['_dump'].invoke unless schema_dumped
+          schema_dumped = true
+        end
+      end
+    end
+    
+    # desc 'Runs the "down" for a given migration VERSION.'
+    task :down => [:environment, :load_config] do
+      version = ENV['VERSION'] ? ENV['VERSION'].to_i : nil
+      raise 'VERSION is required' unless version
+      
+      if ENV['RAILS_ORG'] == 'sessions'
+        ActiveRecord::Base.connect_to_sessions
+        ActiveRecord::Migrator.run(:down, ActiveRecord::Migrator.migrations_paths, version)
+        db_namespace['_dump'].invoke
+      end
+      
+      if ENV['RAILS_ORG'] == 'master'
+        ActiveRecord::Base.connect_to_master
+        ActiveRecord::Migrator.run(:down, ActiveRecord::Migrator.migrations_paths, version)
+        db_namespace['_dump'].invoke
+      end
+      
+      if ![ 'sessions', 'master' ].include?(ENV['RAILS_ORG'])
+        schema_dumped = false
+        MultiDB::Organization.active.each do |org|
+          ActiveRecord::Base.connect_to_organization(org, true)
+          ActiveRecord::Migrator.run(:down, ActiveRecord::Migrator.migrations_paths, version)
+          db_namespace['_dump'].invoke unless schema_dumped
+          schema_dumped = true
+        end
+      end
+    end
+  end
+  
+  desc "Migrate the database (options: VERSION=x, VERBOSE=false)."
+  task :migrate => [:environment, :load_config] do
+    ActiveRecord::Migration.verbose = ENV["VERBOSE"] ? ENV["VERBOSE"] == "true" : true
+    
+    if ENV['RAILS_ORG'] == 'sessions'
+      ActiveRecord::Base.connect_to_sessions
+      ActiveRecord::Migrator.migrate(ActiveRecord::Migrator.migrations_paths, ENV["VERSION"] ? ENV["VERSION"].to_i : nil) do |migration|
+        ENV["SCOPE"].blank? || (ENV["SCOPE"] == migration.scope)
+      end
+      db_namespace['_dump'].invoke
+    end
+    
+    if ENV['RAILS_ORG'] == 'master'
+      ActiveRecord::Base.connect_to_master
+      ActiveRecord::Migrator.migrate(ActiveRecord::Migrator.migrations_paths, ENV["VERSION"] ? ENV["VERSION"].to_i : nil) do |migration|
+        ENV["SCOPE"].blank? || (ENV["SCOPE"] == migration.scope)
+      end
+      db_namespace['_dump'].invoke
+    end
+    
+    if ![ 'sessions', 'master' ].include?(ENV['RAILS_ORG'])
+      schema_dumped = false
+      MultiDB::Organization.active.each do |org|
+        puts "== Migrating: #{org.code}"
+        ActiveRecord::Base.connect_to_organization(org, true)
+        ActiveRecord::Migrator.migrate(ActiveRecord::Migrator.migrations_paths, ENV["VERSION"] ? ENV["VERSION"].to_i : nil) do |migration|
+          ENV["SCOPE"].blank? || (ENV["SCOPE"] == migration.scope)
+        end
+        db_namespace['_dump'].invoke unless schema_dumped
+        schema_dumped = true
+      end
+    end
+  end
+  
+  desc 'Rolls the schema back to the previous version (specify steps w/ STEP=n).'
+  task :rollback => [:environment, :load_config] do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    
+    if ENV['RAILS_ORG'] == 'sessions'
+      ActiveRecord::Base.connect_to_sessions
+      ActiveRecord::Migrator.rollback(ActiveRecord::Migrator.migrations_paths, step)
+      db_namespace['_dump'].invoke
+    end
+    
+    if ENV['RAILS_ORG'] == 'master'
+      ActiveRecord::Base.connect_to_master
+      ActiveRecord::Migrator.rollback(ActiveRecord::Migrator.migrations_paths, step)
+      db_namespace['_dump'].invoke
+    end
+    
+    if ![ 'sessions', 'master' ].include?(ENV['RAILS_ORG'])
+      schema_dumped = false
+      MultiDB::Organization.active.each do |org|
+        puts "== Rollback: #{org.code}"
+        ActiveRecord::Base.connect_to_organization(org, true)
+        ActiveRecord::Migrator.rollback(ActiveRecord::Migrator.migrations_paths, step)
+        db_namespace['_dump'].invoke unless schema_dumped
+        schema_dumped = true
+      end
+    end
+  end
+  
+  # desc 'Pushes the schema to the next version (specify steps w/ STEP=n).'
+  task :forward => [:environment, :load_config] do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    
+    if ENV['RAILS_ORG'] == 'sessions'
+      ActiveRecord::Base.connect_to_sessions
+      ActiveRecord::Migrator.forward(ActiveRecord::Migrator.migrations_paths, step)
+      db_namespace['_dump'].invoke
+    end
+    
+    if ENV['RAILS_ORG'] == 'master'
+      ActiveRecord::Base.connect_to_master
+      ActiveRecord::Migrator.forward(ActiveRecord::Migrator.migrations_paths, step)
+      db_namespace['_dump'].invoke
+    end
+    
+    if ![ 'sessions', 'master' ].include?(ENV['RAILS_ORG'])
+      schema_dumped = false
+      MultiDB::Organization.active.each do |org|
+        ActiveRecord::Base.connect_to_organization(org, true)
+        ActiveRecord::Migrator.forward(ActiveRecord::Migrator.migrations_paths, step)
+        db_namespace['_dump'].invoke unless schema_dumped
+        schema_dumped = true
+      end
+    end
+  end
+  
+  namespace :schema do
+    desc 'Create a db/schema.rb file that can be portably used against any DB supported by AR'
+    task :dump => [:environment, :load_config] do
+      require 'active_record/schema_dumper'
+      
+      databases_to_dump = case ENV['RAILS_ORG']
+        when nil then [ :sessions, :master, :organization ]
+        when 'sessions' then [ :sessions ]
+        when 'master' then [ :master ]
+        else [ :organization ]
+      end
+      
+      databases_to_dump.each do |db|
+        ActiveRecord::Base.send("connect_to_#{db}")
+        filename = ENV['SCHEMA'] || "#{Rails.root}/db/schema_#{db}.rb"
+        File.open(filename, "w:utf-8") do |file|
+          ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, file)
+        end
+      end
+      db_namespace['schema:dump'].reenable
+    end
+    
+    desc 'Load a schema.rb file into the database'
+    task :load => [:environment, :load_config] do
+      databases_to_load = case ENV['RAILS_ORG']
+        when nil then [ :sessions, :master, :organization ]
+        when 'sessions' then [ :sessions ]
+        when 'master' then [ :master ]
+        else [ :organization ]
+      end
+      
+      databases_to_load.each do |db|
+        ActiveRecord::Base.send("connect_to_#{db}")
+        file = ENV['SCHEMA'] || "#{Rails.root}/db/schema_#{db}.rb"
+        if File.exists?(file)
+          load(file)
+        else
+          abort %{#{file} doesn't exist yet. Run `rake db:migrate` to create it then try again. If you do not intend to use a database, you should instead alter #{Rails.root}/config/application.rb to limit the frameworks that will be loaded}
+        end
+      end
+    end
+  end
+  
+  namespace :test do
+    # desc "Recreate the test database from an existent schema.rb file"
+    task :load_schema => 'db:test:purge' do
+      old_env = Rails.env
+      Rails.env = 'test'
+      ActiveRecord::Schema.verbose = false
+      db_namespace["schema:load"].invoke
+      Rails.env = old_env
+    end
+    
+    # desc "Empty the test databases"
+    task :purge => [:environment, :load_config] do
+      abcs = ActiveRecord::Base.configurations
+      
+      # sessions
+      ActiveRecord::Base.connect_to_sessions
+      ActiveRecord::Base.connection.recreate_database(abcs['test']['database'], mysql_creation_options(abcs['test']))
+      
+      # master
+      ActiveRecord::Base.connect_to_master
+      ActiveRecord::Base.connection.recreate_database(ActiveRecord::Base.master_configuration('test')['database'], mysql_creation_options(abcs['test']))
+      
+      # org
+      ActiveRecord::Base.connect_to_organization
+      ActiveRecord::Base.connection.recreate_database(abcs['test']['database'] + '_org1', mysql_creation_options(abcs['test']))
+    end
+  end
+  
+  alias :create_database_without_multidb :create_database
+  def create_database(_config)
+    is_test = ActiveRecord::Base.configurations.invert[_config] == 'test'
+    config = _config.dup
+    
+    case ENV['RAILS_ORG']
+    when nil
+      # set default org database
+      config['database'] += '_org1'
+    when 'sessions'
+      # do nothing
+    when 'master'
+      config = ActiveRecord::Base.master_configuration(is_test ? 'test' : nil)
+    else
+      config['database'] += '_' + ENV['RAILS_ORG']
+    end
+    
+    create_database_without_multidb(config)
+  end
+end
+
+
+alias :drop_database_without_multidb :drop_database
+def drop_database(_config)
+  is_test = ActiveRecord::Base.configurations.invert[_config] == 'test'
+  config = _config.dup
+  
+  case ENV['RAILS_ORG']
+  when nil
+    # set default org database
+    config['database'] += '_org1'
+  when 'sessions'
+    # do nothing
+  when 'master'
+    config = ActiveRecord::Base.master_configuration(is_test ? 'test' : nil)
+  else
+    config['database'] += '_' + ENV['RAILS_ORG']
+  end
+  
+  drop_database_without_multidb(config)
+end
+
+# not sure yet why environment is not loaded before running create_database/drop_database
+# for now, copied this in from lib/multi_db/active_record_patches.rb
+module ActiveRecord
+  class Base
+    class << self
+      
+      def master_configuration(env = nil)
+        env ||= Rails.env
+        
+        # use master db configuration in config/database.yml if present
+        configurations["master_#{env}"] or Proc.new {
+          c = configurations[env].dup
+          c['database'] += '_master'
+          c
+        }.call
+      end
+      
+    end
+  end
+end