added raw_response method check for pagy_elastisearch_rails (#234)

ddnexus · May 8, 2020 · d571719 · d571719
1 parent 7a09bb9
commit d571719
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/pagy/extras/elasticsearch_rails.rb b/lib/pagy/extras/elasticsearch_rails.rb
@@ -28,7 +28,7 @@ def pagy_elasticsearch_rails(pagy_search_args, vars={})
       search_args[-1][:size] = vars[:items]
       search_args[-1][:from] = vars[:items] * (vars[:page] - 1)
       response               = model.search(*search_args)
-      total                  = response.raw_response['hits']['total']
+      total                  = response.respond_to?(:raw_response) ? response.raw_response['hits']['total'] : response.response['hits']['total']
       vars[:count]           = total.is_a?(Hash) ? total['value'] : total
       pagy = Pagy.new(vars)
       # with :last_page overflow we need to re-run the method in order to get the hits