switchman 0.0.1

data/lib/switchman/shackles.rb

@@ -0,0 +1,34 @@
+module Switchman
+  module Shackles
+    module ClassMethods
+      def ensure_handler
+        Shard.default.activate(*Shard.categories) do
+          new_handler = @connection_handlers[self.environment]
+          if !new_handler
+            new_handler = @connection_handlers[self.environment] = ::ActiveRecord::ConnectionAdapters::ConnectionHandler.new
+            pools = ::ActiveRecord::Base.connection_handler.instance_variable_get(:@class_to_pool)
+            pools.each do |model, pool|
+              # don't call establish_connection for pools created just for different sharding categories
+              if model != ::ActiveRecord::Base.name
+                klass = model.constantize
+                next if ::ActiveRecord::Base.connection_handler.retrieve_connection_pool(klass.superclass).default_pool == pools[model].default_pool
+              end
+
+              new_handler.establish_connection(model, pool.spec)
+            end
+          end
+          # make sure it picks up the environment change
+          new_handler.connection_pools.each do |_, pool|
+            pool.spec.instance_variable_set(:@current_config, nil)
+          end
+          new_handler
+        end
+      end
+    end
+
+    def self.included(klass)
+      klass.extend(ClassMethods)
+      klass.singleton_class.send(:remove_method, :ensure_handler)
+    end
+  end
+end

data/lib/switchman/test_helper.rb

@@ -0,0 +1,65 @@
+module Switchman
+  module TestHelper
+    class << self
+      def recreate_persistent_test_shards
+        # recreate the default shard (it got buhleted)
+        if Shard.default.is_a?(DefaultShard)
+          begin
+            Shard.create!(:default => true)
+          rescue
+            # database doesn't exist yet, presumably
+            [nil, nil]
+          end
+          Shard.default(true)
+        end
+
+        # can't auto-create a new shard on the default shard's db server if the
+        # default shard is split across multiple db servers
+        if ::ActiveRecord::Base.connection_handler.connection_pools.length > 1
+          server1 = DatabaseServer.create(:config => Shard.default.database_server.config)
+        else
+          server1 = Shard.default.database_server
+        end
+        server2 = DatabaseServer.create(:config => Shard.default.database_server.config)
+
+        if server1 == Shard.default.database_server && server1.config[:shard1] && server1.config[:shard2]
+          # look for the shards in the db already
+          shard1 = find_existing_test_shard(server1, server1.config[:shard1])
+          shard2 = find_existing_test_shard(server2, server1.config[:shard2])
+
+          shard1 ||= server1.shards.build
+          shard1.name = server1.config[:shard1]
+          shard1.save! if shard1.changed?
+          shard2 ||= server2.shards.build
+          shard2.name = server1.config[:shard2]
+          shard2.save! if shard2.changed?
+
+          recreate_shards = shard1.activate { ::ActiveRecord::Base.connection.tables == [] }
+          if recreate_shards
+            shard1.drop_database rescue nil
+            shard1.destroy
+            shard2.drop_database rescue nil
+            shard2.destroy
+            shard1 = server1.create_new_shard(:db_name => server1.config[:shard1])
+            shard2 = server2.create_new_shard(:db_name => server1.config[:shard2])
+          end
+          [shard1, shard2]
+        else
+          [server1, server2]
+        end
+      end
+
+      private
+      def find_existing_test_shard(server, name)
+        if server == Shard.default.database_server
+          server.shards.find_by_database_name(name)
+        else
+          shard = Shard.where("database_server_id IS NOT NULL AND database_name=?", name).first
+          # if somehow databases got created in a different order, change the shard to match
+          shard.database_server = server if shard
+          shard
+        end
+      end
+    end
+  end
+end

data/lib/switchman/version.rb

@@ -0,0 +1,3 @@
+module Switchman
+  VERSION = "0.0.1"
+end

data/spec/dummy/Rakefile

@@ -0,0 +1,7 @@
+#!/usr/bin/env rake
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+
+Dummy::Application.load_tasks

data/spec/dummy/app/models/appendage.rb

@@ -0,0 +1,24 @@
+class Appendage < ActiveRecord::Base
+  attr_accessible :user, :user_id, :value
+
+  belongs_to :user
+  has_many :digits
+
+  has_many :features, :as => :owner
+
+  scope :has_no_value, where(:value => nil)
+  scope :has_value, where("appendages.value IS NOT NULL")
+
+  attr_writer :associated_shards
+  class << self
+    attr_accessor :associated_shards_map
+  end
+
+  def self.associated_shards_for(global_id)
+    associated_shards_map[global_id] if global_id && associated_shards_map
+  end
+
+  def associated_shards
+    self.class.associated_shards_for(self.global_id) || @associated_shards || [self.shard]
+  end
+end

data/spec/dummy/app/models/digit.rb

@@ -0,0 +1,9 @@
+class Digit < ActiveRecord::Base
+  attr_accessible :appendage, :appendage_id, :value
+
+  belongs_to :appendage
+  has_one :user, :through => :appendage
+
+  scope :has_no_value, where(:value => nil)
+  scope :has_value, where("digits.value IS NOT NULL")
+end

