switchman 2.2.2 → 3.0.0

data/lib/tasks/switchman.rake

@@ -1,20 +1,22 @@
+# frozen_string_literal: true
+
 module Switchman
   module Rake
     def self.filter_database_servers(&block)
       chain = filter_database_servers_chain # use a local variable so that the current chain is closed over in the following lambda
-      @filter_database_servers_chain = lambda { |servers| block.call(servers, chain) }
+      @filter_database_servers_chain = ->(servers) { block.call(servers, chain) }
     end
 
     def self.scope(base_scope = Shard,
-      database_server: ENV['DATABASE_SERVER'],
-      shard: ENV['SHARD'])
+                   database_server: ENV['DATABASE_SERVER'],
+                   shard: ENV['SHARD'])
       servers = DatabaseServer.all
 
       if database_server
         servers = database_server
         if servers.first == '-'
           negative = true
-          servers = servers[1..-1]
+          servers = servers[1..]
         end
         servers = servers.split(',')
         open = servers.delete('open')
@@ -31,30 +33,27 @@ module Switchman
 
       servers = filter_database_servers_chain.call(servers)
 
-      scope = base_scope.order(::Arel.sql("database_server_id IS NOT NULL, database_server_id, id"))
+      scope = base_scope.order(::Arel.sql('database_server_id IS NOT NULL, database_server_id, id'))
       if servers != DatabaseServer.all
-        conditions = ["database_server_id IN (?)", servers.map(&:id)]
-        conditions.first << " OR database_server_id IS NULL" if servers.include?(Shard.default.database_server)
+        conditions = ['database_server_id IN (?)', servers.map(&:id)]
+        conditions.first << ' OR database_server_id IS NULL' if servers.include?(Shard.default.database_server)
         scope = scope.where(conditions)
       end
 
-      if shard
-        scope = shard_scope(scope, shard)
-      end
+      scope = shard_scope(scope, shard) if shard
 
       scope
     end
 
     def self.options
-      # we still pass through both of these options for back-compat purposes
-      { parallel: ENV['PARALLEL']&.to_i, max_procs: ENV['MAX_PARALLEL_PROCS']&.to_i }
+      { parallel: ENV['PARALLEL'].to_i, max_procs: ENV['MAX_PARALLEL_PROCS'] }
     end
 
-    # categories - an array or proc, to activate as the current shard during the
+    # classes - an array or proc, to activate as the current shard during the
     # task. tasks which modify the schema may want to pass all categories in
     # so that schema updates for non-default tables happen against all shards.
     # this is handled automatically for the default migration tasks, below.
-    def self.shardify_task(task_name, categories: [:primary])
+    def self.shardify_task(task_name, classes: [::ActiveRecord::Base])
       old_task = ::Rake::Task[task_name]
       old_actions = old_task.actions.dup
       old_task.actions.clear
@@ -67,43 +66,30 @@ module Switchman
 
         ::GuardRail.activate(:deploy) do
           Shard.default.database_server.unguard do
-            begin
-              categories = categories.call if categories.respond_to?(:call)
-              Shard.with_each_shard(scope, categories, options) do
-                shard = Shard.current
-                puts "#{shard.id}: #{shard.description}"
-                ::ActiveRecord::Base.connection_pool.spec.config[:shard_name] = Shard.current.name
-                if ::Rails.version < '6.0'
-                  ::ActiveRecord::Base.configurations[::Rails.env] = ::ActiveRecord::Base.connection_pool.spec.config.stringify_keys
-                else
-                  # Adopted from the deprecated code that currently lives in rails proper
-                  remaining_configs = ::ActiveRecord::Base.configurations.configurations.reject { |db_config| db_config.env_name == ::Rails.env }
-                  new_config = ::ActiveRecord::DatabaseConfigurations.new(::Rails.env =>
-                    ::ActiveRecord::Base.connection_pool.spec.config.stringify_keys).configurations
-                  new_configs = remaining_configs + new_config
-      
-                  ::ActiveRecord::Base.configurations = new_configs
-                end
-                shard.database_server.unguard do
-                  old_actions.each { |action| action.call(*task_args) }
-                end
-                nil
+            classes = classes.call if classes.respond_to?(:call)
+            Shard.with_each_shard(scope, classes, **options) do
+              shard = Shard.current
+              puts "#{shard.id}: #{shard.description}"
+
+              shard.database_server.unguard do
+                old_actions.each { |action| action.call(*task_args) }
               end
-            rescue => e
-              puts "Exception from #{e.current_shard.id}: #{e.current_shard.description}" if options[:parallel].to_i != 0
-              raise
+              nil
             end
