Sha256: 5f2449b58a7cea517f1936197af86434c1636325ca5239084fb9339f27042515
Contents?: true
Size: 1.36 KB
Versions: 12
Compression:
Stored size: 1.36 KB
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)
scope.with_connection do |connection|
return super unless connection.sqlserver?
return [] if keys.empty?
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
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
12 entries across 12 versions & 1 rubygems