Contents

# frozen_string_literal: true

require "active_record/associations/preloader"

module ActiveRecord
  module ConnectionAdapters
    module SQLServer
      module CoreExt
        module LoaderQuery
          def load_records_for_keys(keys, &block)
            return super unless scope.connection.sqlserver?

            if association_key_name.is_a?(Array)
              query_constraints = Hash.new { |hsh, key| hsh[key] = Set.new }

              keys.each_with_object(query_constraints) do |values_set, constraints|
                association_key_name.zip(values_set).each do |key_name, value|
                  constraints[key_name] << value
                end
              end

              scope.where(query_constraints).load(&block)
            else
              keys.each_slice(in_clause_length).flat_map do |slice|
                scope.where(association_key_name => slice).load(&block).records
              end
            end
          end

          def in_clause_length
            10_000
          end
        end
      end
    end
  end
end

ActiveSupport.on_load(:active_record) do
  mod = ActiveRecord::ConnectionAdapters::SQLServer::CoreExt::LoaderQuery
  ActiveRecord::Associations::Preloader::Association::LoaderQuery.prepend(mod)
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
activerecord-sqlserver-adapter-7.1.1	lib/active_record/connection_adapters/sqlserver/core_ext/preloader.rb
activerecord-sqlserver-adapter-7.1.0	lib/active_record/connection_adapters/sqlserver/core_ext/preloader.rb
activerecord-sqlserver-adapter-7.1.0.rc2	lib/active_record/connection_adapters/sqlserver/core_ext/preloader.rb
activerecord-sqlserver-adapter-7.1.0.rc1	lib/active_record/connection_adapters/sqlserver/core_ext/preloader.rb
activerecord-sqlserver-adapter-7.1.0.beta1	lib/active_record/connection_adapters/sqlserver/core_ext/preloader.rb