dynashard 0.1.0

data/spec/support/models.rb

@@ -0,0 +1,99 @@
+class Shard < ActiveRecord::Base
+  def dsn
+    HashWithIndifferentAccess.new(self.attributes.reject{|k,v| v.nil? || ['name','id'].include?(k)})
+  end
+end
+
+# Owner class that doesn't shard its associations
+class NonShardingOwner < ActiveRecord::Base
+  # Non-sharded associations
+  has_one :non_sharded_has_one
+  has_many :non_sharded_has_manys
+  has_many :non_sharded_joins
+  has_many :non_sharded_has_many_throughs, :through => :non_sharded_joins
+
+  # Sharded associations
+  has_one :sharded_has_one
+  has_many :sharded_has_manys
+  has_many :sharded_joins
+  has_many :sharded_has_many_throughs, :through => :sharded_joins
+end
+
+# Owner class for sharded associations
+class ShardingOwner < ActiveRecord::Base
+  shard :associated, :using => :shard_dsn
+
+  belongs_to :shard
+  def shard_dsn
+    @dsn ||= shard.dsn
+  end
+
+  # Non-sharded associations
+  has_one :non_sharded_has_one
+  has_many :non_sharded_has_manys
+  has_many :non_sharded_joins
+  has_many :non_sharded_has_many_throughs, :through => :non_sharded_joins
+
+  # Sharded associations
+  has_one :sharded_has_one
+  has_many :sharded_has_manys
+  has_many :sharded_joins
+  has_many :sharded_has_many_throughs, :through => :sharded_joins
+end
+
+# Non-sharded has_one association class
+class NonShardedHasOne < ActiveRecord::Base
+  belongs_to :sharding_owner
+
+  validates :name, :uniqueness => true
+end
+
+# Non-sharded has_many association class
+class NonShardedHasMany < ActiveRecord::Base
+  belongs_to :sharding_owner
+  belongs_to :non_sharding_owner
+end
+
+# Join table for has_many :through
+class NonShardedJoin < ActiveRecord::Base
+  belongs_to :sharding_owner
+  belongs_to :non_sharded_has_many_through
+end
+
+# Non-sharded has_many association class
+class NonShardedHasManyThrough < ActiveRecord::Base
+  has_many :non_sharded_joins
+end
+
+# Sharded has_one association class
+class ShardedHasOne < ActiveRecord::Base
+  shard :by => :owner
+
+  validates :name, :uniqueness => true
+
+  belongs_to :sharding_owner
+  belongs_to :non_sharding_owner
+end
+
+# Sharded has_many association class
+class ShardedHasMany < ActiveRecord::Base
+  shard :by => :owner
+
+  belongs_to :sharding_owner
+  belongs_to :non_sharding_owner
+end
+
+# Join table for has_many :through
+class ShardedJoin < ActiveRecord::Base
+  shard :by => :owner
+
+  belongs_to :sharding_owner
+  belongs_to :sharded_has_many_through
+end
+
+# Sharded has_many :through association class
+class ShardedHasManyThrough < ActiveRecord::Base
+  shard :by => :owner
+
+  has_many :sharded_joins
+end

data/spec/validation_spec.rb

