Module: Searchkick

Defined in:
lib/searchkick.rb,
lib/searchkick/index.rb,
lib/searchkick/model.rb,
lib/searchkick/query.rb,
lib/searchkick/indexer.rb,
lib/searchkick/logging.rb,
lib/searchkick/results.rb,
lib/searchkick/version.rb,
lib/searchkick/middleware.rb,
lib/searchkick/index_options.rb,
lib/searchkick/reindex_queue.rb,
lib/searchkick/reindex_v2_job.rb,
lib/searchkick/bulk_reindex_job.rb,
lib/searchkick/process_batch_job.rb,
lib/searchkick/process_queue_job.rb

Defined Under Namespace

Modules: ControllerRuntime, IndexOptions, IndexWithInstrumentation, IndexerWithInstrumentation, Model, QueryWithInstrumentation, SearchkickWithInstrumentation Classes: BulkReindexJob, DangerousOperation, Error, ImportError, Index, Indexer, InvalidQueryError, LogSubscriber, Middleware, MissingIndexError, ProcessBatchJob, ProcessQueueJob, Query, ReindexQueue, ReindexV2Job, Results, UnsupportedVersionError

Constant Summary collapse

VERSION = 
"2.3.1"

Class Attribute Summary collapse

Class Method Summary collapse

Class Attribute Details

.aws_credentials ⇒ Object

Returns the value of attribute aws_credentials.



41
42
43
 
# File 'lib/searchkick.rb', line 41

def aws_credentials
  @aws_credentials
end

.client ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/searchkick.rb', line 50

def self.client
  @client ||= begin
    require "typhoeus/adapters/faraday" if defined?(Typhoeus)

    Elasticsearch::Client.new({
      url: ENV["ELASTICSEARCH_URL"],
      transport_options: {request: {timeout: timeout}, headers: {content_type: "application/json"}}
    }.deep_merge(client_options)) do |f|
      f.use Searchkick::Middleware
      f.request :aws_signers_v4, {
        credentials: Aws::Credentials.new(aws_credentials[:access_key_id], aws_credentials[:secret_access_key]),
        service_name: "es",
        region: aws_credentials[:region] || "us-east-1"
      } if aws_credentials
    end
  end
end

.client_options ⇒ Object

Returns the value of attribute client_options.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def client_options
  @client_options
end

.env ⇒ Object 



68
69
70
 
# File 'lib/searchkick.rb', line 68

def self.env
  @env ||= ENV["RAILS_ENV"] || ENV["RACK_ENV"] || "development"
end

.index_suffix ⇒ Object

Returns the value of attribute index_suffix.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def index_suffix
  @index_suffix
end

.models ⇒ Object

Returns the value of attribute models.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def models
  @models
end

.queue_name ⇒ Object

Returns the value of attribute queue_name.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def queue_name
  @queue_name
end

.redis ⇒ Object

Returns the value of attribute redis.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def redis
  @redis
end

.search_method_name ⇒ Object

Returns the value of attribute search_method_name.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def search_method_name
  @search_method_name
end

.search_timeout ⇒ Object 



72
73
74
 
# File 'lib/searchkick.rb', line 72

def self.search_timeout
  @search_timeout || timeout
end

.timeout ⇒ Object

Returns the value of attribute timeout.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def timeout
  @timeout
end

.wordnet_path ⇒ Object

Returns the value of attribute wordnet_path.



39
40
41
 
# File 'lib/searchkick.rb', line 39

def wordnet_path
  @wordnet_path
end

Class Method Details

.callbacks(value) ⇒ Object 



128
129
130
131
132
133
134
135
136
137
138
139
140
141
 
# File 'lib/searchkick.rb', line 128

def self.callbacks(value)
  if block_given?
    previous_value = callbacks_value
    begin
      self.callbacks_value = value
      yield
      indexer.perform if callbacks_value == :bulk
    ensure
      self.callbacks_value = previous_value
    end
  else
    self.callbacks_value = value
  end
end

.callbacks? ⇒ Boolean

Returns:

  • (Boolean) 