data/spec/dummy/app/models/feature.rb

@@ -0,0 +1,5 @@
+class Feature < ActiveRecord::Base
+  belongs_to :owner, :polymorphic => true
+
+  attr_accessible :owner, :value
+end

data/spec/dummy/app/models/mirror_user.rb

@@ -0,0 +1,5 @@
+class MirrorUser < ActiveRecord::Base
+  self.shard_category = :mirror_universe
+
+  has_one :user
+end

data/spec/dummy/app/models/user.rb

@@ -0,0 +1,23 @@
+class User < ActiveRecord::Base
+  attr_accessible :name
+
+  has_many :appendages, :multishard => true
+  has_many :digits, :through => :appendages, :multishard => true
+
+  has_many :features, :as => :owner, :multishard => true
+
+  belongs_to :mirror_user
+
+  attr_writer :associated_shards
+  class << self
+    attr_accessor :associated_shards_map
+  end
+
+  def self.associated_shards_for(global_id)
+    associated_shards_map[global_id] if global_id && associated_shards_map
+  end
+
+  def associated_shards
+    self.class.associated_shards_for(self.global_id) || @associated_shards || [self.shard]
+  end
+end

data/spec/dummy/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment',  __FILE__)
+run Dummy::Application

data/spec/dummy/config/application.rb

@@ -0,0 +1,59 @@
+require File.expand_path('../boot', __FILE__)
+
+require 'rails/all'
+
+Bundler.require(*Rails.groups)
+require "switchman"
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Custom directories with classes and modules you want to be autoloadable.
+    # config.autoload_paths += %W(#{config.root}/extras)
+
+    # Only load the plugins named here, in the order given (default is alphabetical).
+    # :all can be used as a placeholder for all plugins not explicitly named.
+    # config.plugins = [ :exception_notification, :ssl_requirement, :all ]
+
+    # Activate observers that should always be running.
+    # config.active_record.observers = :cacher, :garbage_collector, :forum_observer
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # Configure the default encoding used in templates for Ruby 1.9.
+    config.encoding = "utf-8"
+
+    # Configure sensitive parameters which will be filtered from the log file.
+    config.filter_parameters += [:password]
+
+    # Enable escaping HTML in JSON.
+    config.active_support.escape_html_entities_in_json = true
+
+    # Use SQL instead of Active Record's schema dumper when creating the database.
+    # This is necessary if your schema can't be completely dumped by the schema dumper,
+    # like if you have constraints or database-specific column types
+    # config.active_record.schema_format = :sql
+
+    # Enforce whitelist mode for mass assignment.
+    # This will create an empty whitelist of attributes available for mass-assignment for all models
+    # in your app. As such, your models will need to explicitly whitelist or blacklist accessible
+    # parameters by using an attr_accessible or attr_protected declaration.
+    config.active_record.whitelist_attributes = true
+
+    # Enable the asset pipeline
+    config.assets.enabled = true
+
+    # Version of your assets, change this if you want to expire all your assets
+    config.assets.version = '1.0'
+  end
+end
+

data/spec/dummy/config/boot.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+gemfile = File.expand_path('../../../../Gemfile', __FILE__)
+
+if File.exist?(gemfile)
+  ENV['BUNDLE_GEMFILE'] = gemfile
+  require 'bundler'
+  Bundler.setup
+end
+
+$:.unshift File.expand_path('../../../../lib', __FILE__)

data/spec/dummy/config/database.yml

@@ -0,0 +1,17 @@
+# SQLite version 3.x
+#   gem install sqlite3
+#
+#   Ensure the SQLite 3 gem is defined in your Gemfile
+#   gem 'sqlite3'
+development:
+  adapter: postgresql
+  database: switchman_development
+  port: 15432
+
+# Warning: The database defined as "test" will be erased and
+# re-generated from your development database when you run "rake".
+# Do not set this db to the same as development or production.
+test:
+  adapter: postgresql
+  database: switchman_test
+  port: 15432

data/spec/dummy/config/database.yml.example

@@ -0,0 +1,25 @@
+# SQLite version 3.x
+#   gem install sqlite3
+#
+#   Ensure the SQLite 3 gem is defined in your Gemfile
+#   gem 'sqlite3'
+development:
+  adapter: sqlite3
+  database: db/development.sqlite3
+  pool: 5
+  timeout: 5000
+
+# Warning: The database defined as "test" will be erased and
+# re-generated from your development database when you run "rake".
+# Do not set this db to the same as development or production.
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+  timeout: 5000
+
+production:
+  adapter: sqlite3
+  database: db/production.sqlite3
+  pool: 5
+  timeout: 5000

data/spec/dummy/config/environment.rb

@@ -0,0 +1,5 @@
+# Load the rails application
+require File.expand_path('../application', __FILE__)
+
+# Initialize the rails application
+Dummy::Application.initialize!

data/spec/dummy/config/environments/development.rb