@@ -0,0 +1,101 @@
+require 'spec_helper'
+
+describe 'Dynashard::ValidationExtensions' do
+  before(:each) do
+    @owner = Factory(:sharding_owner)
+  end
+
+  describe '#find_finder_class_for' do
+    before(:each) do
+      @shard_klass = Dynashard.class_for(@owner.shard_dsn)
+      @sharded_model_subclass = Dynashard.sharded_model_class(@shard_klass, ShardedHasOne)
+    end
+
+    context 'when sharding' do
+      before(:each) do
+        @validator = ShardedHasOne.validators.detect{|v| v.is_a?(ActiveRecord::Validations::UniquenessValidator)}
+      end
+
+      context 'with a generated association owner class' do
+        it 'returns the generated class' do
+          new_record = @owner.build_sharded_has_one
+          @validator.send(:find_finder_class_for, new_record).should == @sharded_model_subclass
+        end
+      end
+
+      context 'with a model class' do
+        it 'returns the generated shard subclass' do
+          Dynashard.with_context(:owner => @owner.shard_dsn) do
+            new_record = ShardedHasOne.new
+            @validator.send(:find_finder_class_for, new_record).should == @sharded_model_subclass
+          end
+        end
+      end
+    end
+
+    context 'when not sharding' do
+      before(:each) do
+        @validator = NonShardedHasOne.validators.detect{|v| v.is_a?(ActiveRecord::Validations::UniquenessValidator)}
+      end
+
+      context 'with a non-sharding association owner class' do
+        it 'returns the model class' do
+          non_sharding_owner = Factory(:non_sharding_owner)
+          new_record = non_sharding_owner.build_non_sharded_has_one
+          @validator.send(:find_finder_class_for, new_record).should == NonShardedHasOne
+        end
+      end
+
+      context 'with a non-sharding model class' do
+        it 'returns the model class' do
+          new_record = NonShardedHasOne.new
+          @validator.send(:find_finder_class_for, new_record).should == NonShardedHasOne
+        end
+      end
+    end
+  end
+
+  context 'when validating uniqueness' do
+    context 'with an association proxy target' do
+      context 'with a conflicting record on the shard' do
+        it 'returns invalid' do
+          conflicting_record = Dynashard.with_context(:owner => @owner.shard_dsn){Factory(:sharded_has_one)}
+          new_record = @owner.build_sharded_has_one(:name => conflicting_record.name)
+          new_record.should_not be_valid
+        end
+      end
+
+      context 'with a conflicting record on a different shard' do
+        it 'returns valid' do
+          other_shard = Shard.find(:all).detect{|shard| shard != @owner.shard}
+          non_conflicting_record = Dynashard.with_context(:owner => other_shard.dsn){Factory(:sharded_has_one)}
+          new_record = @owner.build_sharded_has_one(:name => non_conflicting_record.name)
+          new_record.should be_valid
+        end
+      end
+    end
+
+    context 'with a model class' do
+      context 'with a conflicting record on the shard' do
+        it 'returns invalid' do
+          conflicting_record = Dynashard.with_context(:owner => @owner.shard_dsn){Factory(:sharded_has_one)}
+          new_record = ShardedHasOne.new(:name => conflicting_record.name)
+          Dynashard.with_context(:owner => @owner.shard_dsn) do
+            new_record.should_not be_valid
+          end
+        end
+      end
+
+      context 'with a conflicting record on a different shard' do
+        it 'returns valid' do
+          other_shard = Shard.find(:all).detect{|shard| shard != @owner.shard}
+          non_conflicting_record = Dynashard.with_context(:owner => other_shard.dsn){Factory(:sharded_has_one)}
+          new_record = ShardedHasOne.new(:name => non_conflicting_record.name)
+          Dynashard.with_context(:owner => @owner.shard_dsn) do
+            new_record.should be_valid
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,188 @@
+--- !ruby/object:Gem::Specification 
+name: dynashard
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Nick Hengeveld
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-01-31 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: activerecord
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        version: "3.0"
+  requirement: *id001
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: shoulda
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id002
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: bundler
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  requirement: *id003
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: jeweler
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 1
+        - 5
+        - 2
+        version: 1.5.2
+  requirement: *id004
+- !ruby/object:Gem::Dependency 
+  type: :development
+  prerelease: false
+  name: rcov
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id005
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  prerelease: false
+  name: activerecord
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        version: "3.0"
+  requirement: *id006
+description: Dynashard allows you to shard your ActiveRecord models.  Models can be configured to shard based on context that can be defined dynamically.
+email: nickh@verticalresponse.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.md
+files: 
+- .document
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- lib/dynashard.rb
+- lib/dynashard/arel_sql_engine.rb
+- lib/dynashard/associations.rb
+- lib/dynashard/connection_handler.rb
+- lib/dynashard/model.rb
+- lib/dynashard/validations.rb
+- spec/arel_sql_engine_spec.rb
+- spec/associations_spec.rb
+- spec/connection_handler_spec.rb
+- spec/dynashard_spec.rb
+- spec/model_spec.rb
+- spec/spec_helper.rb
+- spec/support/factories.rb
+- spec/support/models.rb
+- spec/validation_spec.rb
+- spec/db/schema.rb
+has_rdoc: true
+homepage: http://github.com/nickh/dynashard
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Dynamic sharding for ActiveRecord
+test_files: 
+- spec/arel_sql_engine_spec.rb
+- spec/associations_spec.rb
+- spec/connection_handler_spec.rb
+- spec/db/schema.rb
+- spec/dynashard_spec.rb
+- spec/model_spec.rb
+- spec/spec_helper.rb
+- spec/support/factories.rb
+- spec/support/models.rb
+- spec/validation_spec.rb