124
125
126
 
# File 'lib/searchkick.rb', line 124

def self.callbacks?
  Thread.current[:searchkick_callbacks_enabled].nil? || Thread.current[:searchkick_callbacks_enabled]
end

.callbacks_value ⇒ Object

private



198
199
200
 
# File 'lib/searchkick.rb', line 198

def self.callbacks_value
  Thread.current[:searchkick_callbacks_enabled]
end

.callbacks_value=(value) ⇒ Object

private



203
204
205
 
# File 'lib/searchkick.rb', line 203

def self.callbacks_value=(value)
  Thread.current[:searchkick_callbacks_enabled] = value
end

.disable_callbacks ⇒ Object 



120
121
122
 
# File 'lib/searchkick.rb', line 120

def self.disable_callbacks
  self.callbacks_value = false
end

.enable_callbacks ⇒ Object

callbacks



116
117
118
 
# File 'lib/searchkick.rb', line 116

def self.enable_callbacks
  self.callbacks_value = nil
end

.indexer ⇒ Object

private



193
194
195
 
# File 'lib/searchkick.rb', line 193

def self.indexer
  Thread.current[:searchkick_indexer] ||= Searchkick::Indexer.new
end

.load_records(records, ids) ⇒ Object

private

Raises:



172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
 
# File 'lib/searchkick.rb', line 172

def self.load_records(records, ids)
  records =
    if records.respond_to?(:primary_key)
      # ActiveRecord
      records.where(records.primary_key => ids) if records.primary_key
    elsif records.respond_to?(:queryable)
      # Mongoid 3+
      records.queryable.for_ids(ids)
    elsif records.respond_to?(:unscoped) && :id.respond_to?(:in)
      # Nobrainer
      records.unscoped.where(:id.in => ids)
    elsif records.respond_to?(:key_column_names)
      records.where(records.key_column_names.first => ids)
    end

  raise Searchkick::Error, "Not sure how to load records" if !records

  records
end

.multi_search(queries) ⇒ Object 



104
105
106
107
108
109
110
111
112
 
# File 'lib/searchkick.rb', line 104

def self.multi_search(queries)
  if queries.any?
    responses = client.msearch(body: queries.flat_map { |q| [q.params.except(:body), q.body] })["responses"]
    queries.each_with_index do |query, i|
      query.handle_response(responses[i])
    end
  end
  queries
end

.reindex_status(index_name) ⇒ Object 



149
150
151
152
153
154
155
156
157
 
# File 'lib/searchkick.rb', line 149

def self.reindex_status(index_name)
  if redis
    batches_left = Searchkick::Index.new(index_name).batches_left
    {
      completed: batches_left == 0,
      batches_left: batches_left
    }
  end
end

.search(term = "*", **options, &block) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/searchkick.rb', line 84

def self.search(term = "*", **options, &block)
  klass = options[:model]

  # TODO add in next major version
  # if !klass
  #   index_name = Array(options[:index_name])
  #   if index_name.size == 1 && index_name.first.respond_to?(:searchkick_index)
  #     klass = index_name.first
  #   end
  # end

  query = Searchkick::Query.new(klass, term, options.except(:model))
  block.call(query.body) if block
  if options[:execute] == false
    query
  else
    query.execute
  end
end

.server_below?(version) ⇒ Boolean

Returns:

  • (Boolean) 


80
81
82
 
# File 'lib/searchkick.rb', line 80

def self.server_below?(version)
  Gem::Version.new(server_version.sub("-", ".")) < Gem::Version.new(version.sub("-", "."))
end

.server_version ⇒ Object 



76
77
78
 
# File 'lib/searchkick.rb', line 76

def self.server_version
  @server_version ||= client.info["version"]["number"]
end

.with_redis ⇒ Object 



159
160
161
162
163
164
165
166
167
168
169
 
# File 'lib/searchkick.rb', line 159

def self.with_redis
  if redis
    if redis.respond_to?(:with)
      redis.with do |r|
        yield r
      end
    else
      yield redis
    end
  end
end