+          rescue => e
+            puts "Exception from #{e.current_shard.id}: #{e.current_shard.description}" if options[:parallel] != 0
+            raise
+
+            #::ActiveRecord::Base.configurations = old_configurations
           end
         end
       end
     end
 
-    %w{db:migrate db:migrate:up db:migrate:down db:rollback}.each do |task_name|
-      shardify_task(task_name, categories: ->{ Shard.categories })
+    %w[db:migrate db:migrate:up db:migrate:down db:rollback].each do |task_name|
+      shardify_task(task_name, classes: -> { Shard.sharded_models })
     end
 
-    private
-
     def self.shard_scope(scope, raw_shard_ids)
       raw_shard_ids = raw_shard_ids.split(',')
 
@@ -126,6 +112,7 @@ module Switchman
         when /^(-?)(\d+)?\.\.(\.)?(\d+)?$/
           negative, start, open, finish = $1.present?, $2, $3.present?, $4
           raise "Invalid shard id or range: #{id}" unless start || finish
+
           range = []
           range << "id>=#{start}" if start
           range << "id<#{'=' unless open}#{finish}" if finish
@@ -137,12 +124,12 @@ module Switchman
         when %r{^(-?\d+)/(\d+)$}
           numerator = $1.to_i
           denominator = $2.to_i
-          if numerator == 0 || numerator.abs > denominator
-            raise "Invalid fractional chunk: #{id}"
-          end
+          raise "Invalid fractional chunk: #{id}" if numerator.zero? || numerator.abs > denominator
+
           # one chunk means everything
           if denominator == 1
             next if numerator == 1
+
             return scope.none
           end
 
@@ -157,24 +144,25 @@ module Switchman
           select = []
           if index != 1
             subscope = subscope.offset(per_chunk * (index - 1))
-            select << "MIN(id) AS min_id"
+            select << 'MIN(id) AS min_id'
           end
           if index != denominator
             subscope = subscope.limit(per_chunk)
-            select << "MAX(id) AS max_id"
-          end
-
-          result = Shard.from(subscope).select(select.join(", ")).to_a.first
-          if index == 1
-            range = "id<=#{result['max_id']}"
-          elsif index == denominator
-            range = "id>=#{result['min_id']}"
-          else
-            range = "(id>=#{result['min_id']} AND id<=#{result['max_id']})"
+            select << 'MAX(id) AS max_id'
           end
 
-          (numerator < 0 ? negative_ranges : ranges) <<  range
-          else
+          result = Shard.from(subscope).select(select.join(', ')).to_a.first
+          range = case index
+                  when 1
+                    "id<=#{result['max_id']}"
+                  when denominator
+                    "id>=#{result['min_id']}"
+                  else
+                    "(id>=#{result['min_id']} AND id<=#{result['max_id']})"
+                  end
+
+          (numerator.negative? ? negative_ranges : ranges) << range
+        else
           raise "Invalid shard id or range: #{id}"
         end
       end
@@ -183,27 +171,24 @@ module Switchman
       negative_shard_ids.uniq!
       unless shard_ids.empty?
         shard_ids -= negative_shard_ids
-        if shard_ids.empty? && ranges.empty?
-          return scope.none
-        end
+        return scope.none if shard_ids.empty? && ranges.empty?
+
         # we already trimmed them all out; no need to make the server do it as well
         negative_shard_ids = [] if ranges.empty?
       end
 
       conditions = []
       positive_queries = []
-      unless ranges.empty?
-        positive_queries << ranges.join(" OR ")
-      end
+      positive_queries << ranges.join(' OR ') unless ranges.empty?
       unless shard_ids.empty?
-        positive_queries << "id IN (?)"
+        positive_queries << 'id IN (?)'
         conditions << shard_ids
       end
-      positive_query = positive_queries.join(" OR ")
+      positive_query = positive_queries.join(' OR ')
       scope = scope.where(positive_query, *conditions) unless positive_queries.empty?
 
-      scope = scope.where("NOT (#{negative_ranges.join(" OR")})") unless negative_ranges.empty?
-      scope = scope.where("id NOT IN (?)", negative_shard_ids) unless negative_shard_ids.empty?
+      scope = scope.where("NOT (#{negative_ranges.join(' OR')})") unless negative_ranges.empty?
+      scope = scope.where('id NOT IN (?)', negative_shard_ids) unless negative_shard_ids.empty?
       scope
     end
 
@@ -214,18 +199,17 @@ module Switchman
 
   module ActiveRecord
     module PostgreSQLDatabaseTasks
-      def structure_dump(filename, extra_flags=nil)
+      def structure_dump(filename, extra_flags = nil)
         set_psql_env
         args = ['-s', '-x', '-O', '-f', filename]
         args.concat(Array(extra_flags)) if extra_flags
