active_shard 0.2.4 → 0.2.5

data/lib/active_shard.rb

@@ -9,6 +9,7 @@ module ActiveShard
   autoload :ShardCollection
   autoload :ShardDefinition
   autoload :ShardLookupHandler
+  autoload :TestFixtures
 
   class << self
 
@@ -174,6 +175,10 @@ module ActiveShard
       config.shards_by_schema( environment, schema_name )
     end
 
+    def schemas
+      config.schemas( environment )
+    end
+
     def logger
       @logger
     end

data/lib/active_shard/active_record/connection_handler.rb

@@ -81,6 +81,10 @@ module ActiveShard
 
         active_shard_name = shard_lookup.lookup_active_shard( schema_name )
         
+        connection_pool schema_name, active_shard_name
+      end
+
+      def connection_pool( schema_name, active_shard_name = nil )
         ( active_shard_name.nil? ?
             get_schema_pool_for( schema_name ) :
             connection_pools[ connection_pool_id( schema_name, active_shard_name ) ] )

data/lib/active_shard/config.rb

@@ -16,6 +16,10 @@ module ActiveShard
       shard_collection( environment ).by_schema( schema )
     end
 
+    def schemas( environment )
+      shard_collection( environment ).schemas
+    end
+
     def add_shard( environment, shard_definition )
       shard_collection( environment ).add_shard( shard_definition )
     end
@@ -45,4 +49,4 @@ module ActiveShard
 
   end
   
-end
+end

data/lib/active_shard/database_cleaner.rb

@@ -0,0 +1,3 @@
+require 'active_shard/database_cleaner/base'
+require 'active_shard/database_cleaner/truncation'
+require 'active_shard/database_cleaner/transaction'

data/lib/active_shard/database_cleaner/base.rb

@@ -0,0 +1,26 @@
+require 'database_cleaner/active_record/base'
+
+module DatabaseCleaner
+  module ActiveShard
+    def self.acailable_strategies
+      %w[truncation transaction]
+    end
+
+    module Base
+      include ::DatabaseCleaner::Generic::Base
+
+      private
+      def connection(schema, shard)
+        ::ActiveRecord::Base.connection_handler.connection_pool(schema, shard).connection
+      end
+
+      def for_each_shard
+        ::ActiveShard.schemas.each do |schema|
+          ::ActiveShard.shards_by_schema(schema).each do |shard|
+            yield connection(schema, shard.name)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_shard/database_cleaner/transaction.rb

@@ -0,0 +1,23 @@
+require 'database_cleaner/active_record/transaction'
+
+module DatabaseCleaner::ActiveShard
+  class Transaction
+    include ::DatabaseCleaner::ActiveShard::Base
+
+    def start
+      for_each_shard do |c|
+        c.increment_open_transactions
+        c.begin_db_transaction
+      end
+    end
+
+    def clean
+      for_each_shard do |c|
+        if c.open_transactions > 0
+          c.rollback_db_transaction
+          c.decrement_open_transactions
+        end
+      end
+    end
+  end
+end

data/lib/active_shard/database_cleaner/truncation.rb

@@ -0,0 +1,26 @@
+require 'database_cleaner/active_record/truncation'
+
+module DatabaseCleaner::ActiveShard
+  class Truncation
+    include ::DatabaseCleaner::ActiveShard::Base
+    include ::DatabaseCleaner::Generic::Truncation
+
+    def clean
+      for_each_shard do |c|
+        c.disable_referential_integrity do
+          c.truncate_tables(tables_to_truncate(c))
+        end
+      end
+    end
+
+    private
+    def tables_to_truncate(connection)
+      (@only || connection.tables) - @tables_to_exclude - connection.views
+    end
+
+    # overwritten
+    def migration_storage_name
+      'schema_migrations'
+    end
+  end
+end

data/lib/active_shard/fixtures.rb

