42 lines
1.9 KiB
Ruby
42 lines
1.9 KiB
Ruby
![]() |
module Kaminari
|
||
|
# Active Record specific page scope methods implementations
|
||
|
module ActiveRecordRelationMethodsWithLimit
|
||
|
MAX_COUNT_LIMIT = 10_000
|
||
|
|
||
|
# This is a modified version of
|
||
|
# https://github.com/kaminari/kaminari/blob/c5186f5d9b7f23299d115408e62047447fd3189d/kaminari-activerecord/lib/kaminari/activerecord/active_record_relation_methods.rb#L17-L41
|
||
|
# that limit the COUNT query to 10,000 to avoid query timeouts.
|
||
|
# rubocop: disable Gitlab/ModuleWithInstanceVariables
|
||
|
def total_count_with_limit(column_name = :all, _options = nil) #:nodoc:
|
||
|
return @total_count if defined?(@total_count) && @total_count
|
||
|
|
||
|
# There are some cases that total count can be deduced from loaded records
|
||
|
if loaded?
|
||
|
# Total count has to be 0 if loaded records are 0
|
||
|
return @total_count = 0 if (current_page == 1) && @records.empty?
|
||
|
# Total count is calculable at the last page
|
||
|
return @total_count = (current_page - 1) * limit_value + @records.length if @records.any? && (@records.length < limit_value)
|
||
|
end
|
||
|
|
||
|
# #count overrides the #select which could include generated columns referenced in #order, so skip #order here, where it's irrelevant to the result anyway
|
||
|
c = except(:offset, :limit, :order)
|
||
|
# Remove includes only if they are irrelevant
|
||
|
c = c.except(:includes) unless references_eager_loaded_tables?
|
||
|
# .group returns an OrderedHash that responds to #count
|
||
|
# The following line was modified from `c = c.count(:all)`
|
||
|
c = c.limit(MAX_COUNT_LIMIT + 1).count(column_name)
|
||
|
@total_count =
|
||
|
if c.is_a?(Hash) || c.is_a?(ActiveSupport::OrderedHash)
|
||
|
c.count
|
||
|
elsif c.respond_to? :count
|
||
|
c.count(column_name)
|
||
|
else
|
||
|
c
|
||
|
end
|
||
|
end
|
||
|
# rubocop: enable Gitlab/ModuleWithInstanceVariables
|
||
|
|
||
|
Kaminari::ActiveRecordRelationMethods.prepend(self)
|
||
|
end
|
||
|
end
|