active_record_shards 3.0.0.beta4 → 3.1.0

data/lib/active_record_shards.rb

@@ -10,6 +10,7 @@ require 'active_record_shards/migration'
 require 'active_record_shards/default_slave_patches'
 require 'active_record_shards/connection_handler'
 require 'active_record_shards/connection_specification'
+require 'active_record_shards/schema_dumper_extension'
 
 if ActiveRecord::VERSION::MAJOR >= 4
   methods_to_override = [:establish_connection, :remove_connection, :pool_for,
@@ -37,6 +38,10 @@ end
 
 ActiveRecord::Associations::CollectionProxy.send(:include, ActiveRecordShards::AssociationCollectionConnectionSelection)
 
+if ActiveRecord::VERSION::MAJOR >= 4 && RUBY_VERSION >= '2'
+  ActiveRecord::SchemaDumper.send(:prepend, ActiveRecordShards::SchemaDumperExtension)
+end
+
 module ActiveRecordShards
   def self.rails_env
     env = Rails.env if Object.const_defined?(:Rails)

data/lib/active_record_shards/schema_dumper_extension.rb

@@ -0,0 +1,43 @@
+module ActiveRecordShards
+  module SchemaDumperExtension
+
+    def dump(stream)
+      stream = super(stream)
+      original_connection = @connection
+
+      if ActiveRecord::Base.supports_sharding?
+        ActiveRecord::Base.on_first_shard do
+          @connection = ActiveRecord::Base.connection
+          shard_header(stream)
+          extensions(stream)
+          tables(stream)
+          shard_trailer(stream)
+        end
+      end
+
+      stream
+    ensure
+      @connection = original_connection
+    end
+
+
+    def shard_header(stream)
+      define_params = @version ? "version: #{@version}" : ""
+
+      stream.puts <<HEADER
+
+
+# This section generated by active_record_shards
+
+ActiveRecord::Base.on_all_shards do
+ActiveRecord::Schema.define(#{define_params}) do
+
+HEADER
+      end
+
+    def shard_trailer(stream)
+      stream.puts "end\nend"
+    end
+
+  end
+end

metadata

@@ -1,7 +1,8 @@
 --- !ruby/object:Gem::Specification
 name: active_record_shards
 version: !ruby/object:Gem::Version
-  version: 3.0.0.beta4
+  version: 3.1.0
+  prerelease: 
 platform: ruby
 authors:
 - Mick Staugaard
@@ -10,26 +11,28 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-05 00:00:00.000000000 Z
+date: 2014-07-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
   requirement: !ruby/object:Gem::Requirement
+    none: false
     requirements:
-    - - ">="
+    - - ! '>='
       - !ruby/object:Gem::Version
         version: 3.2.16
-    - - "<="
+    - - <=
       - !ruby/object:Gem::Version
         version: '4.1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
+    none: false
     requirements:
-    - - ">="
+    - - ! '>='
       - !ruby/object:Gem::Version
         version: 3.2.16
-    - - "<="
+    - - <=
       - !ruby/object:Gem::Version
         version: '4.1'
 description: Easily run queries on shard and slave databases.
@@ -41,8 +44,6 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- README.md
-- lib/active_record_shards.rb
 - lib/active_record_shards/association_collection_connection_selection.rb
 - lib/active_record_shards/configuration_parser.rb
 - lib/active_record_shards/connection_handler.rb
@@ -52,30 +53,34 @@ files:
 - lib/active_record_shards/default_slave_patches.rb
 - lib/active_record_shards/migration.rb
 - lib/active_record_shards/model.rb
+- lib/active_record_shards/schema_dumper_extension.rb
 - lib/active_record_shards/shard_selection.rb
 - lib/active_record_shards/tasks.rb
+- lib/active_record_shards.rb
+- README.md
 homepage: https://github.com/zendesk/active_record_shards
 licenses:
 - MIT
-metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
   requirements:
-  - - ">="
+  - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
   requirements:
-  - - ">"
+  - - ! '>='
     - !ruby/object:Gem::Version
-      version: 1.3.1
+      version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 1.8.23.2
 signing_key: 
-specification_version: 4
+specification_version: 3
 summary: Simple database switching for ActiveRecord.
 test_files: []

checksums.yaml

@@ -1,7 +0,0 @@
----
-SHA1:
-  metadata.gz: 6e2b0d1263e6c2ba07f208b1c46f252e254cd341
-  data.tar.gz: b5e951385e67395baaf4e6408441fda6350afc1f
-SHA512:
-  metadata.gz: f194e8c0726304edff05e03f501b8edf42dff62ea28bdbe90293b93d10cae94499471934387401682dcb7b69f63f5c4eeb428b14ea91da11d3c54a78d964a4e5
-  data.tar.gz: c87fd04a0217d32e9399f58b9f45d6f9e483444e21285c0d1ad29bf4baad3c6c165d988c3b3fd7800db7a950f63efa8a4f0d829a9c13290fec381230573661bc