@@ -0,0 +1,37 @@
+Dummy::Application.configure do
+  # Settings specified here will take precedence over those in config/application.rb
+
+  # In the development environment your application's code is reloaded on
+  # every request. This slows down response time but is perfect for development
+  # since you don't have to restart the web server when you make code changes.
+  config.cache_classes = false
+
+  # Log error messages when you accidentally call methods on nil.
+  config.whiny_nils = true
+
+  # Show full error reports and disable caching
+  config.consider_all_requests_local       = true
+  config.action_controller.perform_caching = false
+
+  # Don't care if the mailer can't send
+  config.action_mailer.raise_delivery_errors = false
+
+  # Print deprecation notices to the Rails logger
+  config.active_support.deprecation = :log
+
+  # Only use best-standards-support built into browsers
+  config.action_dispatch.best_standards_support = :builtin
+
+  # Raise exception on mass assignment protection for Active Record models
+  config.active_record.mass_assignment_sanitizer = :strict
+
+  # Log the query plan for queries taking more than this (works
+  # with SQLite, MySQL, and PostgreSQL)
+  config.active_record.auto_explain_threshold_in_seconds = 0.5
+
+  # Do not compress assets
+  config.assets.compress = false
+
+  # Expands the lines which load the assets
+  config.assets.debug = true
+end

data/spec/dummy/config/environments/production.rb

@@ -0,0 +1,67 @@
+Dummy::Application.configure do
+  # Settings specified here will take precedence over those in config/application.rb
+
+  # Code is not reloaded between requests
+  config.cache_classes = true
+
+  # Full error reports are disabled and caching is turned on
+  config.consider_all_requests_local       = false
+  config.action_controller.perform_caching = true
+
+  # Disable Rails's static asset server (Apache or nginx will already do this)
+  config.serve_static_assets = false
+
+  # Compress JavaScripts and CSS
+  config.assets.compress = true
+
+  # Don't fallback to assets pipeline if a precompiled asset is missed
+  config.assets.compile = false
+
+  # Generate digests for assets URLs
+  config.assets.digest = true
+
+  # Defaults to nil and saved in location specified by config.assets.prefix
+  # config.assets.manifest = YOUR_PATH
+
+  # Specifies the header that your server uses for sending files
+  # config.action_dispatch.x_sendfile_header = "X-Sendfile" # for apache
+  # config.action_dispatch.x_sendfile_header = 'X-Accel-Redirect' # for nginx
+
+  # Force all access to the app over SSL, use Strict-Transport-Security, and use secure cookies.
+  # config.force_ssl = true
+
+  # See everything in the log (default is :info)
+  # config.log_level = :debug
+
+  # Prepend all log lines with the following tags
+  # config.log_tags = [ :subdomain, :uuid ]
+
+  # Use a different logger for distributed setups
+  # config.logger = ActiveSupport::TaggedLogging.new(SyslogLogger.new)
+
+  # Use a different cache store in production
+  # config.cache_store = :mem_cache_store
+
+  # Enable serving of images, stylesheets, and JavaScripts from an asset server
+  # config.action_controller.asset_host = "http://assets.example.com"
+
+  # Precompile additional assets (application.js, application.css, and all non-JS/CSS are already added)
+  # config.assets.precompile += %w( search.js )
+
+  # Disable delivery errors, bad email addresses will be ignored
+  # config.action_mailer.raise_delivery_errors = false
+
+  # Enable threaded mode
+  # config.threadsafe!
+
+  # Enable locale fallbacks for I18n (makes lookups for any locale fall back to
+  # the I18n.default_locale when a translation can not be found)
+  config.i18n.fallbacks = true
+
+  # Send deprecation notices to registered listeners
+  config.active_support.deprecation = :notify
+
+  # Log the query plan for queries taking more than this (works
+  # with SQLite, MySQL, and PostgreSQL)
+  # config.active_record.auto_explain_threshold_in_seconds = 0.5
+end

data/spec/dummy/config/environments/test.rb

@@ -0,0 +1,37 @@
+Dummy::Application.configure do
+  # Settings specified here will take precedence over those in config/application.rb
+
+  # The test environment is used exclusively to run your application's
+  # test suite. You never need to work with it otherwise. Remember that
+  # your test database is "scratch space" for the test suite and is wiped
+  # and recreated between test runs. Don't rely on the data there!
+  config.cache_classes = true
+
+  # Configure static asset server for tests with Cache-Control for performance
+  config.serve_static_assets = true
+  config.static_cache_control = "public, max-age=3600"
+
+  # Log error messages when you accidentally call methods on nil
+  config.whiny_nils = true
+
+  # Show full error reports and disable caching
+  config.consider_all_requests_local       = true
+  config.action_controller.perform_caching = false
+
+  # Raise exceptions instead of rendering exception templates
+  config.action_dispatch.show_exceptions = false
+
+  # Disable request forgery protection in test environment
+  config.action_controller.allow_forgery_protection    = false
+
+  # Tell Action Mailer not to deliver emails to the real world.
+  # The :test delivery method accumulates sent emails in the
+  # ActionMailer::Base.deliveries array.
+  config.action_mailer.delivery_method = :test
+
+  # Raise exception on mass assignment protection for Active Record models
+  config.active_record.mass_assignment_sanitizer = :strict
+
+  # Print deprecation notices to the stderr
+  config.active_support.deprecation = :stderr
+end