lib/searchkick.rb in searchkick-4.6.2 vs lib/searchkick.rb in searchkick-4.6.3

@@ -32,10 +32,11 @@
 
   # errors
   class Error < StandardError; end
   class MissingIndexError < Error; end
   class UnsupportedVersionError < Error; end
+  # TODO switch to Error
   class InvalidQueryError < Elasticsearch::Transport::Transport::Errors::BadRequest; end
   class DangerousOperation < Error; end
   class ImportError < Error; end
 
   class << self
@@ -275,9 +276,21 @@
     if klass.respond_to?(:current_scope)
       !klass.current_scope.nil?
     elsif defined?(Mongoid::Threaded)
       !Mongoid::Threaded.current_scope(klass).nil?
     end
+  end
+
+  # private
+  def self.not_found_error?(e)
+    (defined?(Elasticsearch) && e.is_a?(Elasticsearch::Transport::Transport::Errors::NotFound)) ||
+    (defined?(OpenSearch) && e.is_a?(OpenSearch::Transport::Transport::Errors::NotFound))
+  end
+
+  # private
+  def self.transport_error?(e)
+    (defined?(Elasticsearch) && e.is_a?(Elasticsearch::Transport::Transport::Error)) ||
+    (defined?(OpenSearch) && e.is_a?(OpenSearch::Transport::Transport::Error))
   end
 end
 
 require "active_model/callbacks"
 ActiveModel::Callbacks.include(Searchkick::Model)