searchkick 2.3.2 → 5.2.1

data/lib/searchkick/bulk_reindex_job.rb

@@ -1,17 +1,21 @@
 module Searchkick
   class BulkReindexJob < ActiveJob::Base
-    queue_as :searchkick
+    queue_as { Searchkick.queue_name }
 
+    # TODO remove min_id and max_id in Searchkick 6
     def perform(class_name:, record_ids: nil, index_name: nil, method_name: nil, batch_id: nil, min_id: nil, max_id: nil)
-      klass = class_name.constantize
-      index = index_name ? Searchkick::Index.new(index_name) : klass.searchkick_index
+      model = Searchkick.load_model(class_name)
+      index = model.searchkick_index(name: index_name)
+
+      # legacy
       record_ids ||= min_id..max_id
-      index.import_scope(
-        Searchkick.load_records(klass, record_ids),
-        method_name: method_name,
-        batch: true,
-        batch_id: batch_id
-      )
+
+      relation = Searchkick.scope(model)
+      relation = Searchkick.load_records(relation, record_ids)
+      relation = relation.search_import if relation.respond_to?(:search_import)
+
+      RecordIndexer.new(index).reindex(relation, mode: :inline, method_name: method_name, full: false)
+      RelationIndexer.new(index).batch_completed(batch_id) if batch_id
     end
   end
 end

data/lib/searchkick/controller_runtime.rb

@@ -0,0 +1,40 @@
+# based on https://gist.github.com/mnutt/566725
+module Searchkick
+  module ControllerRuntime
+    extend ActiveSupport::Concern
+
+    protected
+
+    attr_internal :searchkick_runtime
+
+    def process_action(action, *args)
+      # We also need to reset the runtime before each action
+      # because of queries in middleware or in cases we are streaming
+      # and it won't be cleaned up by the method below.
+      Searchkick::LogSubscriber.reset_runtime
+      super
+    end
+
+    def cleanup_view_runtime
+      searchkick_rt_before_render = Searchkick::LogSubscriber.reset_runtime
+      runtime = super
+      searchkick_rt_after_render = Searchkick::LogSubscriber.reset_runtime
+      self.searchkick_runtime = searchkick_rt_before_render + searchkick_rt_after_render
+      runtime - searchkick_rt_after_render
+    end
+
+    def append_info_to_payload(payload)
+      super
+      payload[:searchkick_runtime] = (searchkick_runtime || 0) + Searchkick::LogSubscriber.reset_runtime
+    end
+
+    module ClassMethods
+      def log_process_action(payload)
+        messages = super
+        runtime = payload[:searchkick_runtime]
+        messages << ("Searchkick: %.1fms" % runtime.to_f) if runtime.to_f > 0
+        messages
+      end
+    end
+  end
+end

data/lib/searchkick/hash_wrapper.rb

@@ -0,0 +1,12 @@
+module Searchkick
+  # Subclass of `Hashie::Mash` to wrap Hash-like structures
+  # (responses from Elasticsearch)
+  #
+  # The primary goal of the subclass is to disable the
+  # warning being printed by Hashie for re-defined
+  # methods, such as `sort`.
+  #
+  class HashWrapper < ::Hashie::Mash
+    disable_warnings if respond_to?(:disable_warnings)
+  end
+end