Sha256: 4c70cd7f056b02c160fcdac38b2ff31ed96e0659657e731f6ec905dbe99bfdd0
Contents?: true
Size: 1.7 KB
Versions: 14
Compression:
Stored size: 1.7 KB
Contents
require 'rsolr'
module Query
#
#
class Solr < Base # :nodoc:all
attr_reader :server, :index_types
def initialize *index_types
@server = RSolr.connect rescue nil
super *index_types
end
# # This runs the actual search.
# #
# # TODO Remove!
# #
# def search tokens, offset = 0
# results = nil
#
# duration = timed do
# results = execute(tokens, offset) || empty_results # TODO Does not work yet
# end
# results.duration = duration
#
# results
# end
#
#
def execute tokens, offset = 0
results = Results::Live.new
if server
similar = {}
new_query = tokens.to_solr_query
return results if new_query.empty?
index_types.each do |index|
begin
response = server.select q: new_query, fq: "type:#{index.name}", hl: true, :'hl.fl' => '*', :'hl.simple.pre' => '<', :'hl.simple.post' => '>', facet: true
rescue RSolr::RequestError => re
return results
end
highlighting = response['highlighting']
possibilities = response['response']['docs'].map do |doc|
highlights = highlighting[doc['id'].to_s]
next unless highlights
selected = doc.select { |key| highlights.has_key?(key) }
selected.values.join ' '
end
possibilities.collect! { |possibility| possibility.strip }.uniq!
similar[index.name] = possibilities unless possibilities.empty?
end
results.add similar: similar
end
# TODO
#
class << results
def to_log query
?* + super
end
end
results
end
end
end
Version data entries
14 entries across 14 versions & 1 rubygems