@@ -0,0 +1,134 @@
+require 'active_record/fixtures'
+module ActiveShard
+  class Fixtures
+    class << self
+      def create_fixtures(fixtures_dir, *args)
+        schemas = ActiveShard.schemas.map(&:to_sym) & Dir.entries(fixtures_dir).map(&:to_sym)
+
+        self.by_schema.merge!(
+          schemas.inject({}) do |hash, schema|
+            hash[schema] = Schema.new(schema, File.join(fixtures_dir, schema.to_s), *args)
+            hash
+          end
+        )
+      end
+
+      def reset!
+        schemas.each(&:reset!)
+
+        @by_schema = {}
+      end
+
+      def by_schema
+        @by_schema ||= {}
+      end
+
+      def schemas
+        by_schema.values
+      end
+
+      def [](key)
+        by_schema[key]
+      end
+
+      def setup_fixture_accessors obj
+        schemas.each do |schema|
+          schema.setup_fixture_accessors obj
+        end
+      end
+    end
+
+    class Schema
+      attr_reader :schema
+
+      def initialize(schema, schema_fixtures_dir, *args)
+        @schema = schema
+
+        shards = ActiveShard.shards_by_schema(schema).map{|s| s.name.to_sym} & Dir.entries(schema_fixtures_dir).map(&:to_sym)
+
+        self.by_shard.merge!(
+          shards.inject({}) do |hash, shard|
+            hash[shard] = Shard.new(schema, shard, File.join(schema_fixtures_dir, shard.to_s), *args)
+            hash
+          end
+        )
+      end
+
+      def reset!
+        shards.each(&:reset!)
+      end
+
+      def by_shard
+        @by_shard ||= {}
+      end
+
+      def shards
+        by_shard.values
+      end
+
+      def [](key)
+        by_shard[key]
+      end
+
+      def setup_fixture_accessors obj
+        shards.each do |shard|
+          shard.setup_fixture_accessors obj
+        end
+      end
+    end
+
+    class Shard
+      attr_reader :schema
+      attr_reader :shard
+      attr_reader :fixture_names
+      attr_reader :fixtures
+
+      def initialize(schema, shard, shard_fixtures_dir, *args)
+        @schema = schema
+        @shard = shard
+
+        @fixtures_dir = shard_fixtures_dir
+        @fixture_names = Dir[File.join(@fixtures_dir, '*.yml')].map {|f| File.basename(f, '.yml') }
+
+        @fixtures = ::Fixtures.create_fixtures(@fixtures_dir, @fixture_names, *args) do
+          connection
+        end
+      end
+
+      def connection
+        @connection ||= ::ActiveRecord::Base.connection_handler.connection_pool(@schema,@shard).connection
+      end
+
+      def setup_fixture_accessors obj
+        fixture_names = self.fixture_names
+        loaded_fixtures = ::Fixtures.cache_for_connection(connection)
+
+        fixture_names.each do |table_name|
+          table_name = table_name.to_s.tr('./', '_')
+
+          obj.class.send(:define_method, table_name) do |*fixtures|
+            force_reload = fixtures.pop if fixtures.last == true || fixtures.last == :reload
+
+            @fixture_cache ||= {}
+            @fixture_cache[table_name] ||= {}
+
+            instances = fixtures.map do |fixture|
+              @fixture_cache[table_name].delete(fixture) if force_reload
+              if loaded_fixtures[table_name][fixture.to_s]
+                @fixture_cache[table_name][fixture] ||= loaded_fixtures[table_name][fixture.to_s].find
+              else
+                raise StandardError, "No fixture with name '#{fixture}' found for table '#{table_name}'"
+              end
+            end
+
+            instances.size == 1 ? instances.first : instances
+          end
+        end
+      end
+
+      def reset!
+        ::Fixtures.reset_cache connection
+      end
+    end
+  end
+end

data/lib/active_shard/version.rb

@@ -1,3 +1,3 @@
 module ActiveShard
-  VERSION = '0.2.4'
+  VERSION = '0.2.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_shard
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.2.5
   prerelease: 
 platform: ruby
 authors:
@@ -10,11 +10,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-12-28 00:00:00.000000000Z
+date: 2012-01-10 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
-  requirement: &85891990 !ruby/object:Gem::Requirement
+  requirement: &86568210 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -22,7 +22,7 @@ dependencies:
         version: 3.0.0
   type: :runtime
   prerelease: false
-  version_requirements: *85891990
+  version_requirements: *86568210
 description: ActiveShard is a library that implements flexible sharding in ActiveRecord
   and Rails.
 email:
@@ -35,12 +35,17 @@ files:
 - README.md
 - lib/active_shard/version.rb
 - lib/active_shard/shard_definition.rb
+- lib/active_shard/database_cleaner.rb
 - lib/active_shard/active_record.rb
 - lib/active_shard/railtie.rb
 - lib/active_shard/scope_manager.rb
 - lib/active_shard/config.rb
 - lib/active_shard/shard_collection.rb
+- lib/active_shard/database_cleaner/truncation.rb
+- lib/active_shard/database_cleaner/transaction.rb
+- lib/active_shard/database_cleaner/base.rb
 - lib/active_shard/rails/database.rake
+- lib/active_shard/fixtures.rb
 - lib/active_shard/active_record/schema_connection_proxy.rb
 - lib/active_shard/active_record/sharded_base.rb
 - lib/active_shard/active_record/connection_specification_adapter.rb