-        search_path = configuration['schema_search_path']
         shard = Shard.current.name
         serialized_search_path = shard
         args << "--schema=#{Shellwords.escape(shard)}"
 
         args << configuration['database']
         run_cmd('pg_dump', args, 'dumping')
-        File.open(filename, "a") { |f| f << "SET search_path TO #{serialized_search_path};\n\n" }
+        File.open(filename, 'a') { |f| f << "SET search_path TO #{serialized_search_path};\n\n" }
       end
     end
   end

metadata

@@ -1,85 +1,85 @@
 --- !ruby/object:Gem::Specification
 name: switchman
 version: !ruby/object:Gem::Version
-  version: 2.2.2
+  version: 3.0.0
 platform: ruby
 authors:
 - Cody Cutrer
 - James Williams
 - Jacob Fugal
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-10-19 00:00:00.000000000 Z
+date: 2021-03-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: railties
+  name: activerecord
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '5.1'
+        version: '6.1'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '6.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '5.1'
+        version: '6.1'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '6.2'
 - !ruby/object:Gem::Dependency
-  name: activerecord
+  name: guardrail
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '5.1'
-    - - "<"
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: 3.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '5.1'
-    - - "<"
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: 3.0.0
 - !ruby/object:Gem::Dependency
-  name: guardrail
+  name: open4
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.0
+        version: 1.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.0
+        version: 1.3.0
 - !ruby/object:Gem::Dependency
-  name: open4
+  name: railties
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: '6.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '6.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: '6.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '6.2'
 - !ruby/object:Gem::Dependency
   name: appraisal
   requirement: !ruby/object:Gem::Requirement
@@ -108,6 +108,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -123,21 +137,21 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: pg
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '13.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '13.0'
 - !ruby/object:Gem::Dependency
-  name: rspec-rails
+  name: rspec-mocks
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
@@ -151,47 +165,75 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.5'
 - !ruby/object:Gem::Dependency
-  name: rspec-mocks
+  name: rspec-rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.5'
+        version: '4.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.5'
+        version: '4.0'
 - !ruby/object:Gem::Dependency
-  name: simplecov
+  name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.15'
+        version: '1.10'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.15'
+        version: '1.10'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: rubocop-rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.0'
+        version: '0.5'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.0'
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.15'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.15'
 description: Sharding
 email:
 - cody@instructure.com
@@ -201,6 +243,7 @@ extra_rdoc_files: []
 files:
 - Rakefile
 - app/models/switchman/shard.rb
+- app/models/switchman/unsharded_record.rb
 - db/migrate/20130328212039_create_switchman_shards.rb
 - db/migrate/20130328224244_create_default_shard.rb
 - db/migrate/20161206323434_add_back_default_string_limits_switchman.rb
@@ -213,10 +256,10 @@ files:
 - lib/switchman/active_record/association.rb
 - lib/switchman/active_record/attribute_methods.rb
 - lib/switchman/active_record/base.rb
-- lib/switchman/active_record/batches.rb
 - lib/switchman/active_record/calculations.rb
-- lib/switchman/active_record/connection_handler.rb
 - lib/switchman/active_record/connection_pool.rb
+- lib/switchman/active_record/database_configurations.rb
+- lib/switchman/active_record/database_configurations/database_config.rb
 - lib/switchman/active_record/finder_methods.rb
 - lib/switchman/active_record/log_subscriber.rb
 - lib/switchman/active_record/migration.rb
@@ -231,12 +274,11 @@ files:
 - lib/switchman/active_record/spawn_methods.rb
 - lib/switchman/active_record/statement_cache.rb
 - lib/switchman/active_record/table_definition.rb
+- lib/switchman/active_record/tasks/database_tasks.rb
 - lib/switchman/active_record/type_caster.rb
-- lib/switchman/active_record/where_clause_factory.rb
 - lib/switchman/active_support/cache.rb
 - lib/switchman/arel.rb
 - lib/switchman/call_super.rb
-- lib/switchman/connection_pool_proxy.rb
 - lib/switchman/database_server.rb
 - lib/switchman/default_shard.rb
 - lib/switchman/engine.rb
@@ -257,7 +299,7 @@ homepage: http://www.instructure.com/
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -265,15 +307,15 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.5'
+      version: '2.6'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.1.4
+signing_key:
 specification_version: 4
 summary: Rails sharding magic
 test_files: []

data/lib/switchman/active_record/batches.rb

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-module Switchman
-  module ActiveRecord
-    module Batches
-      def batch_order
-        ::Arel.sql("#{connection.quote_local_table_name(table_name)}.#{quoted_primary_key} ASC")
-      end
-    end
-  end
-end